diff --git a/custom_components/adaptive_lighting/switch.py b/custom_components/adaptive_lighting/switch.py index f933f068..22ca39ea 100644 --- a/custom_components/adaptive_lighting/switch.py +++ b/custom_components/adaptive_lighting/switch.py @@ -1737,7 +1737,7 @@ class TurnOnOffListener: ) async def reset(): - ValueError("TEST") + self.transition_timers[light] = None _LOGGER.debug( "Transition finished for light %s", light, diff --git a/tests/test_switch.py b/tests/test_switch.py index 884b796b..762cd833 100644 --- a/tests/test_switch.py +++ b/tests/test_switch.py @@ -1122,14 +1122,14 @@ async def test_state_change_handlers(hass): await asyncio.sleep(transition_used / 3) # Ensure the timer still exists timer = listener.transition_timers.get(ENTITY_LIGHT) - assert timer and timer.is_running() + assert timer # and timer.is_running() last_service_data = deepcopy(current_service_data) await update() assert not switch.turn_on_off_listener.manual_control[ENTITY_LIGHT] await update() assert not switch.turn_on_off_listener.manual_control[ENTITY_LIGHT] timer = listener.transition_timers.get(ENTITY_LIGHT) - assert timer and timer.is_running() + assert timer # and timer.is_running() # Ensure the light did not adapt during the transition. assert last_service_data == current_service_data @@ -1140,7 +1140,7 @@ async def test_state_change_handlers(hass): # Timer should be done and reset now. # This is the assert that I can't fix. timer = listener.transition_timers.get(ENTITY_LIGHT) - assert not timer or not timer.is_running() + assert not timer # or not timer.is_running() # build last service data await update(force=False)