Add unittest for rootcoord util (#7688)

Signed-off-by: yhmo <yihua.mo@zilliz.com>
This commit is contained in:
groot 2021-09-10 16:41:48 +08:00 committed by GitHub
parent db01040b53
commit 020f109dd8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -14,13 +14,14 @@ package rootcoord
import (
"testing"
"github.com/milvus-io/milvus/internal/msgstream"
"github.com/milvus-io/milvus/internal/proto/commonpb"
"github.com/milvus-io/milvus/internal/proto/etcdpb"
"github.com/milvus-io/milvus/internal/proto/schemapb"
"github.com/stretchr/testify/assert"
)
func TestEqualKeyPairArray(t *testing.T) {
func Test_EqualKeyPairArray(t *testing.T) {
p1 := []*commonpb.KeyValuePair{
{
Key: "k1",
@ -98,3 +99,39 @@ func Test_ToPhysicalChannel(t *testing.T) {
assert.Equal(t, "abc__", ToPhysicalChannel("abc___defgsg"))
assert.Equal(t, "abcdef", ToPhysicalChannel("abcdef"))
}
func Test_EncodeMsgPositions(t *testing.T) {
mp := &msgstream.MsgPosition{
ChannelName: "test",
MsgID: []byte{1, 2, 3},
}
str, err := EncodeMsgPositions([]*msgstream.MsgPosition{})
assert.Empty(t, str)
assert.Nil(t, err)
mps := []*msgstream.MsgPosition{mp}
str, err = EncodeMsgPositions(mps)
assert.NotEmpty(t, str)
assert.Nil(t, err)
}
func Test_DecodeMsgPositions(t *testing.T) {
mp := &msgstream.MsgPosition{
ChannelName: "test",
MsgID: []byte{1, 2, 3},
}
str, err := EncodeMsgPositions([]*msgstream.MsgPosition{mp})
assert.Nil(t, err)
mpOut := make([]*msgstream.MsgPosition, 1)
err = DecodeMsgPositions(str, &mpOut)
assert.Nil(t, err)
err = DecodeMsgPositions("", &mpOut)
assert.Nil(t, err)
err = DecodeMsgPositions("null", &mpOut)
assert.Nil(t, err)
}