Make more resilient

open-chat-labs · Dec 18, 2024 · b58f8e2 · b58f8e2
1 parent 01c48ce
commit b58f8e2
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/backend/integration_tests/src/client/registry.rs b/backend/integration_tests/src/client/registry.rs
@@ -29,13 +29,19 @@ pub mod happy_path {
 
         assert!(matches!(response, expand_onto_subnet::Response::Success));
 
-        for _ in 0..40 {
+        for i in 1..200 {
             env.advance_time(Duration::from_secs(1));
             env.tick();
-        }
 
-        let subnets::Response::Success(subnets) = super::subnets(env, sender, registry_canister_id, &Empty {});
+            if i % 10 == 0 {
+                let subnets::Response::Success(subnets) = super::subnets(env, sender, registry_canister_id, &Empty {});
+
+                if let Some(subnet) = subnets.into_iter().find(|s| s.subnet_id == subnet_id) {
+                    return subnet;
+                }
+            }
+        }
 
-        subnets.last().unwrap().clone()
+        panic!("Failed to expand onto new subnet")
     }
 }