From c7e98b922a49d05d2ac9e67ceb70054bb64a47dc Mon Sep 17 00:00:00 2001 From: jaime Date: Thu, 14 Apr 2022 10:45:34 +0800 Subject: [PATCH] Fix mq factory ut (#16483) Signed-off-by: yun.zhang --- internal/mq/msgstream/mq_factory_test.go | 25 ++++++++++++++++++++++-- 1 file changed, 23 insertions(+), 2 deletions(-) diff --git a/internal/mq/msgstream/mq_factory_test.go b/internal/mq/msgstream/mq_factory_test.go index 452bcff341..f4055c40e1 100644 --- a/internal/mq/msgstream/mq_factory_test.go +++ b/internal/mq/msgstream/mq_factory_test.go @@ -18,6 +18,7 @@ package msgstream import ( "context" + "io/ioutil" "os" "testing" @@ -41,10 +42,13 @@ func TestPmsFactory(t *testing.T) { func TestRmsFactory(t *testing.T) { 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() - _, err := rmsFactory.NewMsgStream(ctx) + _, err = rmsFactory.NewMsgStream(ctx) assert.Nil(t, err) _, err = rmsFactory.NewTtMsgStream(ctx) @@ -52,4 +56,21 @@ func TestRmsFactory(t *testing.T) { _, err = rmsFactory.NewQueryMsgStream(ctx) 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) }