reenable start_private_shares test now that embedding is configurable

Trust-Machines · Dec 2, 2024 · 9ada98b · 9ada98b
1 parent 8e411c1
commit 9ada98b
Showing 1 changed file with 20 additions and 20 deletions.
diff --git a/src/state_machine/coordinator/frost.rs b/src/state_machine/coordinator/frost.rs
@@ -857,30 +857,30 @@ pub mod test {
         assert_eq!(coordinator.get_state(), State::DkgPublicGather);
         assert_eq!(coordinator.current_dkg_id, 0);
     }
-    /*
-        #[test]
-        fn start_private_shares_v1() {
-            start_private_shares::<v1::Aggregator>();
-        }
 
-        #[test]
-        fn start_private_shares_v2() {
-            start_private_shares::<v2::Aggregator>();
-        }
+    #[test]
+    fn start_private_shares_v1() {
+        start_private_shares::<v1::Aggregator>();
+    }
+
+    #[test]
+    fn start_private_shares_v2() {
+        start_private_shares::<v2::Aggregator>();
+    }
 
-        fn start_private_shares<Aggregator: AggregatorTrait>() {
-            let mut rng = OsRng;
-            let config = Config::new(10, 40, 28, Scalar::random(&mut rng));
-            let mut coordinator = FrostCoordinator::<Aggregator>::new(config);
+    fn start_private_shares<Aggregator: AggregatorTrait>() {
+        let mut rng = OsRng;
+        let config = Config::new(10, 40, 28, Scalar::random(&mut rng));
+        let mut coordinator = FrostCoordinator::<Aggregator>::new(config);
 
-            coordinator.state = State::DkgPrivateDistribute; // Must be in this state before calling start private shares
+        coordinator.state = State::DkgPrivateDistribute; // Must be in this state before calling start private shares
+
+        let message = coordinator.start_private_shares().unwrap();
+        assert!(matches!(message.msg, Message::DkgPrivateBegin(_)));
+        assert_eq!(coordinator.get_state(), State::DkgPrivateGather);
+        assert_eq!(coordinator.current_dkg_id, 0);
+    }
 
-            let message = coordinator.start_private_shares().unwrap();
-            assert!(matches!(message.msg, Message::DkgPrivateBegin(_)));
-            assert_eq!(coordinator.get_state(), State::DkgPrivateGather);
-            assert_eq!(coordinator.current_dkg_id, 0);
-        }
-    */
     #[test]
     fn run_dkg_sign_v1() {
         run_dkg_sign::<FrostCoordinator<v1::Aggregator>, v1::Signer>(5, 2);