40 lines
1.2 KiB
Go
40 lines
1.2 KiB
Go
package frame
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestDataFrameEncode(t *testing.T) {
|
|
var userDataTag byte = 0x15
|
|
d := NewDataFrame()
|
|
d.SetCarriage(userDataTag, []byte("hpds"))
|
|
|
|
tidbuf := []byte(d.TransactionId())
|
|
result := []byte{
|
|
0x80 | byte(TagOfDataFrame), byte(len(tidbuf) + 4 + 8 + 2),
|
|
0x80 | byte(TagOfMetaFrame), byte(len(tidbuf) + 2 + 2),
|
|
byte(TagOfTransactionId), byte(len(tidbuf))}
|
|
result = append(result, tidbuf...)
|
|
result = append(result, byte(TagOfSourceId), 0x0)
|
|
result = append(result, 0x80|byte(TagOfPayloadFrame), 0x06,
|
|
userDataTag, 0x04, 0x79, 0x6F, 0x6D, 0x6F)
|
|
assert.Equal(t, result, d.Encode())
|
|
}
|
|
|
|
func TestDataFrameDecode(t *testing.T) {
|
|
var userDataTag byte = 0x15
|
|
buf := []byte{
|
|
0x80 | byte(TagOfDataFrame), 0x10,
|
|
0x80 | byte(TagOfMetaFrame), 0x06,
|
|
byte(TagOfTransactionId), 0x04, 0x31, 0x32, 0x33, 0x34,
|
|
0x80 | byte(TagOfPayloadFrame), 0x06,
|
|
userDataTag, 0x04, 0x79, 0x6F, 0x6D, 0x6F}
|
|
data, err := DecodeToDataFrame(buf)
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, "1234", data.TransactionId())
|
|
assert.EqualValues(t, userDataTag, data.GetDataTag())
|
|
assert.EqualValues(t, []byte("hpds"), data.GetCarriage())
|
|
}
|