mirror of
https://github.com/apernet/hysteria.git
synced 2025-04-03 04:27:39 +03:00
557 lines
16 KiB
Go
557 lines
16 KiB
Go
package protocol
|
|
|
|
import (
|
|
"bytes"
|
|
"io"
|
|
"reflect"
|
|
"testing"
|
|
)
|
|
|
|
func TestReadTCPRequest(t *testing.T) {
|
|
type args struct {
|
|
r io.Reader
|
|
}
|
|
tests := []struct {
|
|
name string
|
|
args args
|
|
want string
|
|
wantErr bool
|
|
}{
|
|
{
|
|
name: "normal 1",
|
|
args: args{
|
|
r: bytes.NewReader([]byte("\x05hello")),
|
|
},
|
|
want: "hello",
|
|
wantErr: false,
|
|
},
|
|
{
|
|
name: "normal 2",
|
|
args: args{
|
|
r: bytes.NewReader([]byte("\x41\x25We the People We the People We the People We the People We the People We the People We the People We the People We the People We the People We the People We the People We the People We the People We the People We the People We the People We the People We the People We the People We the People")),
|
|
},
|
|
want: "We the People We the People We the People We the People We the People We the People We the People We the People We the People We the People We the People We the People We the People We the People We the People We the People We the People We the People We the People We the People We the People",
|
|
wantErr: false,
|
|
},
|
|
{
|
|
name: "empty",
|
|
args: args{
|
|
r: bytes.NewReader([]byte("\x00")),
|
|
},
|
|
want: "",
|
|
wantErr: true,
|
|
},
|
|
{
|
|
name: "incomplete",
|
|
args: args{
|
|
r: bytes.NewReader([]byte("\x06oh no")),
|
|
},
|
|
want: "oh no\x00",
|
|
wantErr: true,
|
|
},
|
|
{
|
|
name: "too long",
|
|
args: args{
|
|
r: bytes.NewReader([]byte("\x66\x77\x88Whatever")),
|
|
},
|
|
want: "",
|
|
wantErr: true,
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
got, err := ReadTCPRequest(tt.args.r)
|
|
if (err != nil) != tt.wantErr {
|
|
t.Errorf("ReadTCPRequest() error = %v, wantErr %v", err, tt.wantErr)
|
|
return
|
|
}
|
|
if got != tt.want {
|
|
t.Errorf("ReadTCPRequest() got = %v, want %v", got, tt.want)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestWriteTCPRequest(t *testing.T) {
|
|
type args struct {
|
|
addr string
|
|
}
|
|
tests := []struct {
|
|
name string
|
|
args args
|
|
wantW string
|
|
wantErr bool
|
|
}{
|
|
{
|
|
name: "normal 1",
|
|
args: args{
|
|
addr: "hello",
|
|
},
|
|
wantW: "\x44\x01\x05hello",
|
|
wantErr: false,
|
|
},
|
|
{
|
|
name: "normal 2",
|
|
args: args{
|
|
addr: "We the People We the People We the People We the People We the People We the People We the People We the People We the People We the People We the People We the People We the People We the People We the People We the People We the People We the People We the People We the People We the People",
|
|
},
|
|
wantW: "\x44\x01\x41\x25We the People We the People We the People We the People We the People We the People We the People We the People We the People We the People We the People We the People We the People We the People We the People We the People We the People We the People We the People We the People We the People",
|
|
wantErr: false,
|
|
},
|
|
{
|
|
name: "empty",
|
|
args: args{
|
|
addr: "",
|
|
},
|
|
wantW: "\x44\x01\x00",
|
|
wantErr: false,
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
w := &bytes.Buffer{}
|
|
err := WriteTCPRequest(w, tt.args.addr)
|
|
if (err != nil) != tt.wantErr {
|
|
t.Errorf("WriteTCPRequest() error = %v, wantErr %v", err, tt.wantErr)
|
|
return
|
|
}
|
|
if gotW := w.String(); gotW != tt.wantW {
|
|
t.Errorf("WriteTCPRequest() gotW = %v, want %v", []byte(gotW), tt.wantW)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestUDPMessage(t *testing.T) {
|
|
t.Run("buffer too small", func(t *testing.T) {
|
|
// Make sure Serialize returns -1 when the buffer is too small.
|
|
tBuf := make([]byte, 20)
|
|
if (&UDPMessage{
|
|
SessionID: 66,
|
|
PacketID: 77,
|
|
FragID: 2,
|
|
FragCount: 5,
|
|
Addr: "random_addr",
|
|
Data: []byte("random_data"),
|
|
}).Serialize(tBuf) != -1 {
|
|
t.Error("Serialize() did not return -1 when the buffer was too small")
|
|
}
|
|
})
|
|
|
|
type fields struct {
|
|
SessionID uint32
|
|
PacketID uint16
|
|
FragID uint8
|
|
FragCount uint8
|
|
Addr string
|
|
Data []byte
|
|
}
|
|
tests := []struct {
|
|
name string
|
|
fields fields
|
|
want []byte
|
|
}{
|
|
{
|
|
name: "test 1",
|
|
fields: fields{
|
|
SessionID: 1,
|
|
PacketID: 1,
|
|
FragID: 0,
|
|
FragCount: 1,
|
|
Addr: "example.com:80",
|
|
Data: []byte("GET /nothing HTTP/1.1\r\n"),
|
|
},
|
|
want: []byte{0x0, 0x0, 0x0, 0x1, 0x0, 0x1, 0x0, 0x1, 0xe, 0x65, 0x78, 0x61, 0x6d, 0x70, 0x6c, 0x65, 0x2e, 0x63, 0x6f, 0x6d, 0x3a, 0x38, 0x30, 0x47, 0x45, 0x54, 0x20, 0x2f, 0x6e, 0x6f, 0x74, 0x68, 0x69, 0x6e, 0x67, 0x20, 0x48, 0x54, 0x54, 0x50, 0x2f, 0x31, 0x2e, 0x31, 0xd, 0xa},
|
|
},
|
|
{
|
|
name: "test 2",
|
|
fields: fields{
|
|
SessionID: 1329655244,
|
|
Addr: "some_random_goofy_ahh_address_which_is_very_long_some_random_goofy_ahh_address_which_is_very_long_some_random_goofy_ahh_address_which_is_very_long_some_random_goofy_ahh_address_which_is_very_long_some_random_goofy_ahh_address_which_is_very_long_some_random_goofy_ahh_address_which_is_very_long_some_random_goofy_ahh_address_which_is_very_long_some_random_goofy_ahh_address_which_is_very_long_some_random_goofy_ahh_address_which_is_very_long_some_random_goofy_ahh_address_which_is_very_long:9000",
|
|
PacketID: 62233,
|
|
FragID: 8,
|
|
FragCount: 19,
|
|
Data: []byte("God is great, beer is good, and people are crazy."),
|
|
},
|
|
want: []byte{0x4f, 0x40, 0xed, 0xcc, 0xf3, 0x19, 0x8, 0x13, 0x41, 0xee, 0x73, 0x6f, 0x6d, 0x65, 0x5f, 0x72, 0x61, 0x6e, 0x64, 0x6f, 0x6d, 0x5f, 0x67, 0x6f, 0x6f, 0x66, 0x79, 0x5f, 0x61, 0x68, 0x68, 0x5f, 0x61, 0x64, 0x64, 0x72, 0x65, 0x73, 0x73, 0x5f, 0x77, 0x68, 0x69, 0x63, 0x68, 0x5f, 0x69, 0x73, 0x5f, 0x76, 0x65, 0x72, 0x79, 0x5f, 0x6c, 0x6f, 0x6e, 0x67, 0x5f, 0x73, 0x6f, 0x6d, 0x65, 0x5f, 0x72, 0x61, 0x6e, 0x64, 0x6f, 0x6d, 0x5f, 0x67, 0x6f, 0x6f, 0x66, 0x79, 0x5f, 0x61, 0x68, 0x68, 0x5f, 0x61, 0x64, 0x64, 0x72, 0x65, 0x73, 0x73, 0x5f, 0x77, 0x68, 0x69, 0x63, 0x68, 0x5f, 0x69, 0x73, 0x5f, 0x76, 0x65, 0x72, 0x79, 0x5f, 0x6c, 0x6f, 0x6e, 0x67, 0x5f, 0x73, 0x6f, 0x6d, 0x65, 0x5f, 0x72, 0x61, 0x6e, 0x64, 0x6f, 0x6d, 0x5f, 0x67, 0x6f, 0x6f, 0x66, 0x79, 0x5f, 0x61, 0x68, 0x68, 0x5f, 0x61, 0x64, 0x64, 0x72, 0x65, 0x73, 0x73, 0x5f, 0x77, 0x68, 0x69, 0x63, 0x68, 0x5f, 0x69, 0x73, 0x5f, 0x76, 0x65, 0x72, 0x79, 0x5f, 0x6c, 0x6f, 0x6e, 0x67, 0x5f, 0x73, 0x6f, 0x6d, 0x65, 0x5f, 0x72, 0x61, 0x6e, 0x64, 0x6f, 0x6d, 0x5f, 0x67, 0x6f, 0x6f, 0x66, 0x79, 0x5f, 0x61, 0x68, 0x68, 0x5f, 0x61, 0x64, 0x64, 0x72, 0x65, 0x73, 0x73, 0x5f, 0x77, 0x68, 0x69, 0x63, 0x68, 0x5f, 0x69, 0x73, 0x5f, 0x76, 0x65, 0x72, 0x79, 0x5f, 0x6c, 0x6f, 0x6e, 0x67, 0x5f, 0x73, 0x6f, 0x6d, 0x65, 0x5f, 0x72, 0x61, 0x6e, 0x64, 0x6f, 0x6d, 0x5f, 0x67, 0x6f, 0x6f, 0x66, 0x79, 0x5f, 0x61, 0x68, 0x68, 0x5f, 0x61, 0x64, 0x64, 0x72, 0x65, 0x73, 0x73, 0x5f, 0x77, 0x68, 0x69, 0x63, 0x68, 0x5f, 0x69, 0x73, 0x5f, 0x76, 0x65, 0x72, 0x79, 0x5f, 0x6c, 0x6f, 0x6e, 0x67, 0x5f, 0x73, 0x6f, 0x6d, 0x65, 0x5f, 0x72, 0x61, 0x6e, 0x64, 0x6f, 0x6d, 0x5f, 0x67, 0x6f, 0x6f, 0x66, 0x79, 0x5f, 0x61, 0x68, 0x68, 0x5f, 0x61, 0x64, 0x64, 0x72, 0x65, 0x73, 0x73, 0x5f, 0x77, 0x68, 0x69, 0x63, 0x68, 0x5f, 0x69, 0x73, 0x5f, 0x76, 0x65, 0x72, 0x79, 0x5f, 0x6c, 0x6f, 0x6e, 0x67, 0x5f, 0x73, 0x6f, 0x6d, 0x65, 0x5f, 0x72, 0x61, 0x6e, 0x64, 0x6f, 0x6d, 0x5f, 0x67, 0x6f, 0x6f, 0x66, 0x79, 0x5f, 0x61, 0x68, 0x68, 0x5f, 0x61, 0x64, 0x64, 0x72, 0x65, 0x73, 0x73, 0x5f, 0x77, 0x68, 0x69, 0x63, 0x68, 0x5f, 0x69, 0x73, 0x5f, 0x76, 0x65, 0x72, 0x79, 0x5f, 0x6c, 0x6f, 0x6e, 0x67, 0x5f, 0x73, 0x6f, 0x6d, 0x65, 0x5f, 0x72, 0x61, 0x6e, 0x64, 0x6f, 0x6d, 0x5f, 0x67, 0x6f, 0x6f, 0x66, 0x79, 0x5f, 0x61, 0x68, 0x68, 0x5f, 0x61, 0x64, 0x64, 0x72, 0x65, 0x73, 0x73, 0x5f, 0x77, 0x68, 0x69, 0x63, 0x68, 0x5f, 0x69, 0x73, 0x5f, 0x76, 0x65, 0x72, 0x79, 0x5f, 0x6c, 0x6f, 0x6e, 0x67, 0x5f, 0x73, 0x6f, 0x6d, 0x65, 0x5f, 0x72, 0x61, 0x6e, 0x64, 0x6f, 0x6d, 0x5f, 0x67, 0x6f, 0x6f, 0x66, 0x79, 0x5f, 0x61, 0x68, 0x68, 0x5f, 0x61, 0x64, 0x64, 0x72, 0x65, 0x73, 0x73, 0x5f, 0x77, 0x68, 0x69, 0x63, 0x68, 0x5f, 0x69, 0x73, 0x5f, 0x76, 0x65, 0x72, 0x79, 0x5f, 0x6c, 0x6f, 0x6e, 0x67, 0x5f, 0x73, 0x6f, 0x6d, 0x65, 0x5f, 0x72, 0x61, 0x6e, 0x64, 0x6f, 0x6d, 0x5f, 0x67, 0x6f, 0x6f, 0x66, 0x79, 0x5f, 0x61, 0x68, 0x68, 0x5f, 0x61, 0x64, 0x64, 0x72, 0x65, 0x73, 0x73, 0x5f, 0x77, 0x68, 0x69, 0x63, 0x68, 0x5f, 0x69, 0x73, 0x5f, 0x76, 0x65, 0x72, 0x79, 0x5f, 0x6c, 0x6f, 0x6e, 0x67, 0x3a, 0x39, 0x30, 0x30, 0x30, 0x47, 0x6f, 0x64, 0x20, 0x69, 0x73, 0x20, 0x67, 0x72, 0x65, 0x61, 0x74, 0x2c, 0x20, 0x62, 0x65, 0x65, 0x72, 0x20, 0x69, 0x73, 0x20, 0x67, 0x6f, 0x6f, 0x64, 0x2c, 0x20, 0x61, 0x6e, 0x64, 0x20, 0x70, 0x65, 0x6f, 0x70, 0x6c, 0x65, 0x20, 0x61, 0x72, 0x65, 0x20, 0x63, 0x72, 0x61, 0x7a, 0x79, 0x2e},
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
m := &UDPMessage{
|
|
SessionID: tt.fields.SessionID,
|
|
Addr: tt.fields.Addr,
|
|
PacketID: tt.fields.PacketID,
|
|
FragID: tt.fields.FragID,
|
|
FragCount: tt.fields.FragCount,
|
|
Data: tt.fields.Data,
|
|
}
|
|
// Serialize
|
|
buf := make([]byte, MaxUDPSize)
|
|
n := m.Serialize(buf)
|
|
if got := buf[:n]; !reflect.DeepEqual(got, tt.want) {
|
|
t.Errorf("Serialize() = %v, want %v", got, tt.want)
|
|
}
|
|
// Parse back
|
|
if m2, err := ParseUDPMessage(tt.want); err != nil {
|
|
t.Errorf("ParseUDPMessage() error = %v", err)
|
|
} else {
|
|
if !reflect.DeepEqual(m2, m) {
|
|
t.Errorf("ParseUDPMessage() = %v, want %v", m2, m)
|
|
}
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
// TestUDPMessageMalformed is to make sure ParseUDPMessage() fails (but not panic) on malformed data.
|
|
func TestUDPMessageMalformed(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
data []byte
|
|
}{
|
|
{
|
|
name: "empty",
|
|
data: []byte{},
|
|
},
|
|
{
|
|
name: "zeroes 1",
|
|
data: []byte{0, 0, 0, 0},
|
|
},
|
|
{
|
|
name: "zeroes 2",
|
|
data: []byte{0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0},
|
|
},
|
|
{
|
|
name: "incomplete 1",
|
|
data: []byte{0x66, 0xCC, 0xFF, 0xFF, 0x11, 0x22, 0x33, 0x44, 0x55},
|
|
},
|
|
{
|
|
name: "incomplete 2",
|
|
data: []byte{0x66, 0xCC, 0xFF, 0xFF, 0x11, 0x22, 0x33, 0x44, 0x90, 0xAA, 0xBB, 0xCC, 0xDD, 0xEE, 0xFF},
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
if _, err := ParseUDPMessage(tt.data); err == nil {
|
|
t.Errorf("ParseUDPMessage() should fail")
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestReadTCPResponse(t *testing.T) {
|
|
type args struct {
|
|
r io.Reader
|
|
}
|
|
tests := []struct {
|
|
name string
|
|
args args
|
|
want bool
|
|
want1 string
|
|
wantErr bool
|
|
}{
|
|
{
|
|
name: "success 1",
|
|
args: args{
|
|
r: bytes.NewReader([]byte("\x00\x00")),
|
|
},
|
|
want: true,
|
|
want1: "",
|
|
wantErr: false,
|
|
},
|
|
{
|
|
name: "success 2",
|
|
args: args{
|
|
r: bytes.NewReader([]byte("\x00\x12are ya winning son")),
|
|
},
|
|
want: true,
|
|
want1: "are ya winning son",
|
|
wantErr: false,
|
|
},
|
|
{
|
|
name: "failure 1",
|
|
args: args{
|
|
r: bytes.NewReader([]byte("\x01\x00")),
|
|
},
|
|
want: false,
|
|
want1: "",
|
|
wantErr: false,
|
|
},
|
|
{
|
|
name: "failure 2",
|
|
args: args{
|
|
r: bytes.NewReader([]byte("\x01\x15you ain't winning son")),
|
|
},
|
|
want: false,
|
|
want1: "you ain't winning son",
|
|
wantErr: false,
|
|
},
|
|
{
|
|
name: "incomplete",
|
|
args: args{
|
|
r: bytes.NewReader([]byte("\x01\x25princess peach is in another castle")),
|
|
},
|
|
want: false,
|
|
want1: "princess peach is in another castle\x00\x00",
|
|
wantErr: true,
|
|
},
|
|
{
|
|
name: "too long",
|
|
args: args{
|
|
r: bytes.NewReader([]byte("\xAA\xBB\xCCrandom stuff")),
|
|
},
|
|
want: false,
|
|
want1: "",
|
|
wantErr: true,
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
got, got1, err := ReadTCPResponse(tt.args.r)
|
|
if (err != nil) != tt.wantErr {
|
|
t.Errorf("ReadTCPResponse() error = %v, wantErr %v", err, tt.wantErr)
|
|
return
|
|
}
|
|
if got != tt.want {
|
|
t.Errorf("ReadTCPResponse() got = %v, want %v", got, tt.want)
|
|
}
|
|
if got1 != tt.want1 {
|
|
t.Errorf("ReadTCPResponse() got1 = %v, want %v", got1, tt.want1)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestWriteTCPResponse(t *testing.T) {
|
|
type args struct {
|
|
ok bool
|
|
msg string
|
|
}
|
|
tests := []struct {
|
|
name string
|
|
args args
|
|
wantW string
|
|
wantErr bool
|
|
}{
|
|
{
|
|
name: "success 1",
|
|
args: args{
|
|
ok: true,
|
|
msg: "",
|
|
},
|
|
wantW: "\x00\x00",
|
|
},
|
|
{
|
|
name: "success 2",
|
|
args: args{
|
|
ok: true,
|
|
msg: "Welcome XDXDXD",
|
|
},
|
|
wantW: "\x00\x0EWelcome XDXDXD",
|
|
},
|
|
{
|
|
name: "failure 1",
|
|
args: args{
|
|
ok: false,
|
|
msg: "",
|
|
},
|
|
wantW: "\x01\x00",
|
|
},
|
|
{
|
|
name: "failure 2",
|
|
args: args{
|
|
ok: false,
|
|
msg: "me trying to find who u are: ...",
|
|
},
|
|
wantW: "\x01\x20me trying to find who u are: ...",
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
w := &bytes.Buffer{}
|
|
err := WriteTCPResponse(w, tt.args.ok, tt.args.msg)
|
|
if (err != nil) != tt.wantErr {
|
|
t.Errorf("WriteTCPResponse() error = %v, wantErr %v", err, tt.wantErr)
|
|
return
|
|
}
|
|
if gotW := w.String(); gotW != tt.wantW {
|
|
t.Errorf("WriteTCPResponse() gotW = %v, want %v", gotW, tt.wantW)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestWriteUDPRequest(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
wantW string
|
|
wantErr bool
|
|
}{
|
|
{
|
|
name: "normal",
|
|
wantW: "\x44\x02",
|
|
wantErr: false,
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
w := &bytes.Buffer{}
|
|
err := WriteUDPRequest(w)
|
|
if (err != nil) != tt.wantErr {
|
|
t.Errorf("WriteUDPRequest() error = %v, wantErr %v", err, tt.wantErr)
|
|
return
|
|
}
|
|
if gotW := w.String(); gotW != tt.wantW {
|
|
t.Errorf("WriteUDPRequest() gotW = %v, want %v", gotW, tt.wantW)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestReadUDPResponse(t *testing.T) {
|
|
type args struct {
|
|
r io.Reader
|
|
}
|
|
tests := []struct {
|
|
name string
|
|
args args
|
|
want bool
|
|
want1 uint32
|
|
want2 string
|
|
wantErr bool
|
|
}{
|
|
{
|
|
name: "success 1",
|
|
args: args{
|
|
r: bytes.NewReader([]byte("\x00\x00\x00\x00\x02\x00")),
|
|
},
|
|
want: true,
|
|
want1: 2,
|
|
want2: "",
|
|
wantErr: false,
|
|
},
|
|
{
|
|
name: "success 2",
|
|
args: args{
|
|
r: bytes.NewReader([]byte("\x00\x00\x00\x00\x03\x0EWelcome XDXDXD")),
|
|
},
|
|
want: true,
|
|
want1: 3,
|
|
want2: "Welcome XDXDXD",
|
|
wantErr: false,
|
|
},
|
|
{
|
|
name: "failure",
|
|
args: args{
|
|
r: bytes.NewReader([]byte("\x01\x00\x00\x00\x01\x20me trying to find who u are: ...")),
|
|
},
|
|
want: false,
|
|
want1: 1,
|
|
want2: "me trying to find who u are: ...",
|
|
wantErr: false,
|
|
},
|
|
{
|
|
name: "incomplete",
|
|
args: args{
|
|
r: bytes.NewReader([]byte("\x00\x00\x00\x00\x02")),
|
|
},
|
|
want: false,
|
|
want1: 0,
|
|
want2: "",
|
|
wantErr: true,
|
|
},
|
|
{
|
|
name: "too long",
|
|
args: args{
|
|
r: bytes.NewReader([]byte("\x00\x00\x00\x00\x02\xCC\xFF\x66no cap")),
|
|
},
|
|
want: false,
|
|
want1: 0,
|
|
want2: "",
|
|
wantErr: true,
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
got, got1, got2, err := ReadUDPResponse(tt.args.r)
|
|
if (err != nil) != tt.wantErr {
|
|
t.Errorf("ReadUDPResponse() error = %v, wantErr %v", err, tt.wantErr)
|
|
return
|
|
}
|
|
if got != tt.want {
|
|
t.Errorf("ReadUDPResponse() got = %v, want %v", got, tt.want)
|
|
}
|
|
if got1 != tt.want1 {
|
|
t.Errorf("ReadUDPResponse() got1 = %v, want %v", got1, tt.want1)
|
|
}
|
|
if got2 != tt.want2 {
|
|
t.Errorf("ReadUDPResponse() got2 = %v, want %v", got2, tt.want2)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestWriteUDPResponse(t *testing.T) {
|
|
type args struct {
|
|
ok bool
|
|
sessionID uint32
|
|
msg string
|
|
}
|
|
tests := []struct {
|
|
name string
|
|
args args
|
|
wantW string
|
|
wantErr bool
|
|
}{
|
|
{
|
|
name: "success 1",
|
|
args: args{
|
|
ok: true,
|
|
sessionID: 88,
|
|
msg: "",
|
|
},
|
|
wantW: "\x00\x00\x00\x00\x58\x00",
|
|
},
|
|
{
|
|
name: "success 2",
|
|
args: args{
|
|
ok: true,
|
|
sessionID: 233,
|
|
msg: "together forever",
|
|
},
|
|
wantW: "\x00\x00\x00\x00\xE9\x10together forever",
|
|
},
|
|
{
|
|
name: "failure 1",
|
|
args: args{
|
|
ok: false,
|
|
sessionID: 1,
|
|
msg: "",
|
|
},
|
|
wantW: "\x01\x00\x00\x00\x01\x00",
|
|
},
|
|
{
|
|
name: "failure 2",
|
|
args: args{
|
|
ok: false,
|
|
sessionID: 696969,
|
|
msg: "run away run away",
|
|
},
|
|
wantW: "\x01\x00\x0A\xA2\x89\x11run away run away",
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
w := &bytes.Buffer{}
|
|
err := WriteUDPResponse(w, tt.args.ok, tt.args.sessionID, tt.args.msg)
|
|
if (err != nil) != tt.wantErr {
|
|
t.Errorf("WriteUDPResponse() error = %v, wantErr %v", err, tt.wantErr)
|
|
return
|
|
}
|
|
if gotW := w.String(); gotW != tt.wantW {
|
|
t.Errorf("WriteUDPResponse() gotW = %v, want %v", gotW, tt.wantW)
|
|
}
|
|
})
|
|
}
|
|
}
|