diff --git a/internal/msgstream/mq_factory_test.go b/internal/msgstream/mq_factory_test.go
index 72c2c283a1178cbdea2db51a3cd23ba3f7dbd648..c6c04841145eba233ebd60791bf71a4eb9aefd58 100644
--- a/internal/msgstream/mq_factory_test.go
+++ b/internal/msgstream/mq_factory_test.go
@@ -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)
+}