Fix import conditional compilation

ZcashFoundation · Dec 8, 2023 · 8a13d78 · 8a13d78
1 parent 6b67c6c
commit 8a13d78
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/zebra-state/src/service/finalized_state.rs b/zebra-state/src/service/finalized_state.rs
@@ -43,7 +43,7 @@ pub use disk_db::{DiskDb, DiskWriteBatch, ReadDisk, WriteDisk};
 #[allow(unused_imports)]
 pub use disk_format::{
     FromDisk, IntoDisk, OutputIndex, OutputLocation, RawBytes, TransactionIndex,
-    TransactionLocation, KV, MAX_ON_DISK_HEIGHT,
+    TransactionLocation, MAX_ON_DISK_HEIGHT,
 };
 pub use zebra_db::ZebraDb;
 
@@ -53,6 +53,9 @@ pub use disk_format::{
     SaplingScanningKey,
 };
 
+#[cfg(any(test, feature = "proptest-impl"))]
+pub use disk_format::KV;
+
 /// The column families supported by the running `zebra-state` database code.
 ///
 /// Existing column families that aren't listed here are preserved when the database is opened.

diff --git a/zebra-state/src/service/finalized_state/disk_format.rs b/zebra-state/src/service/finalized_state/disk_format.rs
@@ -22,15 +22,15 @@ mod tests;
 pub use block::{TransactionIndex, TransactionLocation, MAX_ON_DISK_HEIGHT};
 pub use transparent::{OutputIndex, OutputLocation};
 
-#[allow(unused_imports)]
-pub use tests::KV;
-
 #[cfg(feature = "shielded-scan")]
 pub use scan::{
     SaplingScannedDatabaseEntry, SaplingScannedDatabaseIndex, SaplingScannedResult,
     SaplingScanningKey,
 };
 
+#[cfg(any(test, feature = "proptest-impl"))]
+pub use tests::KV;
+
 /// Helper type for writing types to disk as raw bytes.
 /// Also used to convert key types to raw bytes for disk lookups.
 pub trait IntoDisk {