diff --git a/tests/auto/test_auto_mode.py b/tests/auto/test_auto_mode.py index a33feba..2cdd79b 100644 --- a/tests/auto/test_auto_mode.py +++ b/tests/auto/test_auto_mode.py @@ -54,8 +54,8 @@ async def change_mode(self) -> None: async def turn_on(self) -> None: await self._auto_mode.turn_on() - assert self._auto_mode.mode in ('scheduler', 'queue') - assert self._auto_mode.state.startswith('auto_') + await until_true(lambda: self._auto_mode.mode in ('scheduler', 'queue')) + await until_true(lambda: self._auto_mode.state.startswith('auto_')) async def turn_off(self) -> None: await self._auto_mode.turn_off()