From db93339f9da98271a2568ce2d802343e1e754f23 Mon Sep 17 00:00:00 2001 From: Gaius Date: Thu, 24 Feb 2022 21:50:54 +0800 Subject: [PATCH] test: FilterParentCount (#1094) Signed-off-by: Gaius --- scheduler/scheduler/scheduler_test.go | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/scheduler/scheduler/scheduler_test.go b/scheduler/scheduler/scheduler_test.go index c1900425a..f9b6f8473 100644 --- a/scheduler/scheduler/scheduler_test.go +++ b/scheduler/scheduler/scheduler_test.go @@ -817,6 +817,7 @@ func TestScheduler_FindParent(t *testing.T) { mockPeers[0].Pieces.Set(0) mockPeers[1].Pieces.Set(0) mockPeers[1].Pieces.Set(1) + mockPeers[1].Pieces.Set(2) md.GetSchedulerClusterConfig().Return(types.SchedulerClusterConfig{ FilterParentCount: 1, @@ -825,7 +826,7 @@ func TestScheduler_FindParent(t *testing.T) { expect: func(t *testing.T, mockPeers []*resource.Peer, parent *resource.Peer, ok bool) { assert := assert.New(t) assert.True(ok) - assert.Equal(mockPeers[0].ID, parent.ID) + assert.Contains([]string{mockPeers[0].ID, mockPeers[1].ID}, parent.ID) }, }, }