admin/milvus
1
0
Fork 0
mirror of https://gitee.com/milvus-io/milvus.git synced 2025-12-08 18:18:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix mq factory ut (#16483)

Browse Source 
Signed-off-by: yun.zhang <yun.zhang@zilliz.com>
This commit is contained in:
jaime 2022-04-14 10:45:34 +08:00 committed by GitHub
parent eae0747f14
commit c7e98b922a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 23 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
internal/mq/msgstream/mq_factory_test.go
View File

@ -18,6 +18,7 @@ package msgstream
import ( import (
"context" "context"
"io/ioutil"
"os" "os"
"testing" "testing"
@ -41,10 +42,13 @@ func TestPmsFactory(t *testing.T) {
func TestRmsFactory(t *testing.T) { func TestRmsFactory(t *testing.T) {
defer os.Unsetenv("ROCKSMQ_PATH") defer os.Unsetenv("ROCKSMQ_PATH")
rmsFactory := NewRmsFactory("tmp/milvus") dir, err := ioutil.TempDir(os.TempDir(), "mq")
assert.Nil(t, err)
rmsFactory := NewRmsFactory(dir)
ctx := context.Background() ctx := context.Background()
_, err := rmsFactory.NewMsgStream(ctx) _, err = rmsFactory.NewMsgStream(ctx)
assert.Nil(t, err) assert.Nil(t, err)
_, err = rmsFactory.NewTtMsgStream(ctx) _, err = rmsFactory.NewTtMsgStream(ctx)
@ -52,4 +56,21 @@ func TestRmsFactory(t *testing.T) {
_, err = rmsFactory.NewQueryMsgStream(ctx) _, err = rmsFactory.NewQueryMsgStream(ctx)
assert.Nil(t, err) assert.Nil(t, err)
err = os.RemoveAll(dir)
assert.Nil(t, err)
}
func TestKafkaFactory(t *testing.T) {
kmsFactory := NewKmsFactory(&Params.KafkaCfg)
ctx := context.Background()
_, err := kmsFactory.NewMsgStream(ctx)
assert.Nil(t, err)
_, err = kmsFactory.NewTtMsgStream(ctx)
assert.Nil(t, err)
_, err = kmsFactory.NewQueryMsgStream(ctx)
assert.Nil(t, err)
} }