milvus/pkg/streaming/util/message/encoder_test.go
chyezh 39c7e06bc5
enhance: add message and msgstream msgpack adaptor (#34874)
issue: #33285

- make message builder and message conversion type safe
- add adaptor type and function to adapt old msgstream msgpack and
message interface

---------

Signed-off-by: chyezh <chyezh@outlook.com>
2024-07-22 20:59:42 +08:00

28 lines
570 B
Go

package message
import (
"testing"
"github.com/stretchr/testify/assert"
)
func TestEncoder(t *testing.T) {
result, err := DecodeInt64(EncodeInt64(1))
assert.NoError(t, err)
assert.Equal(t, int64(1), result)
result2, err := DecodeUint64(EncodeUint64(1))
assert.NoError(t, err)
assert.Equal(t, uint64(1), result2)
result3, err := EncodeProto(&InsertMessageHeader{
CollectionId: 1,
})
assert.NoError(t, err)
var result4 InsertMessageHeader
err = DecodeProto(result3, &result4)
assert.NoError(t, err)
assert.Equal(t, result4.CollectionId, int64(1))
}