Newer
Older
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
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
package packet
import (
"bytes"
"testing"
"github.com/stretchr/testify/assert"
)
func TestNewPaddingTLV(t *testing.T) {
tests := []struct {
name string
input uint8
expected *PaddingTLV
}{
{
name: "A",
input: 2,
expected: &PaddingTLV{
TLVType: 8,
TLVLength: 2,
PaddingData: []byte{0, 0},
},
},
{
name: "B",
input: 4,
expected: &PaddingTLV{
TLVType: 8,
TLVLength: 4,
PaddingData: []byte{0, 0, 0, 0},
},
},
}
for _, test := range tests {
tlv := NewPaddingTLV(test.input)
assert.Equalf(t, test.expected, tlv, "Test %q", test.name)
}
}
func TestPaddingTLV(t *testing.T) {
tlv := NewPaddingTLV(2)
assert.Equal(t, uint8(8), tlv.Type())
assert.Equal(t, uint8(2), tlv.Length())
assert.Equal(t, &PaddingTLV{
TLVType: 8,
TLVLength: 2,
PaddingData: []byte{0, 0},
}, tlv.Value())
}
func TestPaddingTLVSerialize(t *testing.T) {
tests := []struct {
name string
input *PaddingTLV
expected []byte
}{
{
name: "Full",
input: &PaddingTLV{
TLVType: 8,
TLVLength: 2,
PaddingData: []byte{0, 0},
},
expected: []byte{8, 2, 0, 0},
},
{
name: "Full",
input: &PaddingTLV{
TLVType: 8,
TLVLength: 3,
PaddingData: []byte{0, 0, 0},
},
expected: []byte{8, 3, 0, 0, 0},
},
}
for _, test := range tests {
buf := bytes.NewBuffer(nil)
test.input.Serialize(buf)
assert.Equalf(t, test.expected, buf.Bytes(), "Test %q", test.name)
}
}