Change type of TargetComponents to return ComponentId

* `frequenz.client.dispatch.TargetComponents` is now public
* its type has changed from `list[int] | list[ComponentCategory]`
to `list[ComponentId] | list[ComponentCategory]`.
* This change introduces a new dependency on
`frequenz-client-microgrid` (>= v0.7.0, < 0.8.0) for the
`frequenz.client.microgrid.ComponentId` type.

Signed-off-by: Elzbieta Kotulska <[email protected]>

Author: ela-kotulska-frequenz

RELEASE_NOTES.md

@@ -2,19 +2,11 @@
 
 ## Summary
 
-<!-- Here goes a general summary of what this release is about -->
-
-## Upgrading
-
-<!-- Here goes notes on how to upgrade from previous versions, including deprecations and what they should be replaced with -->
-
 ## New Features
 
 * `dispatch-cli` supports now the parameter `--type` and `--running` to filter the list of running services by type and status, respectively.
 * Every call now has a default timeout of 60 seconds, streams terminate after five minutes. This can be influenced by the two new parameters for`DispatchApiClient.__init__()`:
     * `default_timeout: timedelta` (default: 60 seconds)
     * `stream_timeout: timedelta` (default: 5 minutes)
 
-## Bug Fixes
-
-<!-- Here goes notable bug fixes that are worth a special mention or explanation -->
+* `frequenz.client.dispatch.TargetComponents` is now public, and its type has changed from `list[int] | list[ComponentCategory]` to `list[ComponentId] | list[ComponentCategory]`. This change introduces a new dependency on `frequenz-client-microgrid` (>= v0.7.0, < 0.8.0) for the `frequenz.client.microgrid.ComponentId` type.

pyproject.toml

@@ -40,6 +40,7 @@ dependencies = [
   "frequenz-api-dispatch == 1.0.0-rc1",
   "frequenz-client-base >= 0.8.0, < 0.10.0",
   "frequenz-client-common >= 0.1.0, < 0.4.0",
+  "frequenz-client-microgrid >= 0.7.0, < 0.8.0",
   "grpcio >= 1.66.1, < 2",
   "python-dateutil >= 2.8.2, < 3.0",
 ]

src/frequenz/client/dispatch/test/generator.py

@@ -7,6 +7,7 @@
 from datetime import datetime, timedelta, timezone
 
 from frequenz.client.common.microgrid.components import ComponentCategory
+from frequenz.client.microgrid import ComponentId
 
 from .._internal_types import rounded_start_time
 from ..recurrence import EndCriteria, Frequency, RecurrenceRule, Weekday
@@ -99,7 +100,7 @@ def generate_dispatch(self) -> Dispatch:
                         for _ in range(self._rng.randint(1, 10))
                     ],
                     [
-                        self._rng.randint(1, 100)
+                        ComponentId(self._rng.randint(1, 100))
                         for _ in range(self._rng.randint(1, 10))
                     ],
                 ]

src/frequenz/client/dispatch/types.py

@@ -24,10 +24,11 @@
 
 # pylint: enable=no-name-in-module
 from frequenz.client.common.microgrid.components import ComponentCategory
+from frequenz.client.microgrid import ComponentId
 
 from .recurrence import Frequency, RecurrenceRule, Weekday
 
-TargetComponents = list[int] | list[ComponentCategory]
+TargetComponents = list[ComponentId] | list[ComponentCategory]
 """One or more target components specifying which components a dispatch targets.
 
 It can be a list of component IDs or a list of categories.
@@ -50,7 +51,9 @@ def _target_components_from_protobuf(
     """
     match pb_target.WhichOneof("components"):
         case "component_ids":
