From 5488c1a91f18f9307ca24e374512b267a84863ab Mon Sep 17 00:00:00 2001
From: Moshe Shababo <17073733+moshababo@users.noreply.github.com>
Date: Tue, 28 Nov 2023 11:00:38 -0600
Subject: [PATCH] #[cfg(test)] to tests re-exports

---
 node/actors/bft/src/leader/mod.rs  |  9 +++++----
 node/actors/bft/src/replica/mod.rs | 11 ++++++-----
 2 files changed, 11 insertions(+), 9 deletions(-)

diff --git a/node/actors/bft/src/leader/mod.rs b/node/actors/bft/src/leader/mod.rs
index 1ed65346..dfeff178 100644
--- a/node/actors/bft/src/leader/mod.rs
+++ b/node/actors/bft/src/leader/mod.rs
@@ -5,11 +5,12 @@
 mod replica_commit;
 mod replica_prepare;
 mod state_machine;
+
+pub(crate) use state_machine::StateMachine;
+
 #[cfg(test)]
 mod tests;
-
-#[allow(unused_imports)]
+#[cfg(test)]
 pub(crate) use replica_commit::Error as ReplicaCommitError;
-#[allow(unused_imports)]
+#[cfg(test)]
 pub(crate) use replica_prepare::Error as ReplicaPrepareError;
-pub(crate) use state_machine::StateMachine;
diff --git a/node/actors/bft/src/replica/mod.rs b/node/actors/bft/src/replica/mod.rs
index 822a4b48..40df624f 100644
--- a/node/actors/bft/src/replica/mod.rs
+++ b/node/actors/bft/src/replica/mod.rs
@@ -7,12 +7,13 @@ mod leader_commit;
 mod leader_prepare;
 mod new_view;
 mod state_machine;
-#[cfg(test)]
-mod tests;
 mod timer;
 
-#[allow(unused_imports)]
+pub(crate) use state_machine::StateMachine;
+
+#[cfg(test)]
+mod tests;
+#[cfg(test)]
 pub(crate) use leader_commit::Error as LeaderCommitError;
-#[allow(unused_imports)]
+#[cfg(test)]
 pub(crate) use leader_prepare::Error as LeaderPrepareError;
-pub(crate) use state_machine::StateMachine;