1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
|
package hotline
import (
"reflect"
"testing"
)
func TestTrackerRegistration_Payload(t *testing.T) {
type fields struct {
Port []byte
UserCount int
PassID []byte
Name string
Description string
}
tests := []struct {
name string
fields fields
want []byte
}{
{
name: "returns expected payload bytes",
fields: fields{
Port: []byte{0x00, 0x10},
UserCount: 2,
PassID: []byte{0x00, 0x00, 0x00, 0x01},
Name: "Test Serv",
Description: "Fooz",
},
want: []byte{
0x00, 0x01,
0x00, 0x10,
0x00, 0x02,
0x00, 0x00,
0x00, 0x00, 0x00, 0x01,
0x09,
0x54, 0x65, 0x73, 0x74, 0x20, 0x53, 0x65, 0x72, 0x76,
0x04,
0x46, 0x6f, 0x6f, 0x7a,
},
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
tr := &TrackerRegistration{
Port: tt.fields.Port,
UserCount: tt.fields.UserCount,
PassID: tt.fields.PassID,
Name: tt.fields.Name,
Description: tt.fields.Description,
}
if got := tr.Payload(); !reflect.DeepEqual(got, tt.want) {
t.Errorf("Payload() = %v, want %v", got, tt.want)
}
})
}
}
|
