fix downstream manager test (#2950)

pull/2968/head
Todd 3 years ago
parent cf412b95a2
commit 3461caee41

@ -105,7 +105,7 @@ func TestNewDownstreamManager(t *testing.T) {
dm.addConnection("2", w2)
got := dm.Connected().UnmappedKeyIds()
sort.Strings(got)
assert.Equal(t, []string{"1", "2"}, got)
assert.ElementsMatch(t, []string{"1", "2"}, got)
dm.disconnectKeys("1")
assert.ErrorIs(t, w1.SetReadDeadline(time.Now()), io.ErrClosedPipe)
@ -189,7 +189,7 @@ func TestNewDownstreamManager(t *testing.T) {
dm.mapKeyToWorkerId("1", "w1")
dm.mapKeyToWorkerId("2", "w2")
assert.Equal(t, []string{"w1", "w2"}, dm.Connected().WorkerIds())
assert.ElementsMatch(t, []string{"w1", "w2"}, dm.Connected().WorkerIds())
assert.Empty(t, dm.Connected().UnmappedKeyIds())
assert.NoError(t, w1.SetReadDeadline(time.Now()))

Loading…
Cancel
Save