未验证 提交 422b7dcf 编写于 作者: X Xiangyu Wang 提交者: GitHub

Add unit tests for mq_factory (#8140)

Signed-off-by: NXiangyu Wang <xiangyu.wang@zilliz.com>
上级 05efb3a0
......@@ -41,6 +41,19 @@ func TestPmsFactory(t *testing.T) {
assert.Nil(t, err)
}
func TestPmsFactory_SetParams(t *testing.T) {
pmsFactory := (*PmsFactory)(nil)
pulsarAddress, _ := Params.Load("_PulsarAddress")
m := map[string]interface{}{
"PulsarAddress": pulsarAddress,
"receiveBufSize": 1024,
"pulsarBufSize": 1024,
}
err := pmsFactory.SetParams(m)
assert.NotNil(t, err)
}
func TestRmsFactory(t *testing.T) {
os.Setenv("ROCKSMQ_PATH", "/tmp/milvus")
defer os.Unsetenv("ROCKSMQ_PATH")
......@@ -48,8 +61,8 @@ func TestRmsFactory(t *testing.T) {
rmsFactory := NewRmsFactory()
m := map[string]interface{}{
"receiveBufSize": 1024,
"pulsarBufSize": 1024,
"ReceiveBufSize": 1024,
"RmqBufSize": 1024,
}
rmsFactory.SetParams(m)
......@@ -63,3 +76,14 @@ func TestRmsFactory(t *testing.T) {
_, err = rmsFactory.NewQueryMsgStream(ctx)
assert.Nil(t, err)
}
func TestRmsFactory_SetParams(t *testing.T) {
rmsFactory := (*RmsFactory)(nil)
m := map[string]interface{}{
"ReceiveBufSize": 1024,
"RmqBufSize": 1024,
}
err := rmsFactory.SetParams(m)
assert.NotNil(t, err)
}
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册