test/milvus
1
0
Fork 0
mirror of https://gitee.com/milvus-io/milvus.git synced 2024-12-02 11:59:00 +08:00
Code Issues Actions 12 Packages Projects Releases Wiki Activity

Fix unstable ConsistencyHashPolicy ut (#28375)

Browse Source 
Fixes: #28372

Signed-off-by: yangxuan <xuan.yang@zilliz.com>
This commit is contained in:
XuanYang-cn 2023-11-13 19:06:20 +08:00 committed by GitHub
parent c16fc854d1
commit 87465f07a7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
internal/datacoord/policy_test.go
View File

@ -112,11 +112,18 @@ func TestConsistentHashRegisterPolicy(t *testing.T) {
assert.NotNil(t, updates)
assert.Equal(t, 1, len(updates))
// No Delete operation will be generated
assert.EqualValues(t, &ChannelOp{Type: Add, NodeID: 1, Channels: []RWChannel{channels[0]}}, updates[0])
assert.ElementsMatch(t,
NewChannelOpSet(NewAddOp(1, channels[0])).Collect(),
updates,
)
assert.Equal(t, 1, len(updates[0].GetChannelNames()))
channel := updates[0].GetChannelNames()[0]
// Not stable whether to balance chan-1 or chan-2
if channel == "chan-1" {
assert.EqualValues(t, &ChannelOp{Type: Add, NodeID: 1, Channels: []RWChannel{channels[0]}}, updates[0])
}
if channel == "chan-2" {
assert.EqualValues(t, &ChannelOp{Type: Add, NodeID: 1, Channels: []RWChannel{channels[1]}}, updates[0])
}
})
}