fix(dot/network): fix stream manager tests (#1683)

ChainSafe · Jul 8, 2021 · e02eca4 · e02eca4
1 parent d36c3b7
commit e02eca4
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/dot/network/stream_manager_test.go b/dot/network/stream_manager_test.go
@@ -71,11 +71,11 @@ func TestStreamManager(t *testing.T) {
 
 	time.Sleep(cleanupStreamInterval * 2)
 	connsAToB := ha.Network().ConnsToPeer(hb.ID())
-	require.Equal(t, 1, len(connsAToB))
+	require.GreaterOrEqual(t, len(connsAToB), 1)
 	require.Equal(t, 0, len(connsAToB[0].GetStreams()))
 
 	connsBToA := hb.Network().ConnsToPeer(ha.ID())
-	require.Equal(t, 1, len(connsBToA))
+	require.GreaterOrEqual(t, len(connsBToA), 1)
 	require.Equal(t, 0, len(connsBToA[0].GetStreams()))
 }
 
@@ -91,12 +91,12 @@ func TestStreamManager_KeepStream(t *testing.T) {
 	smA.start()
 	smB.start()
 
-	time.Sleep(cleanupStreamInterval / 2)
+	time.Sleep(cleanupStreamInterval / 3)
 	connsAToB := ha.Network().ConnsToPeer(hb.ID())
-	require.Equal(t, 1, len(connsAToB))
+	require.GreaterOrEqual(t, len(connsAToB), 1)
 	require.Equal(t, 1, len(connsAToB[0].GetStreams()))
 
 	connsBToA := hb.Network().ConnsToPeer(ha.ID())
-	require.Equal(t, 1, len(connsBToA))
+	require.GreaterOrEqual(t, len(connsBToA), 1)
 	require.Equal(t, 1, len(connsBToA[0].GetStreams()))
 }