Add timeout for waiting for reshare protocol

entropyxyz · Nov 28, 2024 · b7874c9 · b7874c9
1 parent 2e720b9
commit b7874c9
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/crates/threshold-signature-server/src/validator/tests.rs b/crates/threshold-signature-server/src/validator/tests.rs
@@ -139,6 +139,7 @@ async fn test_reshare() {
 
     // Now wait until signers have changed
     let old_signer_ids = HashSet::from_iter(signer_stash_accounts.into_iter().map(|id| id.0));
+    let mut i = 0;
     loop {
         let new_signer_ids: HashSet<[u8; 32]> = {
             let signer_query = entropy::storage().staking_extension().signers();
@@ -148,6 +149,10 @@ async fn test_reshare() {
         if new_signer_ids != old_signer_ids {
             break;
         }
+        if i > 100 {
+            panic!("Timed out waiting for reshare protocol to finish successfully");
+        }
+        i += 1;
         tokio::time::sleep(std::time::Duration::from_secs(1)).await;
     }