dragonflyoss
/
dragonfly
mirror of https://github.com/dragonflyoss/dragonfly.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

test: FilterParentCount (#1094) 

Signed-off-by: Gaius <gaius.qi@gmail.com>
This commit is contained in:
Gaius 2022-02-24 21:50:54 +08:00
parent d0665728e0
commit db93339f9d
No known key found for this signature in database
GPG Key ID: 8B4E5D1290FA2FFB
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
scheduler/scheduler/scheduler_test.go
View File

@ -817,6 +817,7 @@ func TestScheduler_FindParent(t *testing.T) {
mockPeers[0].Pieces.Set(0) mockPeers[0].Pieces.Set(0)
mockPeers[1].Pieces.Set(0) mockPeers[1].Pieces.Set(0)
mockPeers[1].Pieces.Set(1) mockPeers[1].Pieces.Set(1)
mockPeers[1].Pieces.Set(2)
md.GetSchedulerClusterConfig().Return(types.SchedulerClusterConfig{ md.GetSchedulerClusterConfig().Return(types.SchedulerClusterConfig{
FilterParentCount: 1, 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) { expect: func(t *testing.T, mockPeers []*resource.Peer, parent *resource.Peer, ok bool) {
assert := assert.New(t) assert := assert.New(t)
assert.True(ok) assert.True(ok)
assert.Equal(mockPeers[0].ID, parent.ID) assert.Contains([]string{mockPeers[0].ID, mockPeers[1].ID}, parent.ID)
}, },
}, },
} }