Add a few dispatch types and conversion tests for them

Signed-off-by: Mathias L. Baumann <[email protected]>

Author: Marenz

src/frequenz/client/dispatch/_client.py

@@ -2,7 +2,6 @@
 # Copyright © 2024 Frequenz Energy-as-a-Service GmbH
 
 """Dispatch API client for Python."""
-from dataclasses import dataclass
 from datetime import datetime
 from typing import Awaitable
 
@@ -11,38 +10,23 @@
 
 # pylint: disable=no-name-in-module
 from frequenz.api.dispatch.v1.dispatch_pb2 import (
-    ComponentSelector,
     DispatchFilter,
     DispatchList,
     DispatchListRequest,
 )
 from frequenz.api.dispatch.v1.dispatch_pb2 import (
     TimeIntervalFilter as PBTimeIntervalFilter,
 )
-from frequenz.client.common.microgrid.components.components import (
-    ComponentCategory,
-    _component_category_to_protobuf,
-)
 from google.protobuf.timestamp_pb2 import Timestamp
 
-# pylint: enable=no-name-in-module
-
-
-@dataclass(frozen=True, kw_only=True)
-class TimeIntervalFilter:
-    """Filter for a time interval."""
-
-    start_from: datetime
-    """Filter by start_time >= start_from."""
-
-    start_to: datetime
-    """Filter by start_time < start_to."""
-
-    end_from: datetime
-    """Filter by end_time >= end_from."""
+from ._types import (
+    ComponentSelector,
+    Dispatch,
+    TimeIntervalFilter,
+    _component_selector_to_protobuf,
+)
 
-    end_to: datetime
-    """Filter by end_time < end_to."""
+# pylint: enable=no-name-in-module
 
 
 class Client:
@@ -62,11 +46,11 @@ def __init__(self, grpc_channel: grpc.aio.Channel, svc_addr: str) -> None:
     async def list(
         self,
         microgrid_id: int,
-        component_selectors: list[list[int] | ComponentCategory] | None = None,
+        component_selectors: list[ComponentSelector] | None = None,
         interval_filter: TimeIntervalFilter | None = None,
         is_active: bool | None = None,
         is_dry_run: bool | None = None,
-    ) -> DispatchList:
+    ) -> list[Dispatch]:
         """List dispatches.
 
         Args:
@@ -96,16 +80,8 @@ def to_timestamp(dt: datetime) -> Timestamp:
 
         selectors = []
 
-        if component_selectors is not None:
-            for selector in component_selectors:
-                proto_selector = ComponentSelector()
-                if isinstance(selector, list):
-                    proto_selector.component_ids.component_ids.extend(selector)
-                elif isinstance(selector, ComponentCategory):
-                    proto_selector.component_category = _component_category_to_protobuf(
-                        selector
-                    )
-                selectors.append(proto_selector)
+        for selector in component_selectors if component_selectors else []:
+            selectors.append(_component_selector_to_protobuf(selector))
 
         filters = DispatchFilter(
             selectors=selectors,
@@ -117,4 +93,4 @@ def to_timestamp(dt: datetime) -> Timestamp:
 
         response: Awaitable[DispatchList]
         response = self._stub.ListMicrogridDispatches(request)  # type: ignore
-        return await response
+        return list(map(Dispatch.from_protobuf, (await response).dispatches))