sub_test.go

package wkproto

import (
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestSbuEncodeAndDecode(t *testing.T) {
	packet := &SubPacket{
		Setting:     Setting(1),
		ChannelID:   "123456",
		ChannelType: 1,
		Action:      Subscribe,
	}

	codec := New()
	// 编码
	packetBytes, err := codec.EncodeFrame(packet, 1)
	assert.NoError(t, err)

	// 解码
	resultPacket, _, err := codec.DecodeFrame(packetBytes, 1)
	assert.NoError(t, err)
	resultSubPacket, ok := resultPacket.(*SubPacket)
	assert.Equal(t, true, ok)

	// 比较
	assert.Equal(t, packet.Setting, resultSubPacket.Setting)
	assert.Equal(t, packet.ChannelID, resultSubPacket.ChannelID)
	assert.Equal(t, packet.ChannelType, resultSubPacket.ChannelType)
	assert.Equal(t, packet.Action, resultSubPacket.Action)

}