Expose test utils behind the test-utils feature gate

Signed-off-by: Simon Völcker <simon.voelcker@frequenz.com>

Author: simonvoelcker

Cargo.toml

@@ -23,12 +23,15 @@ tonic = "0.14"
 tonic-prost = "0.14"
 tracing = { version = "0.1" }
 tracing-subscriber = { version = "0.3" }
+tokio-stream = { version = "0.1.17", features = ["sync"], optional = true }
 
 [dev-dependencies]
 tracing-subscriber = { version = "0.3", features = ["std", "env-filter"] }
 tokio = { version = "1.48", features = ["test-util"] }
 tokio-stream = { version = "0.1.17", features = ["sync"] }
 
-
 [build-dependencies]
 tonic-prost-build = { version = "0.14", features = ["cleanup-markdown"] }
+
+[features]
+test-utils = ["dep:tokio-stream"]

src/client.rs

@@ -18,5 +18,6 @@ pub use proto::common::microgrid::electrical_components::{
     ElectricalComponent, ElectricalComponentCategory,
 };
 
-#[cfg(test)]
-pub(crate) mod test_utils;
+#[cfg(any(test, feature = "test-utils"))]
+#[expect(clippy::unwrap_used, clippy::panic, clippy::expect_used)]
+pub mod test_utils;

src/client/proto.rs

@@ -17,7 +17,7 @@ mod pb {
 // Only export what we need
 pub use pb::frequenz::api::common::v1alpha8 as common;
 pub use pb::frequenz::api::microgrid::v1alpha18 as microgrid;
-#[cfg(test)]
+#[cfg(any(test, feature = "test-utils"))]
 pub use pb::google;
 
 mod electrical_component;

src/lib.rs

@@ -38,3 +38,6 @@ pub(crate) mod wall_clock_timer;
 
 mod microgrid;
 pub use microgrid::{BatteryPool, Microgrid};
+
+#[cfg(any(test, feature = "test-utils"))]
+pub use client::test_utils;