Use timeout to check if a receiver is empty

shsms · shsms · commit 4f2fe34201aa · 2024-03-12T14:03:38.000+01:00
Checking the length of a receiver is not supported by generic
receivers, but only by `Broadcast` receivers.  Using a timeout to
confirm a channel is empty is the more general approach.

Signed-off-by: Sahas Subramanian &lt;sahas.subramanian@proton.me&gt;
diff --git a/tests/actor/test_battery_status.py b/tests/actor/test_battery_status.py
@@ -713,7 +713,8 @@ async def test_async_battery_status(self, mocker: MockerFixture) -> None:
                 )
                 time.shift(10)
                 await asyncio.sleep(0.3)
-                assert len(status_receiver) == 0
+                with pytest.raises(asyncio.TimeoutError):
+                    await asyncio.wait_for(status_receiver.receive(), timeout=0.1)
 
                 await mock_microgrid.mock_client.send(
                     inverter_data(component_id=INVERTER_ID)

Original file line number	Diff line number	Diff line change
`@@ -713,7 +713,8 @@ async def test_async_battery_status(self, mocker: MockerFixture) -> None:`
`713`	`713`	`)`
`714`	`714`	`time.shift(10)`
`715`	`715`	`await asyncio.sleep(0.3)`
`716`		`- assert len(status_receiver) == 0`
	`716`	`+ with pytest.raises(asyncio.TimeoutError):`
	`717`	`+ await asyncio.wait_for(status_receiver.receive(), timeout=0.1)`
`717`	`718`
`718`	`719`	`await mock_microgrid.mock_client.send(`
`719`	`720`	`inverter_data(component_id=INVERTER_ID)`