hide mock() methods behind cfg(test) or feature

ZcashFoundation · Jan 31, 2024 · 0e1061d · 0e1061d
1 parent 3742711
commit 0e1061d
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 0 deletions.
diff --git a/zebra-scan/Cargo.toml b/zebra-scan/Cargo.toml
@@ -17,6 +17,7 @@ categories = ["cryptography::cryptocurrencies"]
 [[bin]] # Bin to run the Scanner gRPC server
 name = "scanner-grpc-server"
 path = "src/bin/rpc_server.rs"
+required-features = ["proptest-impl"]
 
 [features]
 

diff --git a/zebra-scan/src/init.rs b/zebra-scan/src/init.rs
@@ -51,6 +51,7 @@ pub struct ScanTask {
 
 impl ScanTask {
     /// Spawns a new [`ScanTask`] for tests.
+    #[cfg(any(test, feature = "proptest-impl"))]
     pub fn mock() -> (Self, Receiver<ScanTaskCommand>) {
         let (cmd_sender, cmd_receiver) = mpsc::channel();
 

diff --git a/zebra-scan/src/service.rs b/zebra-scan/src/service.rs
@@ -46,6 +46,7 @@ impl ScanService {
     }
 
     /// Create a new [`ScanService`] with a mock `ScanTask`
+    #[cfg(any(test, feature = "proptest-impl"))]
     pub fn new_with_mock_scanner(db: Storage) -> (Self, Receiver<ScanTaskCommand>) {
         let (scan_task, cmd_receiver) = ScanTask::mock();
         (Self { db, scan_task }, cmd_receiver)