Capture calls to augment_electrical_component_bounds

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

Author: simonvoelcker

Cargo.toml

@@ -23,11 +23,11 @@ tonic = "0.14"
 tonic-prost = "0.14"
 tracing = { version = "0.1" }
 tracing-subscriber = { version = "0.3" }
+tokio-stream = { version = "0.1.17", features = ["sync"] }
 
 [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]

src/client/test_utils.rs

@@ -3,24 +3,25 @@
 
 //! A mock implementation of the MicrogridApiClient for testing.
 
+use std::sync::Mutex;
 use std::{sync::Arc, time::SystemTime};
-
 use tokio_stream::wrappers::ReceiverStream;
 use tonic::Response;
 
+use super::MicrogridApiClient;
 use crate::{
     proto::{
         common::{
             metrics::{
-                Bounds, Metric, MetricSample, MetricValueVariant, SimpleMetricValue, metric_value_variant,
+                Bounds, Metric, MetricSample, MetricValueVariant, SimpleMetricValue,
+                metric_value_variant,
             },
             microgrid::electrical_components::{
                 ElectricalComponent, ElectricalComponentCategory,
                 ElectricalComponentCategorySpecificInfo, ElectricalComponentConnection,
                 ElectricalComponentStateCode, ElectricalComponentStateSnapshot,
-                ElectricalComponentTelemetry, Inverter, InverterType,
+                ElectricalComponentTelemetry, Inverter, InverterType, MetricConfigBounds,
                 electrical_component_category_specific_info::Kind,
-                MetricConfigBounds,
             },
         },
         google::protobuf,
@@ -34,7 +35,6 @@ use crate::{
     },
     quantity::{Current, Power, ReactivePower, Voltage},
 };
-use super::MicrogridApiClient;
 
 /// A mock implementation of the `MicrogridApiClient` trait for testing purposes.
 ///
@@ -43,6 +43,7 @@ use super::MicrogridApiClient;
 pub struct MockMicrogridApiClient {
     pub components: Vec<Arc<MockComponent>>,
     pub connections: Vec<ElectricalComponentConnection>,
+    pub augment_bounds_calls: Arc<Mutex<Vec<AugmentElectricalComponentBoundsRequest>>>,
 }
 
 #[derive(Default, Debug, Clone)]
@@ -162,13 +163,18 @@ impl MockComponent {
         self
     }
 
-    pub fn add_component_bounds(mut self, metric: i32, lower: Option<f32>, upper: Option<f32>) -> Self {
-        self.component.metric_config_bounds.push(
-            MetricConfigBounds {
+    pub fn add_component_bounds(
+        mut self,
+        metric: i32,
+        lower: Option<f32>,
+        upper: Option<f32>,
+    ) -> Self {
+        self.component
+            .metric_config_bounds
+            .push(MetricConfigBounds {
                 metric,
-                config_bounds: Some(Bounds {lower, upper}),
-            }
-        );
+                config_bounds: Some(Bounds { lower, upper }),
+            });
         self
     }
 
@@ -242,6 +248,7 @@ impl MockMicrogridApiClient {
         let mut this_client = Self {
             components: vec![],
             connections: vec![],
+            augment_bounds_calls: Arc::new(Mutex::new(Vec::new())),
         };
 
         let now = SystemTime::now();
@@ -283,6 +290,13 @@ impl MockMicrogridApiClient {
 
         this_client
     }
+
+    /// Return a handle to captured augment bounds requests.
+    pub fn augment_bounds_calls_handle(
+        &self,
+    ) -> Arc<Mutex<Vec<AugmentElectricalComponentBoundsRequest>>> {
+        self.augment_bounds_calls.clone()
+    }
 }
 
 #[async_trait::async_trait]
@@ -458,7 +472,13 @@ impl MicrogridApiClient for MockMicrogridApiClient {
         _request: impl tonic::IntoRequest<AugmentElectricalComponentBoundsRequest> + Send,
     ) -> std::result::Result<tonic::Response<AugmentElectricalComponentBoundsResponse>, tonic::Status>
     {
-        unimplemented!()
+        // Capture calls for tests
+        let req = _request.into_request().into_inner();
+        self.augment_bounds_calls.lock().unwrap().push(req);
+
+        Ok(Response::new(AugmentElectricalComponentBoundsResponse {
+            valid_until_time: None,
+        }))
     }
 }