From 7e830f6185ccc493054ac7bb787fd40cc137d066 Mon Sep 17 00:00:00 2001 From: XuanYang-cn Date: Fri, 19 Apr 2024 15:11:20 +0800 Subject: [PATCH] fix: [skip-e2e] TestAvgReassignPolicy ut (#32450) See also: #32448 Signed-off-by: yangxuan --- internal/datacoord/policy_test.go | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/internal/datacoord/policy_test.go b/internal/datacoord/policy_test.go index 2ccbdfa6eb..fbd7c9d95d 100644 --- a/internal/datacoord/policy_test.go +++ b/internal/datacoord/policy_test.go @@ -22,6 +22,7 @@ import ( "github.com/samber/lo" "github.com/stretchr/testify/assert" + "github.com/stretchr/testify/require" memkv "github.com/milvus-io/milvus/internal/kv/mem" ) @@ -409,7 +410,15 @@ func TestAvgReassignPolicy(t *testing.T) { } t.Run(tt.name, func(t *testing.T) { got := AverageReassignPolicy(tt.args.store, tt.args.reassigns) - assert.ElementsMatch(t, tt.want.Collect(), got.Collect()) + + wantMap, gotMap := tt.want.SplitByChannel(), got.SplitByChannel() + assert.ElementsMatch(t, lo.Keys(wantMap), lo.Keys(gotMap)) + + for k, opSet := range wantMap { + gotOpSet, ok := gotMap[k] + require.True(t, ok) + assert.ElementsMatch(t, opSet.Collect(), gotOpSet.Collect()) + } }) } }