test: wait for indexer health before proof calls (fix flaky test-validator races)

Author: ananas-block

account-comparison/programs/account-comparison/tests/test_compressed_account.rs

@@ -54,7 +54,16 @@ async fn start_validator_and_connect() -> LightClient {
         tokio::time::sleep(Duration::from_secs(1)).await;
         rpc = LightClient::new(LightClientConfig::local()).await.unwrap();
     }
-    tokio::time::sleep(Duration::from_secs(10)).await;
+
+    // Wait for the indexer + prover to be ready (not just the validator RPC),
+    // otherwise proof requests race the still-initializing indexer/prover.
+    for attempt in 0..120 {
+        if matches!(rpc.get_indexer_health(None).await, Ok(true)) {
+            break;
+        }
+        assert!(attempt < 119, "indexer did not become healthy in time");
+        tokio::time::sleep(Duration::from_secs(1)).await;
+    }
 
     rpc
 }

basic-operations/anchor/burn/programs/burn/tests/test.rs

@@ -52,7 +52,16 @@ async fn start_validator_and_connect() -> LightClient {
         tokio::time::sleep(Duration::from_secs(1)).await;
         rpc = LightClient::new(LightClientConfig::local()).await.unwrap();
     }
-    tokio::time::sleep(Duration::from_secs(10)).await;
+
+    // Wait for the indexer + prover to be ready (not just the validator RPC),
+    // otherwise proof requests race the still-initializing indexer/prover.
+    for attempt in 0..120 {
+        if matches!(rpc.get_indexer_health(None).await, Ok(true)) {
+            break;
+        }
+        assert!(attempt < 119, "indexer did not become healthy in time");
+        tokio::time::sleep(Duration::from_secs(1)).await;
+    }
 
     rpc
 }

basic-operations/anchor/close/programs/close/tests/test.rs

@@ -51,7 +51,16 @@ async fn start_validator_and_connect() -> LightClient {
         tokio::time::sleep(Duration::from_secs(1)).await;
         rpc = LightClient::new(LightClientConfig::local()).await.unwrap();
     }
-    tokio::time::sleep(Duration::from_secs(10)).await;
+
+    // Wait for the indexer + prover to be ready (not just the validator RPC),
+    // otherwise proof requests race the still-initializing indexer/prover.
+    for attempt in 0..120 {
+        if matches!(rpc.get_indexer_health(None).await, Ok(true)) {
+            break;
+        }
+        assert!(attempt < 119, "indexer did not become healthy in time");
+        tokio::time::sleep(Duration::from_secs(1)).await;
+    }
 
     rpc
 }

basic-operations/anchor/create/programs/create/tests/test.rs

@@ -55,7 +55,16 @@ async fn start_validator_and_connect() -> LightClient {
         tokio::time::sleep(Duration::from_secs(1)).await;
         rpc = LightClient::new(LightClientConfig::local()).await.unwrap();
     }
-    tokio::time::sleep(Duration::from_secs(10)).await;
+
+    // Wait for the indexer + prover to be ready (not just the validator RPC),
+    // otherwise proof requests race the still-initializing indexer/prover.
+    for attempt in 0..120 {
+        if matches!(rpc.get_indexer_health(None).await, Ok(true)) {
+            break;
+        }
+        assert!(attempt < 119, "indexer did not become healthy in time");
+        tokio::time::sleep(Duration::from_secs(1)).await;
+    }
 
     rpc
 }

basic-operations/anchor/reinit/programs/reinit/tests/test.rs

@@ -54,7 +54,16 @@ async fn start_validator_and_connect() -> LightClient {
         tokio::time::sleep(Duration::from_secs(1)).await;
         rpc = LightClient::new(LightClientConfig::local()).await.unwrap();
     }
-    tokio::time::sleep(Duration::from_secs(10)).await;
+
+    // Wait for the indexer + prover to be ready (not just the validator RPC),
+    // otherwise proof requests race the still-initializing indexer/prover.
+    for attempt in 0..120 {
+        if matches!(rpc.get_indexer_health(None).await, Ok(true)) {
+            break;
+        }
+        assert!(attempt < 119, "indexer did not become healthy in time");
+        tokio::time::sleep(Duration::from_secs(1)).await;
+    }
 
     rpc
 }

