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
|
package codec
import (
"bytes"
"testing"
"git.fd.io/govpp.git/api"
)
type MyMsg struct {
Index uint16
Label []byte `struc:"[16]byte"`
Port uint16
}
func (*MyMsg) GetMessageName() string {
return "my_msg"
}
func (*MyMsg) GetCrcString() string {
return "xxxxx"
}
func (*MyMsg) GetMessageType() api.MessageType {
return api.OtherMessage
}
func TestEncode(t *testing.T) {
tests := []struct {
name string
msg api.Message
msgID uint16
expData []byte
}{
{name: "basic",
msg: &MyMsg{Index: 1, Label: []byte("Abcdef"), Port: 1000},
msgID: 100,
expData: []byte{0x00, 0x64, 0x00, 0x01, 0x41, 0x62, 0x63, 0x64, 0x65, 0x66, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x03, 0xE8},
},
}
for _, test := range tests {
t.Run(test.name, func(t *testing.T) {
c := &MsgCodec{}
data, err := c.EncodeMsg(test.msg, test.msgID)
if err != nil {
t.Fatalf("expected nil error, got: %v", err)
}
if !bytes.Equal(data, test.expData) {
t.Fatalf("expected data: % 0X, got: % 0X", test.expData, data)
}
})
}
}
func TestEncodePanic(t *testing.T) {
c := &MsgCodec{}
msg := &MyMsg{Index: 1, Label: []byte("thisIsLongerThan16Bytes"), Port: 1000}
_, err := c.EncodeMsg(msg, 100)
if err == nil {
t.Fatalf("expected non-nil error, got: %v", err)
}
}
|
