From c2fa2e940db0cfd9953c6e6a13edddc22a563611 Mon Sep 17 00:00:00 2001 From: godchen Date: Thu, 21 Oct 2021 10:57:03 +0800 Subject: [PATCH] Check cluster error (#10116) Signed-off-by: godchen --- internal/datacoord/cluster_test.go | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/internal/datacoord/cluster_test.go b/internal/datacoord/cluster_test.go index e9c381e9ed..f9eb78e4f9 100644 --- a/internal/datacoord/cluster_test.go +++ b/internal/datacoord/cluster_test.go @@ -392,7 +392,8 @@ func TestConsistentHashPolicy(t *testing.T) { } hash.Remove("1") - cluster.UnRegister(nodeInfo1) + err = cluster.UnRegister(nodeInfo1) + assert.Nil(t, err) for _, c := range channels { idstr, err := hash.Get(c) assert.Nil(t, err) @@ -403,7 +404,8 @@ func TestConsistentHashPolicy(t *testing.T) { } hash.Remove("2") - cluster.UnRegister(nodeInfo2) + err = cluster.UnRegister(nodeInfo2) + assert.Nil(t, err) for _, c := range channels { idstr, err := hash.Get(c) assert.Nil(t, err) @@ -414,7 +416,8 @@ func TestConsistentHashPolicy(t *testing.T) { } hash.Remove("3") - cluster.UnRegister(nodeInfo3) + err = cluster.UnRegister(nodeInfo3) + assert.Nil(t, err) bufferChannels := channelManager.GetBuffer() assert.EqualValues(t, 3, len(bufferChannels.Channels)) }