basic-operations/anchor/update/programs/update/tests/test.rs

@@ -55,7 +55,16 @@ async fn start_validator_and_connect() -> LightClient {
         tokio::time::sleep(Duration::from_secs(1)).await;
         rpc = LightClient::new(LightClientConfig::local()).await.unwrap();
     }
-    tokio::time::sleep(Duration::from_secs(10)).await;
+
+    // Wait for the indexer + prover to be ready (not just the validator RPC),
+    // otherwise proof requests race the still-initializing indexer/prover.
+    for attempt in 0..120 {
+        if matches!(rpc.get_indexer_health(None).await, Ok(true)) {
+            break;
+        }
+        assert!(attempt < 119, "indexer did not become healthy in time");
+        tokio::time::sleep(Duration::from_secs(1)).await;
+    }
 
     rpc
 }

basic-operations/native/programs/burn/tests/test.rs

@@ -51,7 +51,16 @@ async fn start_validator_and_connect() -> LightClient {
         tokio::time::sleep(Duration::from_secs(1)).await;
         rpc = LightClient::new(LightClientConfig::local()).await.unwrap();
     }
-    tokio::time::sleep(Duration::from_secs(10)).await;
+
+    // Wait for the indexer + prover to be ready (not just the validator RPC),
+    // otherwise proof requests race the still-initializing indexer/prover.
+    for attempt in 0..120 {
+        if matches!(rpc.get_indexer_health(None).await, Ok(true)) {
+            break;
+        }
+        assert!(attempt < 119, "indexer did not become healthy in time");
+        tokio::time::sleep(Duration::from_secs(1)).await;
+    }
 
     rpc
 }

basic-operations/native/programs/close/tests/test.rs

@@ -51,7 +51,16 @@ async fn start_validator_and_connect() -> LightClient {
         tokio::time::sleep(Duration::from_secs(1)).await;
         rpc = LightClient::new(LightClientConfig::local()).await.unwrap();
     }
-    tokio::time::sleep(Duration::from_secs(10)).await;
+
+    // Wait for the indexer + prover to be ready (not just the validator RPC),
+    // otherwise proof requests race the still-initializing indexer/prover.
+    for attempt in 0..120 {
+        if matches!(rpc.get_indexer_health(None).await, Ok(true)) {
+            break;
+        }
+        assert!(attempt < 119, "indexer did not become healthy in time");
+        tokio::time::sleep(Duration::from_secs(1)).await;
+    }
 
     rpc
 }

basic-operations/native/programs/create/tests/test.rs

@@ -50,7 +50,16 @@ async fn start_validator_and_connect() -> LightClient {
         tokio::time::sleep(Duration::from_secs(1)).await;
         rpc = LightClient::new(LightClientConfig::local()).await.unwrap();
     }
-    tokio::time::sleep(Duration::from_secs(10)).await;
+
+    // Wait for the indexer + prover to be ready (not just the validator RPC),
+    // otherwise proof requests race the still-initializing indexer/prover.
+    for attempt in 0..120 {
+        if matches!(rpc.get_indexer_health(None).await, Ok(true)) {
+            break;
+        }
+        assert!(attempt < 119, "indexer did not become healthy in time");
+        tokio::time::sleep(Duration::from_secs(1)).await;
+    }
 
     rpc
 }

basic-operations/native/programs/reinit/tests/test.rs

@@ -52,7 +52,16 @@ async fn start_validator_and_connect() -> LightClient {
         tokio::time::sleep(Duration::from_secs(1)).await;
         rpc = LightClient::new(LightClientConfig::local()).await.unwrap();
     }
-    tokio::time::sleep(Duration::from_secs(10)).await;
+
+    // Wait for the indexer + prover to be ready (not just the validator RPC),
+    // otherwise proof requests race the still-initializing indexer/prover.
+    for attempt in 0..120 {
+        if matches!(rpc.get_indexer_health(None).await, Ok(true)) {
+            break;
+        }
+        assert!(attempt < 119, "indexer did not become healthy in time");
+        tokio::time::sleep(Duration::from_secs(1)).await;
+    }
 
     rpc
 }