mirror of
https://gitee.com/milvus-io/milvus.git
synced 2026-01-06 02:42:53 +08:00
cb7f2fa6fd
Related to #39095 https://go.dev/doc/modules/version-numbers Update pkg version according to golang dep version convention --------- Signed-off-by: Congqi Xia <congqi.xia@zilliz.com>
37 lines
882 B
Go
37 lines
882 B
Go
package client
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/milvus-io/milvus/pkg/v2/mq/common"
|
|
"github.com/milvus-io/milvus/pkg/v2/mq/mqimpl/rocksmq/server"
|
|
)
|
|
|
|
func TestStreaming(t *testing.T) {
|
|
payload, err := marshalStreamingMessage(&common.ProducerMessage{
|
|
Payload: []byte("payload"),
|
|
Properties: map[string]string{
|
|
"key": "value",
|
|
},
|
|
})
|
|
assert.NoError(t, err)
|
|
assert.NotNil(t, payload)
|
|
|
|
msg, err := unmarshalStreamingMessage("topic", server.ConsumerMessage{
|
|
MsgID: 1,
|
|
Payload: payload,
|
|
})
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, string(msg.Payload()), "payload")
|
|
assert.Equal(t, msg.Properties()["key"], "value")
|
|
msg, err = unmarshalStreamingMessage("topic", server.ConsumerMessage{
|
|
MsgID: 1,
|
|
Payload: payload[1:],
|
|
})
|
|
assert.Error(t, err)
|
|
assert.ErrorIs(t, err, errNotStreamingServiceMessage)
|
|
assert.Nil(t, msg)
|
|
}
|