Skip to content

Commit e02eca4

Browse files
authored
fix(dot/network): fix stream manager tests (ChainSafe#1683)
1 parent d36c3b7 commit e02eca4

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

dot/network/stream_manager_test.go

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -71,11 +71,11 @@ func TestStreamManager(t *testing.T) {
7171

7272
time.Sleep(cleanupStreamInterval * 2)
7373
connsAToB := ha.Network().ConnsToPeer(hb.ID())
74-
require.Equal(t, 1, len(connsAToB))
74+
require.GreaterOrEqual(t, len(connsAToB), 1)
7575
require.Equal(t, 0, len(connsAToB[0].GetStreams()))
7676

7777
connsBToA := hb.Network().ConnsToPeer(ha.ID())
78-
require.Equal(t, 1, len(connsBToA))
78+
require.GreaterOrEqual(t, len(connsBToA), 1)
7979
require.Equal(t, 0, len(connsBToA[0].GetStreams()))
8080
}
8181

@@ -91,12 +91,12 @@ func TestStreamManager_KeepStream(t *testing.T) {
9191
smA.start()
9292
smB.start()
9393

94-
time.Sleep(cleanupStreamInterval / 2)
94+
time.Sleep(cleanupStreamInterval / 3)
9595
connsAToB := ha.Network().ConnsToPeer(hb.ID())
96-
require.Equal(t, 1, len(connsAToB))
96+
require.GreaterOrEqual(t, len(connsAToB), 1)
9797
require.Equal(t, 1, len(connsAToB[0].GetStreams()))
9898

9999
connsBToA := hb.Network().ConnsToPeer(ha.ID())
100-
require.Equal(t, 1, len(connsBToA))
100+
require.GreaterOrEqual(t, len(connsBToA), 1)
101101
require.Equal(t, 1, len(connsBToA[0].GetStreams()))
102102
}

0 commit comments

Comments
 (0)