Allow setting grid metadata in mocks

This commit allows tests to specify the `rated_fuse_current` member in
the metadata objects of grid connection points. The test data supplied
using this mechanism can be used in downstream validations.

Signed-off-by: Tiyash Basu <[email protected]>

Author: tiyash-basu-frequenz

tests/microgrid/mock_api.py

@@ -30,6 +30,7 @@
 from frequenz.api.microgrid.battery_pb2 import Battery
 from frequenz.api.microgrid.battery_pb2 import Data as BatteryData
 from frequenz.api.microgrid.ev_charger_pb2 import EvCharger
+from frequenz.api.microgrid.grid_pb2 import Metadata as GridMetadata
 from frequenz.api.microgrid.inverter_pb2 import Inverter
 from frequenz.api.microgrid.inverter_pb2 import Metadata as InverterMetadata
 from frequenz.api.microgrid.meter_pb2 import Data as MeterData
@@ -57,6 +58,8 @@
 from google.protobuf.timestamp_pb2 import Timestamp
 from google.protobuf.wrappers_pb2 import BoolValue
 
+from frequenz.sdk.timeseries import Current
+
 
 class MockMicrogridServicer(  # pylint: disable=too-many-public-methods
     MicrogridServicer
@@ -89,6 +92,7 @@ def add_component(
         self,
         component_id: int,
         component_category: ComponentCategory.V,
+        max_current: Optional[Current] = None,
         inverter_type: InverterType.V = InverterType.INVERTER_TYPE_UNSPECIFIED,
     ) -> None:
         """Add a component to the mock service."""
@@ -100,6 +104,17 @@ def add_component(
                     inverter=InverterMetadata(type=inverter_type),
                 )
             )
+        elif (
+            component_category == ComponentCategory.COMPONENT_CATEGORY_GRID
+            and max_current is not None
+        ):
+            self._components.append(
+                Component(
+                    id=component_id,
+                    category=component_category,
+                    grid=GridMetadata(rated_fuse_current=int(max_current.as_amperes())),
+                )
+            )
         else:
             self._components.append(
                 Component(id=component_id, category=component_category)

tests/microgrid/test_client.py

@@ -26,6 +26,7 @@
     InverterType,
     MeterData,
 )
+from frequenz.sdk.timeseries import Current
 
 from . import mock_api
 
@@ -107,7 +108,6 @@ async def test_components(self) -> None:
                         100,
                         components_pb.ComponentCategory.COMPONENT_CATEGORY_UNSPECIFIED,
                     ),
-                    (101, components_pb.ComponentCategory.COMPONENT_CATEGORY_GRID),
                     (104, components_pb.ComponentCategory.COMPONENT_CATEGORY_METER),
                     (105, components_pb.ComponentCategory.COMPONENT_CATEGORY_INVERTER),
                     (106, components_pb.ComponentCategory.COMPONENT_CATEGORY_BATTERY),
@@ -119,13 +119,19 @@ async def test_components(self) -> None:
                 ]
             )
 
+            servicer.add_component(
+                101,
+                components_pb.ComponentCategory.COMPONENT_CATEGORY_GRID,
+                Current.from_amperes(123.0),
+            )
+
             assert set(await microgrid.components()) == {
                 Component(100, ComponentCategory.NONE),
                 Component(
                     101,
                     ComponentCategory.GRID,
                     None,
-                    GridMetadata(max_current=0.0),
+                    GridMetadata(max_current=123.0),
                 ),
                 Component(104, ComponentCategory.METER),
                 Component(105, ComponentCategory.INVERTER, InverterType.NONE),
@@ -143,6 +149,7 @@ async def test_components(self) -> None:
             servicer.add_component(
                 99,
                 components_pb.ComponentCategory.COMPONENT_CATEGORY_INVERTER,
+                None,
                 components_pb.InverterType.INVERTER_TYPE_BATTERY,
             )