milvus/pkg/streaming/util/message/message_test.go
chyezh d2bc4a53be
enhance: implement rmq and pulsar as wal (#34046)
issue: #33285

- use reader but not consumer for pulsar
- advanced test framework
- move some streaming related package into pkg

---------

Signed-off-by: chyezh <chyezh@outlook.com>
2024-06-27 15:11:05 +08:00

28 lines
576 B
Go

package message
import (
"testing"
"github.com/stretchr/testify/assert"
)
func TestMessageType(t *testing.T) {
s := MessageTypeUnknown.marshal()
assert.Equal(t, "0", s)
typ := unmarshalMessageType("0")
assert.Equal(t, MessageTypeUnknown, typ)
typ = unmarshalMessageType("882s9")
assert.Equal(t, MessageTypeUnknown, typ)
}
func TestVersion(t *testing.T) {
v := newMessageVersionFromString("")
assert.Equal(t, VersionOld, v)
assert.Panics(t, func() {
newMessageVersionFromString("s1")
})
v = newMessageVersionFromString("1")
assert.Equal(t, VersionV1, v)
}