add test

Signed-off-by: aoiasd <[email protected]>

Author: aoiasd

internal/flushcommon/pipeline/flow_graph_embedding_node_test.go

@@ -69,6 +69,7 @@ func TestEmbeddingNode_BM25_Operator(t *testing.T) {
 	t.Run("normal case", func(t *testing.T) {
 		node, err := newEmbeddingNode("test-channel", collSchema)
 		assert.NoError(t, err)
+		defer node.Close()
 
 		var output []Msg
 		assert.NotPanics(t, func() {
@@ -110,6 +111,7 @@ func TestEmbeddingNode_BM25_Operator(t *testing.T) {
 	t.Run("with close msg", func(t *testing.T) {
 		node, err := newEmbeddingNode("test-channel", collSchema)
 		assert.NoError(t, err)
+		defer node.Close()
 
 		var output []Msg
 
@@ -127,6 +129,7 @@ func TestEmbeddingNode_BM25_Operator(t *testing.T) {
 	t.Run("prepare insert failed", func(t *testing.T) {
 		node, err := newEmbeddingNode("test-channel", collSchema)
 		assert.NoError(t, err)
+		defer node.Close()
 
 		assert.Panics(t, func() {
 			node.Operate([]Msg{
@@ -148,6 +151,7 @@ func TestEmbeddingNode_BM25_Operator(t *testing.T) {
 	t.Run("embedding failed", func(t *testing.T) {
 		node, err := newEmbeddingNode("test-channel", collSchema)
 		assert.NoError(t, err)
+		defer node.Close()
 
 		node.functionRunners[0].GetSchema().Type = 0
 		assert.Panics(t, func() {