fix: state trigger working correctly on consequetive executions

bramstroker · May 3, 2024 · ec631ac · ec631ac
1 parent 8daf8c8
commit ec631ac
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/custom_components/powercalc/strategy/playbook.py b/custom_components/powercalc/strategy/playbook.py
@@ -92,6 +92,7 @@ async def activate_playbook(self, playbook_id: str) -> None:
 
         _LOGGER.debug("Activating playbook %s", playbook_id)
         playbook = await self._load_playbook(playbook_id=playbook_id)
+        playbook.queue.reset()
         self._active_playbook = playbook
         self._start_time = dt.utcnow()
 

diff --git a/tests/strategy/test_playbook.py b/tests/strategy/test_playbook.py
@@ -355,6 +355,16 @@ async def test_state_trigger(hass: HomeAssistant) -> None:
 
     await elapse_and_assert_power(hass, 1, "0.10")
 
+    hass.states.async_set("media_player.sonos", STATE_IDLE)
+    await hass.async_block_till_done()
+
+    await elapse_and_assert_power(hass, 2, "5.00")
+
+    hass.states.async_set("media_player.sonos", STATE_OFF)
+    await hass.async_block_till_done()
+
+    await elapse_and_assert_power(hass, 1, "0.10")
+
 
 async def elapse_and_assert_power(
     hass: HomeAssistant,