Skip to content

Commit fa0d3fb

Browse files
Fix warnings in power distributing actor tests
The issue was caused by missing calls to the cleanup() method of the mock microgrid, resulting in warnings during test execution. The warnings occurred because the actor background service attempted to cancel tasks when the event loop was already closed upon test completion. Signed-off-by: Daniel Zullo <[email protected]>
1 parent 81de64a commit fa0d3fb

File tree

1 file changed

+34
-0
lines changed

1 file changed

+34
-0
lines changed

tests/actor/power_distributing/test_power_distributing.py

Lines changed: 34 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -150,6 +150,8 @@ async def test_power_distributor_one_user(self, mocker: MockerFixture) -> None:
150150
assert result.excess_power.isclose(Power.from_watts(200.0))
151151
assert result.request == request
152152

153+
await mockgrid.cleanup()
154+
153155
async def test_power_distributor_exclusion_bounds(
154156
self, mocker: MockerFixture
155157
) -> None:
@@ -242,6 +244,8 @@ async def test_power_distributor_exclusion_bounds(
242244
assert result.bounds == PowerBounds(-1000, -600, 600, 1000)
243245
assert result.request == request
244246

247+
await mockgrid.cleanup()
248+
245249
# pylint: disable=too-many-locals
246250
async def test_two_batteries_one_inverters(self, mocker: MockerFixture) -> None:
247251
"""Test if power distribution works with two batteries connected to one inverter."""
@@ -312,6 +316,8 @@ async def test_two_batteries_one_inverters(self, mocker: MockerFixture) -> None:
312316
assert result.excess_power.isclose(Power.from_watts(700.0))
313317
assert result.request == request
314318

319+
await mockgrid.cleanup()
320+
315321
async def test_two_batteries_one_broken_one_inverters(
316322
self, mocker: MockerFixture
317323
) -> None:
@@ -393,6 +399,8 @@ async def test_two_batteries_one_broken_one_inverters(
393399
result.msg == "No data for at least one of the given batteries {9, 19}"
394400
)
395401

402+
await mockgrid.cleanup()
403+
396404
async def test_battery_two_inverters(self, mocker: MockerFixture) -> None:
397405
"""Test if power distribution works with two inverters for one battery."""
398406
gen = GraphGenerator()
@@ -464,6 +472,8 @@ async def test_battery_two_inverters(self, mocker: MockerFixture) -> None:
464472
assert result.excess_power.isclose(Power.from_watts(200.0))
465473
assert result.request == request
466474

475+
await mockgrid.cleanup()
476+
467477
async def test_two_batteries_three_inverters(self, mocker: MockerFixture) -> None:
468478
"""Test if power distribution works with two batteries connected to three inverters."""
469479
gen = GraphGenerator()
@@ -540,6 +550,8 @@ async def test_two_batteries_three_inverters(self, mocker: MockerFixture) -> Non
540550
assert result.excess_power.isclose(Power.from_watts(200.0))
541551
assert result.request == request
542552

553+
await mockgrid.cleanup()
554+
543555
async def test_two_batteries_one_inverter_different_exclusion_bounds_2(
544556
self, mocker: MockerFixture
545557
) -> None:
@@ -625,6 +637,8 @@ async def test_two_batteries_one_inverter_different_exclusion_bounds_2(
625637
assert result.request == request
626638
assert result.bounds == PowerBounds(-1000, -500, 500, 1000)
627639

640+
await mockgrid.cleanup()
641+
628642
async def test_two_batteries_one_inverter_different_exclusion_bounds(
629643
self, mocker: MockerFixture
630644
) -> None:
@@ -712,6 +726,8 @@ async def test_two_batteries_one_inverter_different_exclusion_bounds(
712726
# each inverter is bounded at 500
713727
assert result.bounds == PowerBounds(-500, -400, 400, 500)
714728

729+
await mockgrid.cleanup()
730+
715731
async def test_connected_but_not_requested_batteries(
716732
self, mocker: MockerFixture
717733
) -> None:
@@ -785,6 +801,8 @@ async def test_connected_but_not_requested_batteries(
785801
)
786802
assert err_msg is not None
787803

804+
await mockgrid.cleanup()
805+
788806
async def test_battery_soc_nan(self, mocker: MockerFixture) -> None:
789807
"""Test if battery with SoC==NaN is not used."""
790808
mockgrid = MockMicrogrid(grid_meter=False)
@@ -847,6 +865,8 @@ async def test_battery_soc_nan(self, mocker: MockerFixture) -> None:
847865
assert result.excess_power.isclose(Power.from_watts(700.0))
848866
assert result.request == request
849867

868+
await mockgrid.cleanup()
869+
850870
async def test_battery_capacity_nan(self, mocker: MockerFixture) -> None:
851871
"""Test battery with capacity set to NaN is not used."""
852872
mockgrid = MockMicrogrid(grid_meter=False)
@@ -902,6 +922,8 @@ async def test_battery_capacity_nan(self, mocker: MockerFixture) -> None:
902922
assert result.excess_power.isclose(Power.from_watts(700.0))
903923
assert result.request == request
904924

925+
await mockgrid.cleanup()
926+
905927
async def test_battery_power_bounds_nan(self, mocker: MockerFixture) -> None:
906928
"""Test battery with power bounds set to NaN is not used."""
907929
mockgrid = MockMicrogrid(grid_meter=False)
@@ -972,6 +994,8 @@ async def test_battery_power_bounds_nan(self, mocker: MockerFixture) -> None:
972994
assert result.excess_power.isclose(Power.from_watts(200.0))
973995
assert result.request == request
974996

997+
await mockgrid.cleanup()
998+
975999
async def test_power_distributor_invalid_battery_id(
9761000
self, mocker: MockerFixture
9771001
) -> None:
@@ -1017,6 +1041,8 @@ async def test_power_distributor_invalid_battery_id(
10171041
err_msg = re.search(r"No battery 100, available batteries:", result.msg)
10181042
assert err_msg is not None
10191043

1044+
await mockgrid.cleanup()
1045+
10201046
async def test_power_distributor_one_user_adjust_power_consume(
10211047
self, mocker: MockerFixture
10221048
) -> None:
@@ -1067,6 +1093,8 @@ async def test_power_distributor_one_user_adjust_power_consume(
10671093
assert result.request == request
10681094
assert result.bounds.inclusion_upper == 1000
10691095

1096+
await mockgrid.cleanup()
1097+
10701098
async def test_power_distributor_one_user_adjust_power_supply(
10711099
self, mocker: MockerFixture
10721100
) -> None:
@@ -1117,6 +1145,8 @@ async def test_power_distributor_one_user_adjust_power_supply(
11171145
assert result.request == request
11181146
assert result.bounds.inclusion_lower == -1000
11191147

1148+
await mockgrid.cleanup()
1149+
11201150
async def test_power_distributor_one_user_adjust_power_success(
11211151
self, mocker: MockerFixture
11221152
) -> None:
@@ -1167,6 +1197,8 @@ async def test_power_distributor_one_user_adjust_power_success(
11671197
assert result.excess_power.isclose(Power.zero(), abs_tol=1e-9)
11681198
assert result.request == request
11691199

1200+
await mockgrid.cleanup()
1201+
11701202
async def test_not_all_batteries_are_working(self, mocker: MockerFixture) -> None:
11711203
"""Test if power distribution works if not all batteries are working."""
11721204
mockgrid = MockMicrogrid(grid_meter=False)
@@ -1215,3 +1247,5 @@ async def test_not_all_batteries_are_working(self, mocker: MockerFixture) -> Non
12151247
assert result.excess_power.isclose(Power.from_watts(700.0))
12161248
assert result.succeeded_power.isclose(Power.from_watts(500.0))
12171249
assert result.request == request
1250+
1251+
await mockgrid.cleanup()

0 commit comments

Comments
 (0)