add polling instead of hard sleep

input-output-hk · Oct 28, 2020 · fac26e2 · fac26e2
1 parent 33e3ea3
commit fac26e2
Showing 1 changed file with 23 additions and 4 deletions.
diff --git a/testing/jormungandr-integration-tests/src/jormungandr/recovery.rs b/testing/jormungandr-integration-tests/src/jormungandr/recovery.rs
@@ -97,15 +97,23 @@ pub fn test_node_recovers_from_node_restart() {
     let snapshot_before = take_snapshot(&account_receiver, &jormungandr, new_utxo.clone());
     jormungandr.stop();
 
-    std::thread::sleep(std::time::Duration::from_secs(10));
+    std::thread::sleep(std::time::Duration::from_secs(2));
 
     let jormungandr = Starter::new()
         .config(config)
         .role(Role::Leader)
         .start()
         .unwrap();
 
-    std::thread::sleep(std::time::Duration::from_secs(10));
+    jormungandr
+        .rest()
+        .raw()
+        .send_until_ok(
+            |raw| raw.account_state(&account_receiver),
+            Default::default(),
+        )
+        .expect("timeout occured when pooling address endpoint");
+
     let snapshot_after = take_snapshot(&account_receiver, &jormungandr, new_utxo);
 
     assert_eq!(
@@ -143,13 +151,24 @@ pub fn test_node_recovers_kill_signal() {
     );
     let snapshot_before = take_snapshot(&account_receiver, &jormungandr, new_utxo.clone());
     jormungandr.stop();
-    std::thread::sleep(std::time::Duration::from_secs(10));
+
+    std::thread::sleep(std::time::Duration::from_secs(2));
+
     let jormungandr = Starter::new()
         .config(config)
         .role(Role::Leader)
         .start()
         .unwrap();
-    std::thread::sleep(std::time::Duration::from_secs(10));
+
+    jormungandr
+        .rest()
+        .raw()
+        .send_until_ok(
+            |raw| raw.account_state(&account_receiver),
+            Default::default(),
+        )
+        .expect("timeout occured when pooling address endpoint");
+
     let snapshot_after = take_snapshot(&account_receiver, &jormungandr, new_utxo);
 
     assert_eq!(