-            id_list: list[int] = list(pb_target.component_ids.ids)
+            id_list: list[ComponentId] = list(
+                map(ComponentId, pb_target.component_ids.ids)
+            )
             return id_list
         case "component_categories":
             category_list: list[ComponentCategory] = list(
@@ -80,8 +83,10 @@ def _target_components_to_protobuf(
     """
     pb_target = PBTargetComponents()
     match target:
-        case list(component_ids) if all(isinstance(id, int) for id in component_ids):
-            pb_target.component_ids.ids.extend(cast(list[int], component_ids))
+        case list(component_ids) if all(
+            isinstance(id, ComponentId) for id in component_ids
+        ):
+            pb_target.component_ids.ids.extend(map(lambda x: int(x), component_ids))
         case list(categories) if all(
             isinstance(cat, ComponentCategory) for cat in categories
         ):

tests/test_cli.py

@@ -13,6 +13,7 @@
 from tzlocal import get_localzone
 
 from frequenz.client.common.microgrid.components import ComponentCategory
+from frequenz.client.microgrid import ComponentId
 from frequenz.client.dispatch.__main__ import cli
 from frequenz.client.dispatch.recurrence import (
     EndCriteria,
@@ -68,7 +69,7 @@ def mock_client(fake_client: FakeClient) -> Generator[None, None, None]:
                         type="test",
                         start_time=datetime(2023, 1, 1, 0, 0, 0),
                         duration=timedelta(seconds=3600),
-                        target=[1, 2, 3],
+                        target=list(map(ComponentId, [1, 2, 3])),
                         active=True,
                         dry_run=False,
                         payload={},
@@ -91,7 +92,7 @@ def mock_client(fake_client: FakeClient) -> Generator[None, None, None]:
                         type="test",
                         start_time=datetime(2023, 1, 1, 0, 0, 0),
                         duration=timedelta(seconds=3600),
-                        target=[1, 2, 3],
+                        target=list(map(ComponentId, [1, 2, 3])),
                         active=True,
                         dry_run=False,
                         payload={},
@@ -113,7 +114,7 @@ def mock_client(fake_client: FakeClient) -> Generator[None, None, None]:
                         type="test",
                         start_time=datetime(2023, 1, 1, 0, 0, 0),
                         duration=timedelta(seconds=3600),
-                        target=[1, 2, 3],
+                        target=list(map(ComponentId, [1, 2, 3])),
                         active=True,
                         dry_run=False,
                         payload={},
@@ -128,7 +129,7 @@ def mock_client(fake_client: FakeClient) -> Generator[None, None, None]:
                         type="test",
                         start_time=datetime(2023, 1, 1, 0, 0, 0),
                         duration=timedelta(seconds=3600),
-                        target=[1, 2, 3],
+                        target=list(map(ComponentId, [1, 2, 3])),
                         active=True,
                         dry_run=False,
                         payload={},
@@ -156,7 +157,7 @@ def mock_client(fake_client: FakeClient) -> Generator[None, None, None]:
                         type="test",
                         start_time=datetime(2023, 1, 1, 0, 0, 0),
                         duration=timedelta(seconds=3600),
-                        target=[1, 2, 3],
+                        target=list(map(ComponentId, [1, 2, 3])),
                         active=True,
                         dry_run=False,
                         payload={},
@@ -169,7 +170,7 @@ def mock_client(fake_client: FakeClient) -> Generator[None, None, None]:
                         type="filtered",
                         start_time=datetime(2023, 1, 1, 0, 0, 0),
                         duration=timedelta(seconds=1800),
-                        target=[3],
+                        target=list(map(ComponentId, [3])),
                         active=True,
                         dry_run=False,
                         payload={},
@@ -376,7 +377,7 @@ async def test_create_command(
     expected_type: str,
     expected_start_time_delta: timedelta | Literal["NOW"],
     expected_duration: timedelta,
-    expected_target: list[int] | list[ComponentCategory],
+    expected_target: list[ComponentId] | list[ComponentCategory],
     expected_options: dict[str, Any],
     expected_reccurence: RecurrenceRule | None,
     expected_return_code: int,
@@ -528,7 +529,7 @@ async def test_create_command(
                     type="test",
                     start_time=datetime(2023, 1, 1, 0, 0, 0),
                     duration=timedelta(seconds=3600),
-                    target=[500, 501],
+                    target=list(map(ComponentId, [500, 501])),
                     active=True,
                     dry_run=False,
                     payload={},
@@ -633,7 +634,7 @@ async def test_update_command(
                     type="test",
                     start_time=datetime(2023, 1, 1, 0, 0, 0),
                     duration=timedelta(seconds=3600),
-                    target=[1, 2, 3],
+                    target=list(map(ComponentId, [1, 2, 3])),
                     active=True,
                     dry_run=False,
                     payload={},
@@ -680,7 +681,7 @@ async def test_get_command(
                     type="test",
                     start_time=datetime(2023, 1, 1, 0, 0, 0),
                     duration=timedelta(seconds=3600),
-                    target=[1, 2, 3],
+                    target=list(map(ComponentId, [1, 2, 3])),
                     active=True,
                     dry_run=False,
                     payload={},

tests/test_proto.py

@@ -18,13 +18,14 @@
     _target_components_from_protobuf,
     _target_components_to_protobuf,
 )
+from frequenz.client.microgrid import ComponentId
 
 
 def test_target_components() -> None:
     """Test the target components."""
     for components in (
-        [1, 2, 3],
-        [10, 20, 30],
+        list(map(ComponentId, [1, 2, 3])),
+        list(map(ComponentId, [10, 20, 30])),
         [ComponentCategory.BATTERY],
         [ComponentCategory.GRID],
         [ComponentCategory.METER],
@@ -99,7 +100,7 @@ def test_dispatch() -> None:
             start_time=datetime(2024, 10, 10, tzinfo=timezone.utc),
             end_time=datetime(2024, 10, 20, tzinfo=timezone.utc),
             duration=timedelta(days=10),
-            target=[1, 2, 3],
+            target=list(map(ComponentId, [1, 2, 3])),
             active=True,
             dry_run=False,
             payload={"key": "value"},
@@ -163,7 +164,7 @@ def test_dispatch_create_request_with_no_recurrence() -> None:
         type="test",
         start_time=datetime(2024, 10, 10, tzinfo=timezone.utc),
         duration=timedelta(days=10),
-        target=[1, 2, 3],
+        target=list(map(ComponentId, [1, 2, 3])),
         active=True,
         dry_run=False,
         payload={"key": "value"},
@@ -180,7 +181,7 @@ def test_dispatch_create_start_immediately() -> None:
         type="test",
         start_time="NOW",
         duration=timedelta(days=10),
-        target=[1, 2, 3],
+        target=list(map(ComponentId, [1, 2, 3])),
         active=True,
         dry_run=False,
         payload={"key": "value"},