From 3e82cc1b4fad9428fd376c318d37c37a97f1db0f Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jaroslav=20Hansl=C3=ADk?= Date: Fri, 16 Oct 2020 23:01:06 +0200 Subject: [PATCH] Add arming state to template alarm control panel (#41964) --- homeassistant/components/template/alarm_control_panel.py | 2 ++ tests/components/template/test_alarm_control_panel.py | 7 +++++++ 2 files changed, 9 insertions(+) diff --git a/homeassistant/components/template/alarm_control_panel.py b/homeassistant/components/template/alarm_control_panel.py index e37c7e2982e..ccefab767be 100644 --- a/homeassistant/components/template/alarm_control_panel.py +++ b/homeassistant/components/template/alarm_control_panel.py @@ -22,6 +22,7 @@ from homeassistant.const import ( STATE_ALARM_ARMED_AWAY, STATE_ALARM_ARMED_HOME, STATE_ALARM_ARMED_NIGHT, + STATE_ALARM_ARMING, STATE_ALARM_DISARMED, STATE_ALARM_PENDING, STATE_ALARM_TRIGGERED, @@ -42,6 +43,7 @@ _VALID_STATES = [ STATE_ALARM_ARMED_AWAY, STATE_ALARM_ARMED_HOME, STATE_ALARM_ARMED_NIGHT, + STATE_ALARM_ARMING, STATE_ALARM_DISARMED, STATE_ALARM_PENDING, STATE_ALARM_TRIGGERED, diff --git a/tests/components/template/test_alarm_control_panel.py b/tests/components/template/test_alarm_control_panel.py index d8d4b8fdcf2..7860622e658 100644 --- a/tests/components/template/test_alarm_control_panel.py +++ b/tests/components/template/test_alarm_control_panel.py @@ -6,6 +6,7 @@ from homeassistant.const import ( STATE_ALARM_ARMED_AWAY, STATE_ALARM_ARMED_HOME, STATE_ALARM_ARMED_NIGHT, + STATE_ALARM_ARMING, STATE_ALARM_DISARMED, STATE_ALARM_PENDING, STATE_ALARM_TRIGGERED, @@ -76,6 +77,12 @@ async def test_template_state_text(hass): state = hass.states.get("alarm_control_panel.test_template_panel") assert state.state == STATE_ALARM_ARMED_NIGHT + hass.states.async_set("alarm_control_panel.test", STATE_ALARM_ARMING) + await hass.async_block_till_done() + + state = hass.states.get("alarm_control_panel.test_template_panel") + assert state.state == STATE_ALARM_ARMING + hass.states.async_set("alarm_control_panel.test", STATE_ALARM_DISARMED) await hass.async_block_till_done()