refactor: enhance code coverage

Tochemey · Feb 20, 2025 · 1c8517c · 1c8517c
1 parent c77a03f
commit 1c8517c
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/actor/rebalancer_test.go b/actor/rebalancer_test.go
@@ -222,12 +222,14 @@ func TestRebalancing(t *testing.T) {
 
 		// take down node1 since it is the first node created in the cluster
 		require.NoError(t, node1.Stop(ctx))
-		util.Pause(time.Minute)
 		require.NoError(t, sd1.Close())
-		util.Pause(time.Minute)
 
-		_, _, err = node2.ActorOf(ctx, "actorName")
-		require.NoError(t, err)
+		require.Eventually(t, func() bool {
+			if _, _, err := node2.ActorOf(ctx, "actorName"); err != nil {
+				return false
+			}
+			return true
+		}, 2*time.Minute, time.Second)
 
 		assert.NoError(t, node2.Stop(ctx))
 		assert.NoError(t, node3.Stop(ctx))