Skip to content
Merged
Show file tree
Hide file tree
Changes from 2 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 6 additions & 0 deletions src/frequenz/sdk/microgrid/_data_pipeline.py
Original file line number Diff line number Diff line change
Expand Up @@ -495,8 +495,14 @@ async def _stop(self) -> None:
if self._resampling_actor:
await self._resampling_actor.actor.stop()
await self._battery_power_wrapper.stop()
await self._ev_power_wrapper.stop()
await self._pv_power_wrapper.stop()
for pool in self._battery_pool_reference_stores.values():
await pool.stop()
for evpool in self._ev_charger_pool_reference_stores.values():
await evpool.stop()
for pvpool in self._pv_pool_reference_stores.values():
await pvpool.stop()


_DATA_PIPELINE: _DataPipeline | None = None
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -63,6 +63,14 @@ async def mocks(mocker: MockerFixture) -> typing.AsyncIterator[_Mocks]:
dp._ev_power_wrapper.status_channel.new_sender(),
)

await asyncio.gather(
*[
dp._stop(),
streamer.stop(),
mockgrid.cleanup(),
]
)


class TestEVChargerPoolControl:
"""Test the EV charger pool control methods."""
Expand Down
8 changes: 8 additions & 0 deletions tests/timeseries/_pv_pool/test_pv_pool_control_methods.py
Original file line number Diff line number Diff line change
Expand Up @@ -56,6 +56,14 @@ async def mocks(mocker: MockerFixture) -> typing.AsyncIterator[_Mocks]:
dp._pv_power_wrapper.status_channel.new_sender(),
)

await asyncio.gather(
*[
dp._stop(),
streamer.stop(),
mockgrid.cleanup(),
]
)


class TestPVPoolControl:
"""Test control methods for the PVPool."""
Expand Down
Loading