node-data: move faker methods to relevant modules

dusk-network · Jul 3, 2024 · bef750e · bef750e
1 parent 114a6f1
commit bef750e
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 15 deletions.
diff --git a/node-data/src/ledger.rs b/node-data/src/ledger.rs
@@ -154,21 +154,6 @@ pub mod faker {
     use super::*;
     use crate::bls::PublicKeyBytes;
     use rand::Rng;
-    use transaction::faker::gen_dummy_tx;
-
-    impl<T> Dummy<T> for Block {
-        /// Creates a block with 3 transactions and random header.
-        fn dummy_with_rng<R: Rng + ?Sized>(_config: &T, rng: &mut R) -> Self {
-            let txs = vec![
-                gen_dummy_tx(rng.gen()),
-                gen_dummy_tx(rng.gen()),
-                gen_dummy_tx(rng.gen()),
-            ];
-            let header: Header = Faker.fake();
-
-            Block::new(header, txs).expect("valid hash")
-        }
-    }
 
     impl<T> Dummy<T> for PublicKeyBytes {
         fn dummy_with_rng<R: Rng + ?Sized>(_config: &T, rng: &mut R) -> Self {

diff --git a/node-data/src/ledger/block.rs b/node-data/src/ledger/block.rs
@@ -84,3 +84,24 @@ impl BlockWithLabel {
         matches!(self.label(), Label::Final(_)) || self.blk.header().height == 0
     }
 }
+
+#[cfg(any(feature = "faker", test))]
+pub mod faker {
+    use super::*;
+    use rand::Rng;
+    use transaction::faker::gen_dummy_tx;
+
+    impl<T> Dummy<T> for Block {
+        /// Creates a block with 3 transactions and random header.
+        fn dummy_with_rng<R: Rng + ?Sized>(_config: &T, rng: &mut R) -> Self {
+            let txs = vec![
+                gen_dummy_tx(rng.gen()),
+                gen_dummy_tx(rng.gen()),
+                gen_dummy_tx(rng.gen()),
+            ];
+            let header: Header = Faker.fake();
+
+            Block::new(header, txs).expect("valid hash")
+        }
+    }
+}