fix(tests): fix FraudProofHandling test

vgonkivs · Sep 28, 2023 · 40beba5 · 40beba5
1 parent 849cb67
commit 40beba5
Showing 1 changed file with 16 additions and 4 deletions.
diff --git a/nodebuilder/tests/fraud_test.go b/nodebuilder/tests/fraud_test.go
@@ -109,11 +109,23 @@ func TestFraudProofHandling(t *testing.T) {
 	// FIXME: Eventually, this should be a check on service registry managing and keeping
 	//  lifecycles of each Module.
 	// 6.
-	syncCtx, syncCancel := context.WithTimeout(context.Background(), blockTime*5)
-	_, err = fullClient.Header.WaitForHeight(syncCtx, 15)
-	require.ErrorIs(t, err, context.DeadlineExceeded)
-	syncCancel()
+	attempts := 0
+	timeOut := blockTime * 5
+	height := uint64(15)
+	for attempts < 5 {
+		syncCtx, syncCancel := context.WithTimeout(context.Background(), timeOut)
+		_, err = full.HeaderServ.WaitForHeight(syncCtx, height)
+		syncCancel()
+		if err != nil {
+			break
+		}
+		attempts++
+		timeOut = blockTime
+		height++
+	}
 
+	require.NotNil(t, err)
+	require.ErrorIs(t, err, context.DeadlineExceeded)
 	// 7.
 	cfg = nodebuilder.DefaultConfig(node.Light)
 	cfg.Header.TrustedPeers = append(cfg.Header.TrustedPeers, addrs[0].String())