More timeout fixes

andyleiserson · Dec 3, 2024 · dae5c6d · dae5c6d
1 parent 8155640
commit dae5c6d
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/ipa-core/src/protocol/context/dzkp_validator.rs b/ipa-core/src/protocol/context/dzkp_validator.rs
@@ -1376,7 +1376,10 @@ mod tests {
         let a: Vec<BA8> = repeat_with(|| rng.gen()).take(count).collect();
         let b: Vec<BA8> = repeat_with(|| rng.gen()).take(count).collect();
 
-        let [ab0, ab1, ab2]: [Vec<Replicated<BA8>>; 3] = TestWorld::default()
+        let config = TestWorldConfig::default().with_timeout_secs(60);
+        let world = TestWorld::<NotSharded>::with_config(&config);
+
+        let [ab0, ab1, ab2]: [Vec<Replicated<BA8>>; 3] = world
             .malicious(
                 zip(bit.clone(), zip(a.clone(), b.clone())),
                 |ctx, inputs| async move {

diff --git a/ipa-core/src/protocol/ipa_prf/boolean_ops/share_conversion_aby.rs b/ipa-core/src/protocol/ipa_prf/boolean_ops/share_conversion_aby.rs
@@ -386,8 +386,9 @@ mod tests {
         rand::thread_rng,
         secret_sharing::SharedValue,
         seq_join::{seq_join, SeqJoin},
+        sharding::NotSharded,
         test_executor::run,
-        test_fixture::{ReconstructArr, Runner, TestWorld},
+        test_fixture::{ReconstructArr, Runner, TestWorld, TestWorldConfig},
     };
 
     #[test]
@@ -457,7 +458,8 @@ mod tests {
             const COUNT: usize = CONV_CHUNK * PROOF_CHUNK * 2 + 1;
             const TOTAL_RECORDS: usize = COUNT.div_ceil(CONV_CHUNK);
 
-            let world = TestWorld::default();
+            let config = TestWorldConfig::default().with_timeout_secs(60);
+            let world = TestWorld::<NotSharded>::with_config(&config);
 
             let mut rng = thread_rng();