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
@@ -89,6 +90,7 @@ def add_component(
         self,
         component_id: int,
         component_category: ComponentCategory.V,
+        max_current: Optional[int] = None,
         inverter_type: InverterType.V = InverterType.INVERTER_TYPE_UNSPECIFIED,
     ) -> None:
         """Add a component to the mock service."""
@@ -100,6 +102,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=max_current),
+                )
+            )
         else:
             self._components.append(
                 Component(id=component_id, category=component_category)

tests/microgrid/test_client.py

@@ -107,7 +107,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,10 +118,14 @@ async def test_components(self) -> None:
                 ]
             )
 
+            servicer.add_component(
+                101, components_pb.ComponentCategory.COMPONENT_CATEGORY_GRID, 123
+            )
+
             assert set(await microgrid.components()) == {
                 Component(100, ComponentCategory.NONE),
                 Component(
-                    101, ComponentCategory.GRID, None, GridMetadata(max_current=0.0)
+                    101, ComponentCategory.GRID, None, GridMetadata(max_current=123.0)
                 ),
                 Component(104, ComponentCategory.METER),
                 Component(105, ComponentCategory.INVERTER, InverterType.NONE),
@@ -140,6 +143,7 @@ async def test_components(self) -> None:
             servicer.add_component(
                 99,
                 components_pb.ComponentCategory.COMPONENT_CATEGORY_INVERTER,
+                None,
                 components_pb.InverterType.INVERTER_TYPE_BATTERY,
             )