From 3c9840d6d3daff609a57cf71f44642c657f60e60 Mon Sep 17 00:00:00 2001 From: congqixia Date: Mon, 26 Dec 2022 16:59:30 +0800 Subject: [PATCH] Fix some datanode instance not stopped in unit tests (#21385) Signed-off-by: Congqi Xia --- internal/datanode/data_node_test.go | 1 + internal/datanode/flow_graph_manager_test.go | 1 + 2 files changed, 2 insertions(+) diff --git a/internal/datanode/data_node_test.go b/internal/datanode/data_node_test.go index d96389304b..f74c9bf21c 100644 --- a/internal/datanode/data_node_test.go +++ b/internal/datanode/data_node_test.go @@ -185,6 +185,7 @@ func TestDataNode(t *testing.T) { t.Run("Test BackGroundGC", func(t *testing.T) { ctx, cancel := context.WithCancel(context.Background()) node := newIDLEDataNodeMock(ctx, schemapb.DataType_Int64) + defer node.Stop() vchanNameCh := make(chan string) node.clearSignal = vchanNameCh diff --git a/internal/datanode/flow_graph_manager_test.go b/internal/datanode/flow_graph_manager_test.go index cc6c5e2f39..28d6e74f9b 100644 --- a/internal/datanode/flow_graph_manager_test.go +++ b/internal/datanode/flow_graph_manager_test.go @@ -45,6 +45,7 @@ func TestFlowGraphManager(t *testing.T) { defer etcdCli.Close() node := newIDLEDataNodeMock(ctx, schemapb.DataType_Int64) + defer node.Stop() node.SetEtcdClient(etcdCli) err = node.Init() require.Nil(t, err)