Fix broken update test and fake service

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

Author: Marenz

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

@@ -123,12 +123,15 @@ async def UpdateMicrogridDispatch(
         request: DispatchUpdateRequest,
     ) -> Empty:
         """Update a dispatch."""
-        dispatch = next((d for d in self.dispatches if d.id == request.id), None)
+        index = next(
+            (i for i, d in enumerate(self.dispatches) if d.id == request.id),
+            None,
+        )
 
-        if dispatch is None:
+        if index is None:
             return Empty()
 
-        pb_dispatch = dispatch.to_protobuf()
+        pb_dispatch = self.dispatches[index].to_protobuf()
 
         # Go through the paths in the update mask and update the dispatch
         for path in request.update_mask.paths:
@@ -174,6 +177,8 @@ async def UpdateMicrogridDispatch(
         dispatch = Dispatch.from_protobuf(pb_dispatch)
         dispatch.update_time = datetime.now(tz=timezone.utc)
 
+        self.dispatches[index] = dispatch
+
         return Empty()
 
     async def GetMicrogridDispatch(

tests/test_dispatch_client.py

@@ -81,6 +81,7 @@ async def test_update_dispatch() -> None:
     assert dispatch == sample
 
     await client.update(dispatch.id, {"recurrence.interval": 4})
+    assert client.dispatches[0].recurrence.interval == 4
 
 
 async def test_get_dispatch() -> None: