Use tighter types for SetPowerResult

Signed-off-by: Sahas Subramanian <[email protected]>

Author: shsms

src/frequenz/sdk/actor/power_distributing/_component_status.py

@@ -5,7 +5,6 @@
 
 
 import enum
-import typing
 from abc import ABC, abstractmethod
 from collections import abc
 from dataclasses import dataclass
@@ -70,10 +69,10 @@ class ComponentStatus:
 class SetPowerResult:
     """Lists of components for which the last set power command succeeded or failed."""
 
-    succeeded: typing.Iterable[int]
+    succeeded: abc.Set[int]
     """Component IDs for which the last set power command succeeded."""
 
-    failed: typing.Iterable[int]
+    failed: abc.Set[int]
     """Component IDs for which the last set power command failed."""
 
 

tests/actor/test_battery_status.py

@@ -538,7 +538,7 @@ async def test_sync_blocking_interrupted_with_invalid_message(
         assert tracker._get_new_status_if_changed() is None
 
         tracker._handle_status_set_power_result(
-            SetPowerResult(succeeded={BATTERY_ID}, failed={})
+            SetPowerResult(succeeded={BATTERY_ID}, failed=set())
         )
         assert tracker._get_new_status_if_changed() is None
 
@@ -648,7 +648,7 @@ async def test_async_battery_status(self, mocker: MockerFixture) -> None:
             assert status.value is ComponentStatusEnum.WORKING
 
             await set_power_result_sender.send(
-                SetPowerResult(succeeded={}, failed={BATTERY_ID})
+                SetPowerResult(succeeded=set(), failed={BATTERY_ID})
             )
             status = await asyncio.wait_for(status_receiver.receive(), timeout=0.1)
             assert status.value is ComponentStatusEnum.UNCERTAIN
@@ -669,7 +669,7 @@ async def test_async_battery_status(self, mocker: MockerFixture) -> None:
             assert status.value is ComponentStatusEnum.NOT_WORKING
 
             await set_power_result_sender.send(
-                SetPowerResult(succeeded={}, failed={BATTERY_ID})
+                SetPowerResult(succeeded=set(), failed={BATTERY_ID})
             )
             await asyncio.sleep(0.3)
             assert len(status_receiver) == 0