cleanup: pass rng

EspressoSystems · Dec 9, 2024 · e19186a · e19186a
1 parent a55cfd6
commit e19186a
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/contracts/rust/adapter/src/stake_table.rs b/contracts/rust/adapter/src/stake_table.rs
@@ -198,9 +198,8 @@ mod test {
     use super::*;
 
     impl NodeInfoJf {
-        fn random() -> Self {
+        fn random(rng: &mut impl RngCore) -> Self {
             let mut seed = [0u8; 32];
-            let mut rng = rand::thread_rng();
             rng.fill_bytes(&mut seed);
 
             let (stake_table_key, _) = BLSPubKey::generated_from_seed_indexed(seed, 0);
@@ -215,8 +214,9 @@ mod test {
 
     #[test]
     fn test_node_info_round_trip() {
+        let mut rng = rand::thread_rng();
         for _ in 0..20 {
-            let jf = NodeInfoJf::random();
+            let jf = NodeInfoJf::random(&mut rng);
             let sol: NodeInfo = jf.clone().into();
             let jf2: NodeInfoJf = sol.into();
             assert_eq!(jf2, jf);