Pass through timeout arguments for the client

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

Author: Marenz

pyproject.toml

@@ -40,7 +40,7 @@ dependencies = [
   # plugins.mkdocstrings.handlers.python.import)
   "frequenz-sdk >= 1.0.0-rc1900, < 1.0.0-rc2100",
   "frequenz-channels >= 1.6.1, < 2.0.0",
-  "frequenz-client-dispatch >= 0.10.1, < 0.11.0",
+  "frequenz-client-dispatch >= 0.10.2, < 0.11.0",
 ]
 dynamic = ["version"]
 

src/frequenz/dispatch/_dispatcher.py

@@ -198,23 +198,33 @@ async def run():
         ```
     """
 
+    # pylint: disable-next=too-many-arguments
     def __init__(
         self,
         *,
         microgrid_id: int,
         server_url: str,
         key: str,
+        call_timeout: timedelta = timedelta(seconds=60),
+        stream_timeout: timedelta = timedelta(minutes=5),
     ):
         """Initialize the dispatcher.
 
         Args:
             microgrid_id: The microgrid id.
             server_url: The URL of the dispatch service.
             key: The key to access the service.
+            call_timeout: The timeout for API calls.
+            stream_timeout: The timeout for streaming API calls.
         """
         super().__init__()
 
-        self._client = DispatchApiClient(server_url=server_url, key=key)
+        self._client = DispatchApiClient(
+            server_url=server_url,
+            key=key,
+            call_timeout=call_timeout,
+            stream_timeout=stream_timeout,
+        )
         self._bg_service = DispatchScheduler(
             microgrid_id,
             self._client,

tests/test_managing_actor.py

@@ -327,9 +327,18 @@ async def test_manage_abstraction(
     class MyFakeClient(FakeClient):
         """Fake client for testing."""
 
-        def __init__(self, *, server_url: str, key: str):
+        def __init__(
+            self,
+            *,
+            server_url: str,
+            key: str,
+            call_timeout: timedelta,
+            stream_timeout: timedelta,
+        ) -> None:
             assert server_url
             assert key
+            assert call_timeout
+            assert stream_timeout
             super().__init__()
 
     mid = 1