simplify wait condition

0xPolygon · Oct 23, 2023 · 940408f · 940408f
1 parent 173b2cd
commit 940408f
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/e2e-polybft/property/property_test.go b/e2e-polybft/property/property_test.go
@@ -124,10 +124,6 @@ func TestProperty_DropValidators(t *testing.T) {
 			node.Start()
 		}
 
-		time.Sleep(2 * blockTime)
-		currentBlock, err = activeValidator.JSONRPC().Eth().GetBlockByNumber(ethgo.Latest, false)
-		require.NoError(t, err)
-		// check that block production is restarted
-		require.True(t, oldBlockNumber < currentBlock.Number)
+		require.NoError(t, cluster.WaitForBlock(oldBlockNumber+1, 30*time.Second))
 	})
 }