"""The tests for the manual Alarm Control Panel component.""" from datetime import timedelta from unittest.mock import patch, MagicMock from homeassistant.components.demo import alarm_control_panel as demo from homeassistant.setup import async_setup_component from homeassistant.const import ( STATE_ALARM_DISARMED, STATE_ALARM_ARMED_HOME, STATE_ALARM_ARMED_AWAY, STATE_ALARM_ARMED_NIGHT, STATE_ALARM_ARMED_CUSTOM_BYPASS, STATE_ALARM_PENDING, STATE_ALARM_TRIGGERED) from homeassistant.components import alarm_control_panel import homeassistant.util.dt as dt_util from tests.common import (async_fire_time_changed, mock_component, mock_restore_cache) from tests.components.alarm_control_panel import common from homeassistant.core import State, CoreState CODE = 'HELLO_CODE' async def test_setup_demo_platform(hass): """Test setup.""" mock = MagicMock() add_entities = mock.MagicMock() await demo.async_setup_platform(hass, {}, add_entities) assert add_entities.call_count == 1 async def test_arm_home_no_pending(hass): """Test arm home method.""" assert await async_setup_component( hass, alarm_control_panel.DOMAIN, {'alarm_control_panel': { 'platform': 'manual', 'name': 'test', 'code': CODE, 'pending_time': 0, 'disarm_after_trigger': False }}) entity_id = 'alarm_control_panel.test' assert STATE_ALARM_DISARMED == \ hass.states.get(entity_id).state await common.async_alarm_arm_home(hass, CODE) assert STATE_ALARM_ARMED_HOME == \ hass.states.get(entity_id).state async def test_arm_home_no_pending_when_code_not_req(hass): """Test arm home method.""" assert await async_setup_component( hass, alarm_control_panel.DOMAIN, {'alarm_control_panel': { 'platform': 'manual', 'name': 'test', 'code': CODE, 'code_arm_required': False, 'pending_time': 0, 'disarm_after_trigger': False }}) entity_id = 'alarm_control_panel.test' assert STATE_ALARM_DISARMED == \ hass.states.get(entity_id).state common.async_alarm_arm_home(hass, 0) await hass.async_block_till_done() assert STATE_ALARM_ARMED_HOME == \ hass.states.get(entity_id).state async def test_arm_home_with_pending(hass): """Test arm home method.""" assert await async_setup_component( hass, alarm_control_panel.DOMAIN, {'alarm_control_panel': { 'platform': 'manual', 'name': 'test', 'code': CODE, 'pending_time': 1, 'disarm_after_trigger': False }}) entity_id = 'alarm_control_panel.test' assert STATE_ALARM_DISARMED == \ hass.states.get(entity_id).state await common.async_alarm_arm_home(hass, CODE, entity_id) assert STATE_ALARM_PENDING == \ hass.states.get(entity_id).state state = hass.states.get(entity_id) assert state.attributes['post_pending_state'] == STATE_ALARM_ARMED_HOME future = dt_util.utcnow() + timedelta(seconds=1) with patch(('homeassistant.components.manual.alarm_control_panel.' 'dt_util.utcnow'), return_value=future): async_fire_time_changed(hass, future) await hass.async_block_till_done() state = hass.states.get(entity_id) assert state.state == STATE_ALARM_ARMED_HOME async def test_arm_home_with_invalid_code(hass): """Attempt to arm home without a valid code.""" assert await async_setup_component( hass, alarm_control_panel.DOMAIN, {'alarm_control_panel': { 'platform': 'manual', 'name': 'test', 'code': CODE, 'pending_time': 1, 'disarm_after_trigger': False }}) entity_id = 'alarm_control_panel.test' assert STATE_ALARM_DISARMED == \ hass.states.get(entity_id).state await common.async_alarm_arm_home(hass, CODE + '2') assert STATE_ALARM_DISARMED == \ hass.states.get(entity_id).state async def test_arm_away_no_pending(hass): """Test arm home method.""" assert await async_setup_component( hass, alarm_control_panel.DOMAIN, {'alarm_control_panel': { 'platform': 'manual', 'name': 'test', 'code': CODE, 'pending_time': 0, 'disarm_after_trigger': False }}) entity_id = 'alarm_control_panel.test' assert STATE_ALARM_DISARMED == \ hass.states.get(entity_id).state await common.async_alarm_arm_away(hass, CODE, entity_id) assert STATE_ALARM_ARMED_AWAY == \ hass.states.get(entity_id).state async def test_arm_away_no_pending_when_code_not_req(hass): """Test arm home method.""" assert await async_setup_component( hass, alarm_control_panel.DOMAIN, {'alarm_control_panel': { 'platform': 'manual', 'name': 'test', 'code': CODE, 'code_arm_required': False, 'pending_time': 0, 'disarm_after_trigger': False }}) entity_id = 'alarm_control_panel.test' assert STATE_ALARM_DISARMED == \ hass.states.get(entity_id).state common.async_alarm_arm_away(hass, 0, entity_id) await hass.async_block_till_done() assert STATE_ALARM_ARMED_AWAY == \ hass.states.get(entity_id).state async def test_arm_home_with_template_code(hass): """Attempt to arm with a template-based code.""" assert await async_setup_component( hass, alarm_control_panel.DOMAIN, {'alarm_control_panel': { 'platform': 'manual', 'name': 'test', 'code_template': '{{ "abc" }}', 'pending_time': 0, 'disarm_after_trigger': False }}) entity_id = 'alarm_control_panel.test' assert STATE_ALARM_DISARMED == \ hass.states.get(entity_id).state await common.async_alarm_arm_home(hass, 'abc') state = hass.states.get(entity_id) assert STATE_ALARM_ARMED_HOME == state.state async def test_arm_away_with_pending(hass): """Test arm home method.""" assert await async_setup_component( hass, alarm_control_panel.DOMAIN, {'alarm_control_panel': { 'platform': 'manual', 'name': 'test', 'code': CODE, 'pending_time': 1, 'disarm_after_trigger': False }}) entity_id = 'alarm_control_panel.test' assert STATE_ALARM_DISARMED == \ hass.states.get(entity_id).state await common.async_alarm_arm_away(hass, CODE) assert STATE_ALARM_PENDING == \ hass.states.get(entity_id).state state = hass.states.get(entity_id) assert state.attributes['post_pending_state'] == STATE_ALARM_ARMED_AWAY future = dt_util.utcnow() + timedelta(seconds=1) with patch(('homeassistant.components.manual.alarm_control_panel.' 'dt_util.utcnow'), return_value=future): async_fire_time_changed(hass, future) await hass.async_block_till_done() state = hass.states.get(entity_id) assert state.state == STATE_ALARM_ARMED_AWAY async def test_arm_away_with_invalid_code(hass): """Attempt to arm away without a valid code.""" assert await async_setup_component( hass, alarm_control_panel.DOMAIN, {'alarm_control_panel': { 'platform': 'manual', 'name': 'test', 'code': CODE, 'pending_time': 1, 'disarm_after_trigger': False }}) entity_id = 'alarm_control_panel.test' assert STATE_ALARM_DISARMED == \ hass.states.get(entity_id).state await common.async_alarm_arm_away(hass, CODE + '2') assert STATE_ALARM_DISARMED == \ hass.states.get(entity_id).state async def test_arm_night_no_pending(hass): """Test arm night method.""" assert await async_setup_component( hass, alarm_control_panel.DOMAIN, {'alarm_control_panel': { 'platform': 'manual', 'name': 'test', 'code': CODE, 'pending_time': 0, 'disarm_after_trigger': False }}) entity_id = 'alarm_control_panel.test' assert STATE_ALARM_DISARMED == \ hass.states.get(entity_id).state await common.async_alarm_arm_night(hass, CODE) assert STATE_ALARM_ARMED_NIGHT == \ hass.states.get(entity_id).state async def test_arm_night_no_pending_when_code_not_req(hass): """Test arm night method.""" assert await async_setup_component( hass, alarm_control_panel.DOMAIN, {'alarm_control_panel': { 'platform': 'manual', 'name': 'test', 'code': CODE, 'code_arm_required': False, 'pending_time': 0, 'disarm_after_trigger': False }}) entity_id = 'alarm_control_panel.test' assert STATE_ALARM_DISARMED == \ hass.states.get(entity_id).state common.async_alarm_arm_night(hass, 0) await hass.async_block_till_done() assert STATE_ALARM_ARMED_NIGHT == \ hass.states.get(entity_id).state async def test_arm_night_with_pending(hass): """Test arm night method.""" assert await async_setup_component( hass, alarm_control_panel.DOMAIN, {'alarm_control_panel': { 'platform': 'manual', 'name': 'test', 'code': CODE, 'pending_time': 1, 'disarm_after_trigger': False }}) entity_id = 'alarm_control_panel.test' assert STATE_ALARM_DISARMED == \ hass.states.get(entity_id).state await common.async_alarm_arm_night(hass, CODE, entity_id) assert STATE_ALARM_PENDING == \ hass.states.get(entity_id).state state = hass.states.get(entity_id) assert state.attributes['post_pending_state'] == \ STATE_ALARM_ARMED_NIGHT future = dt_util.utcnow() + timedelta(seconds=1) with patch(('homeassistant.components.manual.alarm_control_panel.' 'dt_util.utcnow'), return_value=future): async_fire_time_changed(hass, future) await hass.async_block_till_done() state = hass.states.get(entity_id) assert state.state == STATE_ALARM_ARMED_NIGHT # Do not go to the pending state when updating to the same state await common.async_alarm_arm_night(hass, CODE, entity_id) assert STATE_ALARM_ARMED_NIGHT == \ hass.states.get(entity_id).state async def test_arm_night_with_invalid_code(hass): """Attempt to night home without a valid code.""" assert await async_setup_component( hass, alarm_control_panel.DOMAIN, {'alarm_control_panel': { 'platform': 'manual', 'name': 'test', 'code': CODE, 'pending_time': 1, 'disarm_after_trigger': False }}) entity_id = 'alarm_control_panel.test' assert STATE_ALARM_DISARMED == \ hass.states.get(entity_id).state await common.async_alarm_arm_night(hass, CODE + '2') assert STATE_ALARM_DISARMED == \ hass.states.get(entity_id).state async def test_trigger_no_pending(hass): """Test triggering when no pending submitted method.""" assert await async_setup_component( hass, alarm_control_panel.DOMAIN, {'alarm_control_panel': { 'platform': 'manual', 'name': 'test', 'trigger_time': 1, 'disarm_after_trigger': False }}) entity_id = 'alarm_control_panel.test' assert STATE_ALARM_DISARMED == \ hass.states.get(entity_id).state await common.async_alarm_trigger(hass, entity_id=entity_id) assert STATE_ALARM_PENDING == \ hass.states.get(entity_id).state future = dt_util.utcnow() + timedelta(seconds=60) with patch(('homeassistant.components.manual.alarm_control_panel.' 'dt_util.utcnow'), return_value=future): async_fire_time_changed(hass, future) await hass.async_block_till_done() assert STATE_ALARM_TRIGGERED == \ hass.states.get(entity_id).state async def test_trigger_with_delay(hass): """Test trigger method and switch from pending to triggered.""" assert await async_setup_component( hass, alarm_control_panel.DOMAIN, {'alarm_control_panel': { 'platform': 'manual', 'name': 'test', 'code': CODE, 'delay_time': 1, 'pending_time': 0, 'disarm_after_trigger': False }}) entity_id = 'alarm_control_panel.test' assert STATE_ALARM_DISARMED == \ hass.states.get(entity_id).state await common.async_alarm_arm_away(hass, CODE) assert STATE_ALARM_ARMED_AWAY == \ hass.states.get(entity_id).state await common.async_alarm_trigger(hass, entity_id=entity_id) state = hass.states.get(entity_id) assert STATE_ALARM_PENDING == state.state assert STATE_ALARM_TRIGGERED == \ state.attributes['post_pending_state'] future = dt_util.utcnow() + timedelta(seconds=1) with patch(('homeassistant.components.manual.alarm_control_panel.' 'dt_util.utcnow'), return_value=future): async_fire_time_changed(hass, future) await hass.async_block_till_done() state = hass.states.get(entity_id) assert STATE_ALARM_TRIGGERED == state.state async def test_trigger_zero_trigger_time(hass): """Test disabled trigger.""" assert await async_setup_component( hass, alarm_control_panel.DOMAIN, {'alarm_control_panel': { 'platform': 'manual', 'name': 'test', 'pending_time': 0, 'trigger_time': 0, 'disarm_after_trigger': False }}) entity_id = 'alarm_control_panel.test' assert STATE_ALARM_DISARMED == \ hass.states.get(entity_id).state await common.async_alarm_trigger(hass) assert STATE_ALARM_DISARMED == \ hass.states.get(entity_id).state async def test_trigger_zero_trigger_time_with_pending(hass): """Test disabled trigger.""" assert await async_setup_component( hass, alarm_control_panel.DOMAIN, {'alarm_control_panel': { 'platform': 'manual', 'name': 'test', 'pending_time': 2, 'trigger_time': 0, 'disarm_after_trigger': False }}) entity_id = 'alarm_control_panel.test' assert STATE_ALARM_DISARMED == \ hass.states.get(entity_id).state await common.async_alarm_trigger(hass) assert STATE_ALARM_DISARMED == \ hass.states.get(entity_id).state async def test_trigger_with_pending(hass): """Test arm home method.""" assert await async_setup_component( hass, alarm_control_panel.DOMAIN, {'alarm_control_panel': { 'platform': 'manual', 'name': 'test', 'pending_time': 2, 'trigger_time': 3, 'disarm_after_trigger': False }}) entity_id = 'alarm_control_panel.test' assert STATE_ALARM_DISARMED == \ hass.states.get(entity_id).state await common.async_alarm_trigger(hass) assert STATE_ALARM_PENDING == \ hass.states.get(entity_id).state state = hass.states.get(entity_id) assert state.attributes['post_pending_state'] == STATE_ALARM_TRIGGERED future = dt_util.utcnow() + timedelta(seconds=2) with patch(('homeassistant.components.manual.alarm_control_panel.' 'dt_util.utcnow'), return_value=future): async_fire_time_changed(hass, future) await hass.async_block_till_done() state = hass.states.get(entity_id) assert state.state == STATE_ALARM_TRIGGERED future = dt_util.utcnow() + timedelta(seconds=5) with patch(('homeassistant.components.manual.alarm_control_panel.' 'dt_util.utcnow'), return_value=future): async_fire_time_changed(hass, future) await hass.async_block_till_done() state = hass.states.get(entity_id) assert state.state == STATE_ALARM_DISARMED async def test_trigger_with_unused_specific_delay(hass): """Test trigger method and switch from pending to triggered.""" assert await async_setup_component( hass, alarm_control_panel.DOMAIN, {'alarm_control_panel': { 'platform': 'manual', 'name': 'test', 'code': CODE, 'delay_time': 5, 'pending_time': 0, 'armed_home': { 'delay_time': 10 }, 'disarm_after_trigger': False }}) entity_id = 'alarm_control_panel.test' assert STATE_ALARM_DISARMED == \ hass.states.get(entity_id).state await common.async_alarm_arm_away(hass, CODE) assert STATE_ALARM_ARMED_AWAY == \ hass.states.get(entity_id).state await common.async_alarm_trigger(hass, entity_id=entity_id) state = hass.states.get(entity_id) assert STATE_ALARM_PENDING == state.state assert STATE_ALARM_TRIGGERED == \ state.attributes['post_pending_state'] future = dt_util.utcnow() + timedelta(seconds=5) with patch(('homeassistant.components.manual.alarm_control_panel.' 'dt_util.utcnow'), return_value=future): async_fire_time_changed(hass, future) await hass.async_block_till_done() state = hass.states.get(entity_id) assert state.state == STATE_ALARM_TRIGGERED async def test_trigger_with_specific_delay(hass): """Test trigger method and switch from pending to triggered.""" assert await async_setup_component( hass, alarm_control_panel.DOMAIN, {'alarm_control_panel': { 'platform': 'manual', 'name': 'test', 'code': CODE, 'delay_time': 10, 'pending_time': 0, 'armed_away': { 'delay_time': 1 }, 'disarm_after_trigger': False }}) entity_id = 'alarm_control_panel.test' assert STATE_ALARM_DISARMED == \ hass.states.get(entity_id).state await common.async_alarm_arm_away(hass, CODE) assert STATE_ALARM_ARMED_AWAY == \ hass.states.get(entity_id).state await common.async_alarm_trigger(hass, entity_id=entity_id) state = hass.states.get(entity_id) assert STATE_ALARM_PENDING == state.state assert STATE_ALARM_TRIGGERED == \ state.attributes['post_pending_state'] future = dt_util.utcnow() + timedelta(seconds=1) with patch(('homeassistant.components.manual.alarm_control_panel.' 'dt_util.utcnow'), return_value=future): async_fire_time_changed(hass, future) await hass.async_block_till_done() state = hass.states.get(entity_id) assert state.state == STATE_ALARM_TRIGGERED async def test_trigger_with_pending_and_delay(hass): """Test trigger method and switch from pending to triggered.""" assert await async_setup_component( hass, alarm_control_panel.DOMAIN, {'alarm_control_panel': { 'platform': 'manual', 'name': 'test', 'code': CODE, 'delay_time': 1, 'pending_time': 0, 'triggered': { 'pending_time': 1 }, 'disarm_after_trigger': False }}) entity_id = 'alarm_control_panel.test' assert STATE_ALARM_DISARMED == \ hass.states.get(entity_id).state await common.async_alarm_arm_away(hass, CODE) assert STATE_ALARM_ARMED_AWAY == \ hass.states.get(entity_id).state await common.async_alarm_trigger(hass, entity_id=entity_id) state = hass.states.get(entity_id) assert state.state == STATE_ALARM_PENDING assert state.attributes['post_pending_state'] == STATE_ALARM_TRIGGERED future = dt_util.utcnow() + timedelta(seconds=1) with patch(('homeassistant.components.manual.alarm_control_panel.' 'dt_util.utcnow'), return_value=future): async_fire_time_changed(hass, future) await hass.async_block_till_done() state = hass.states.get(entity_id) assert state.state == STATE_ALARM_PENDING assert state.attributes['post_pending_state'] == STATE_ALARM_TRIGGERED future += timedelta(seconds=1) with patch(('homeassistant.components.manual.alarm_control_panel.' 'dt_util.utcnow'), return_value=future): async_fire_time_changed(hass, future) await hass.async_block_till_done() state = hass.states.get(entity_id) assert state.state == STATE_ALARM_TRIGGERED async def test_trigger_with_pending_and_specific_delay(hass): """Test trigger method and switch from pending to triggered.""" assert await async_setup_component( hass, alarm_control_panel.DOMAIN, {'alarm_control_panel': { 'platform': 'manual', 'name': 'test', 'code': CODE, 'delay_time': 10, 'pending_time': 0, 'armed_away': { 'delay_time': 1 }, 'triggered': { 'pending_time': 1 }, 'disarm_after_trigger': False }}) entity_id = 'alarm_control_panel.test' assert STATE_ALARM_DISARMED == \ hass.states.get(entity_id).state await common.async_alarm_arm_away(hass, CODE) assert STATE_ALARM_ARMED_AWAY == \ hass.states.get(entity_id).state await common.async_alarm_trigger(hass, entity_id=entity_id) state = hass.states.get(entity_id) assert state.state == STATE_ALARM_PENDING assert state.attributes['post_pending_state'] == STATE_ALARM_TRIGGERED future = dt_util.utcnow() + timedelta(seconds=1) with patch(('homeassistant.components.manual.alarm_control_panel.' 'dt_util.utcnow'), return_value=future): async_fire_time_changed(hass, future) await hass.async_block_till_done() state = hass.states.get(entity_id) assert state.state == STATE_ALARM_PENDING assert state.attributes['post_pending_state'] == STATE_ALARM_TRIGGERED future += timedelta(seconds=1) with patch(('homeassistant.components.manual.alarm_control_panel.' 'dt_util.utcnow'), return_value=future): async_fire_time_changed(hass, future) await hass.async_block_till_done() state = hass.states.get(entity_id) assert state.state == STATE_ALARM_TRIGGERED async def test_armed_home_with_specific_pending(hass): """Test arm home method.""" assert await async_setup_component( hass, alarm_control_panel.DOMAIN, {'alarm_control_panel': { 'platform': 'manual', 'name': 'test', 'pending_time': 10, 'armed_home': { 'pending_time': 2 } }}) entity_id = 'alarm_control_panel.test' await common.async_alarm_arm_home(hass) assert STATE_ALARM_PENDING == \ hass.states.get(entity_id).state future = dt_util.utcnow() + timedelta(seconds=2) with patch(('homeassistant.components.manual.alarm_control_panel.' 'dt_util.utcnow'), return_value=future): async_fire_time_changed(hass, future) await hass.async_block_till_done() assert STATE_ALARM_ARMED_HOME == \ hass.states.get(entity_id).state async def test_armed_away_with_specific_pending(hass): """Test arm home method.""" assert await async_setup_component( hass, alarm_control_panel.DOMAIN, {'alarm_control_panel': { 'platform': 'manual', 'name': 'test', 'pending_time': 10, 'armed_away': { 'pending_time': 2 } }}) entity_id = 'alarm_control_panel.test' await common.async_alarm_arm_away(hass) assert STATE_ALARM_PENDING == \ hass.states.get(entity_id).state future = dt_util.utcnow() + timedelta(seconds=2) with patch(('homeassistant.components.manual.alarm_control_panel.' 'dt_util.utcnow'), return_value=future): async_fire_time_changed(hass, future) await hass.async_block_till_done() assert STATE_ALARM_ARMED_AWAY == \ hass.states.get(entity_id).state async def test_armed_night_with_specific_pending(hass): """Test arm home method.""" assert await async_setup_component( hass, alarm_control_panel.DOMAIN, {'alarm_control_panel': { 'platform': 'manual', 'name': 'test', 'pending_time': 10, 'armed_night': { 'pending_time': 2 } }}) entity_id = 'alarm_control_panel.test' await common.async_alarm_arm_night(hass) assert STATE_ALARM_PENDING == \ hass.states.get(entity_id).state future = dt_util.utcnow() + timedelta(seconds=2) with patch(('homeassistant.components.manual.alarm_control_panel.' 'dt_util.utcnow'), return_value=future): async_fire_time_changed(hass, future) await hass.async_block_till_done() assert STATE_ALARM_ARMED_NIGHT == \ hass.states.get(entity_id).state async def test_trigger_with_specific_pending(hass): """Test arm home method.""" assert await async_setup_component( hass, alarm_control_panel.DOMAIN, {'alarm_control_panel': { 'platform': 'manual', 'name': 'test', 'pending_time': 10, 'triggered': { 'pending_time': 2 }, 'trigger_time': 3, 'disarm_after_trigger': False }}) entity_id = 'alarm_control_panel.test' await common.async_alarm_trigger(hass) assert STATE_ALARM_PENDING == \ hass.states.get(entity_id).state future = dt_util.utcnow() + timedelta(seconds=2) with patch(('homeassistant.components.manual.alarm_control_panel.' 'dt_util.utcnow'), return_value=future): async_fire_time_changed(hass, future) await hass.async_block_till_done() assert STATE_ALARM_TRIGGERED == \ hass.states.get(entity_id).state future = dt_util.utcnow() + timedelta(seconds=5) with patch(('homeassistant.components.manual.alarm_control_panel.' 'dt_util.utcnow'), return_value=future): async_fire_time_changed(hass, future) await hass.async_block_till_done() assert STATE_ALARM_DISARMED == \ hass.states.get(entity_id).state async def test_trigger_with_disarm_after_trigger(hass): """Test disarm after trigger.""" assert await async_setup_component( hass, alarm_control_panel.DOMAIN, {'alarm_control_panel': { 'platform': 'manual', 'name': 'test', 'trigger_time': 5, 'pending_time': 0, 'disarm_after_trigger': True }}) entity_id = 'alarm_control_panel.test' assert STATE_ALARM_DISARMED == \ hass.states.get(entity_id).state await common.async_alarm_trigger(hass, entity_id=entity_id) assert STATE_ALARM_TRIGGERED == \ hass.states.get(entity_id).state future = dt_util.utcnow() + timedelta(seconds=5) with patch(('homeassistant.components.manual.alarm_control_panel.' 'dt_util.utcnow'), return_value=future): async_fire_time_changed(hass, future) await hass.async_block_till_done() assert STATE_ALARM_DISARMED == \ hass.states.get(entity_id).state async def test_trigger_with_zero_specific_trigger_time(hass): """Test trigger method.""" assert await async_setup_component( hass, alarm_control_panel.DOMAIN, {'alarm_control_panel': { 'platform': 'manual', 'name': 'test', 'trigger_time': 5, 'disarmed': { 'trigger_time': 0 }, 'pending_time': 0, 'disarm_after_trigger': True }}) entity_id = 'alarm_control_panel.test' assert STATE_ALARM_DISARMED == \ hass.states.get(entity_id).state await common.async_alarm_trigger(hass, entity_id=entity_id) assert STATE_ALARM_DISARMED == \ hass.states.get(entity_id).state async def test_trigger_with_unused_zero_specific_trigger_time(hass): """Test disarm after trigger.""" assert await async_setup_component( hass, alarm_control_panel.DOMAIN, {'alarm_control_panel': { 'platform': 'manual', 'name': 'test', 'trigger_time': 5, 'armed_home': { 'trigger_time': 0 }, 'pending_time': 0, 'disarm_after_trigger': True }}) entity_id = 'alarm_control_panel.test' assert STATE_ALARM_DISARMED == \ hass.states.get(entity_id).state await common.async_alarm_trigger(hass, entity_id=entity_id) assert STATE_ALARM_TRIGGERED == \ hass.states.get(entity_id).state future = dt_util.utcnow() + timedelta(seconds=5) with patch(('homeassistant.components.manual.alarm_control_panel.' 'dt_util.utcnow'), return_value=future): async_fire_time_changed(hass, future) await hass.async_block_till_done() assert STATE_ALARM_DISARMED == \ hass.states.get(entity_id).state async def test_trigger_with_specific_trigger_time(hass): """Test disarm after trigger.""" assert await async_setup_component( hass, alarm_control_panel.DOMAIN, {'alarm_control_panel': { 'platform': 'manual', 'name': 'test', 'disarmed': { 'trigger_time': 5 }, 'pending_time': 0, 'disarm_after_trigger': True }}) entity_id = 'alarm_control_panel.test' assert STATE_ALARM_DISARMED == \ hass.states.get(entity_id).state await common.async_alarm_trigger(hass, entity_id=entity_id) assert STATE_ALARM_TRIGGERED == \ hass.states.get(entity_id).state future = dt_util.utcnow() + timedelta(seconds=5) with patch(('homeassistant.components.manual.alarm_control_panel.' 'dt_util.utcnow'), return_value=future): async_fire_time_changed(hass, future) await hass.async_block_till_done() assert STATE_ALARM_DISARMED == \ hass.states.get(entity_id).state async def test_trigger_with_no_disarm_after_trigger(hass): """Test disarm after trigger.""" assert await async_setup_component( hass, alarm_control_panel.DOMAIN, {'alarm_control_panel': { 'platform': 'manual', 'name': 'test', 'trigger_time': 5, 'pending_time': 0, 'disarm_after_trigger': False }}) entity_id = 'alarm_control_panel.test' assert STATE_ALARM_DISARMED == \ hass.states.get(entity_id).state await common.async_alarm_arm_away(hass, CODE, entity_id) assert STATE_ALARM_ARMED_AWAY == \ hass.states.get(entity_id).state await common.async_alarm_trigger(hass, entity_id=entity_id) assert STATE_ALARM_TRIGGERED == \ hass.states.get(entity_id).state future = dt_util.utcnow() + timedelta(seconds=5) with patch(('homeassistant.components.manual.alarm_control_panel.' 'dt_util.utcnow'), return_value=future): async_fire_time_changed(hass, future) await hass.async_block_till_done() assert STATE_ALARM_ARMED_AWAY == \ hass.states.get(entity_id).state async def test_back_to_back_trigger_with_no_disarm_after_trigger(hass): """Test disarm after trigger.""" assert await async_setup_component( hass, alarm_control_panel.DOMAIN, {'alarm_control_panel': { 'platform': 'manual', 'name': 'test', 'trigger_time': 5, 'pending_time': 0, 'disarm_after_trigger': False }}) entity_id = 'alarm_control_panel.test' assert STATE_ALARM_DISARMED == \ hass.states.get(entity_id).state await common.async_alarm_arm_away(hass, CODE, entity_id) assert STATE_ALARM_ARMED_AWAY == \ hass.states.get(entity_id).state await common.async_alarm_trigger(hass, entity_id=entity_id) assert STATE_ALARM_TRIGGERED == \ hass.states.get(entity_id).state future = dt_util.utcnow() + timedelta(seconds=5) with patch(('homeassistant.components.manual.alarm_control_panel.' 'dt_util.utcnow'), return_value=future): async_fire_time_changed(hass, future) await hass.async_block_till_done() assert STATE_ALARM_ARMED_AWAY == \ hass.states.get(entity_id).state await common.async_alarm_trigger(hass, entity_id=entity_id) assert STATE_ALARM_TRIGGERED == \ hass.states.get(entity_id).state future = dt_util.utcnow() + timedelta(seconds=5) with patch(('homeassistant.components.manual.alarm_control_panel.' 'dt_util.utcnow'), return_value=future): async_fire_time_changed(hass, future) await hass.async_block_till_done() assert STATE_ALARM_ARMED_AWAY == \ hass.states.get(entity_id).state async def test_disarm_while_pending_trigger(hass): """Test disarming while pending state.""" assert await async_setup_component( hass, alarm_control_panel.DOMAIN, {'alarm_control_panel': { 'platform': 'manual', 'name': 'test', 'trigger_time': 5, 'disarm_after_trigger': False }}) entity_id = 'alarm_control_panel.test' assert STATE_ALARM_DISARMED == \ hass.states.get(entity_id).state await common.async_alarm_trigger(hass) assert STATE_ALARM_PENDING == \ hass.states.get(entity_id).state await common.async_alarm_disarm(hass, entity_id=entity_id) assert STATE_ALARM_DISARMED == \ hass.states.get(entity_id).state future = dt_util.utcnow() + timedelta(seconds=5) with patch(('homeassistant.components.manual.alarm_control_panel.' 'dt_util.utcnow'), return_value=future): async_fire_time_changed(hass, future) await hass.async_block_till_done() assert STATE_ALARM_DISARMED == \ hass.states.get(entity_id).state async def test_disarm_during_trigger_with_invalid_code(hass): """Test disarming while code is invalid.""" assert await async_setup_component( hass, alarm_control_panel.DOMAIN, {'alarm_control_panel': { 'platform': 'manual', 'name': 'test', 'pending_time': 5, 'code': CODE + '2', 'disarm_after_trigger': False }}) entity_id = 'alarm_control_panel.test' assert STATE_ALARM_DISARMED == \ hass.states.get(entity_id).state await common.async_alarm_trigger(hass) assert STATE_ALARM_PENDING == \ hass.states.get(entity_id).state await common.async_alarm_disarm(hass, entity_id=entity_id) assert STATE_ALARM_PENDING == \ hass.states.get(entity_id).state future = dt_util.utcnow() + timedelta(seconds=5) with patch(('homeassistant.components.manual.alarm_control_panel.' 'dt_util.utcnow'), return_value=future): async_fire_time_changed(hass, future) await hass.async_block_till_done() assert STATE_ALARM_TRIGGERED == \ hass.states.get(entity_id).state async def test_disarm_with_template_code(hass): """Attempt to disarm with a valid or invalid template-based code.""" assert await async_setup_component( hass, alarm_control_panel.DOMAIN, {'alarm_control_panel': { 'platform': 'manual', 'name': 'test', 'code_template': '{{ "" if from_state == "disarmed" else "abc" }}', 'pending_time': 0, 'disarm_after_trigger': False }}) entity_id = 'alarm_control_panel.test' assert STATE_ALARM_DISARMED == \ hass.states.get(entity_id).state await common.async_alarm_arm_home(hass, 'def') state = hass.states.get(entity_id) assert STATE_ALARM_ARMED_HOME == state.state await common.async_alarm_disarm(hass, 'def') state = hass.states.get(entity_id) assert STATE_ALARM_ARMED_HOME == state.state await common.async_alarm_disarm(hass, 'abc') state = hass.states.get(entity_id) assert STATE_ALARM_DISARMED == state.state async def test_arm_custom_bypass_no_pending(hass): """Test arm custom bypass method.""" assert await async_setup_component( hass, alarm_control_panel.DOMAIN, {'alarm_control_panel': { 'platform': 'manual', 'name': 'test', 'code': CODE, 'pending_time': 0, 'disarm_after_trigger': False }}) entity_id = 'alarm_control_panel.test' assert STATE_ALARM_DISARMED == \ hass.states.get(entity_id).state await common.async_alarm_arm_custom_bypass(hass, CODE) assert STATE_ALARM_ARMED_CUSTOM_BYPASS == \ hass.states.get(entity_id).state async def test_arm_custom_bypass_no_pending_when_code_not_req(hass): """Test arm custom bypass method.""" assert await async_setup_component( hass, alarm_control_panel.DOMAIN, {'alarm_control_panel': { 'platform': 'manual', 'name': 'test', 'code': CODE, 'code_arm_required': False, 'pending_time': 0, 'disarm_after_trigger': False }}) entity_id = 'alarm_control_panel.test' assert STATE_ALARM_DISARMED == \ hass.states.get(entity_id).state common.async_alarm_arm_custom_bypass(hass, 0) await hass.async_block_till_done() assert STATE_ALARM_ARMED_CUSTOM_BYPASS == \ hass.states.get(entity_id).state async def test_arm_custom_bypass_with_pending(hass): """Test arm custom bypass method.""" assert await async_setup_component( hass, alarm_control_panel.DOMAIN, {'alarm_control_panel': { 'platform': 'manual', 'name': 'test', 'code': CODE, 'pending_time': 1, 'disarm_after_trigger': False }}) entity_id = 'alarm_control_panel.test' assert STATE_ALARM_DISARMED == \ hass.states.get(entity_id).state await common.async_alarm_arm_custom_bypass(hass, CODE, entity_id) assert STATE_ALARM_PENDING == \ hass.states.get(entity_id).state state = hass.states.get(entity_id) assert state.attributes['post_pending_state'] == \ STATE_ALARM_ARMED_CUSTOM_BYPASS future = dt_util.utcnow() + timedelta(seconds=1) with patch(('homeassistant.components.manual.alarm_control_panel.' 'dt_util.utcnow'), return_value=future): async_fire_time_changed(hass, future) await hass.async_block_till_done() state = hass.states.get(entity_id) assert state.state == STATE_ALARM_ARMED_CUSTOM_BYPASS async def test_arm_custom_bypass_with_invalid_code(hass): """Attempt to custom bypass without a valid code.""" assert await async_setup_component( hass, alarm_control_panel.DOMAIN, {'alarm_control_panel': { 'platform': 'manual', 'name': 'test', 'code': CODE, 'pending_time': 1, 'disarm_after_trigger': False }}) entity_id = 'alarm_control_panel.test' assert STATE_ALARM_DISARMED == \ hass.states.get(entity_id).state await common.async_alarm_arm_custom_bypass(hass, CODE + '2') assert STATE_ALARM_DISARMED == \ hass.states.get(entity_id).state async def test_armed_custom_bypass_with_specific_pending(hass): """Test arm custom bypass method.""" assert await async_setup_component( hass, alarm_control_panel.DOMAIN, {'alarm_control_panel': { 'platform': 'manual', 'name': 'test', 'pending_time': 10, 'armed_custom_bypass': { 'pending_time': 2 } }}) entity_id = 'alarm_control_panel.test' await common.async_alarm_arm_custom_bypass(hass) assert STATE_ALARM_PENDING == \ hass.states.get(entity_id).state future = dt_util.utcnow() + timedelta(seconds=2) with patch(('homeassistant.components.manual.alarm_control_panel.' 'dt_util.utcnow'), return_value=future): async_fire_time_changed(hass, future) await hass.async_block_till_done() assert STATE_ALARM_ARMED_CUSTOM_BYPASS == \ hass.states.get(entity_id).state async def test_arm_away_after_disabled_disarmed(hass): """Test pending state with and without zero trigger time.""" assert await async_setup_component( hass, alarm_control_panel.DOMAIN, {'alarm_control_panel': { 'platform': 'manual', 'name': 'test', 'code': CODE, 'pending_time': 0, 'delay_time': 1, 'armed_away': { 'pending_time': 1, }, 'disarmed': { 'trigger_time': 0 }, 'disarm_after_trigger': False }}) entity_id = 'alarm_control_panel.test' assert STATE_ALARM_DISARMED == \ hass.states.get(entity_id).state await common.async_alarm_arm_away(hass, CODE) state = hass.states.get(entity_id) assert STATE_ALARM_PENDING == state.state assert STATE_ALARM_DISARMED == \ state.attributes['pre_pending_state'] assert STATE_ALARM_ARMED_AWAY == \ state.attributes['post_pending_state'] await common.async_alarm_trigger(hass, entity_id=entity_id) state = hass.states.get(entity_id) assert STATE_ALARM_PENDING == state.state assert STATE_ALARM_DISARMED == \ state.attributes['pre_pending_state'] assert STATE_ALARM_ARMED_AWAY == \ state.attributes['post_pending_state'] future = dt_util.utcnow() + timedelta(seconds=1) with patch(('homeassistant.components.manual.alarm_control_panel.' 'dt_util.utcnow'), return_value=future): async_fire_time_changed(hass, future) await hass.async_block_till_done() state = hass.states.get(entity_id) assert STATE_ALARM_ARMED_AWAY == state.state await common.async_alarm_trigger(hass, entity_id=entity_id) state = hass.states.get(entity_id) assert STATE_ALARM_PENDING == state.state assert STATE_ALARM_ARMED_AWAY == \ state.attributes['pre_pending_state'] assert STATE_ALARM_TRIGGERED == \ state.attributes['post_pending_state'] future += timedelta(seconds=1) with patch(('homeassistant.components.manual.alarm_control_panel.' 'dt_util.utcnow'), return_value=future): async_fire_time_changed(hass, future) await hass.async_block_till_done() state = hass.states.get(entity_id) assert STATE_ALARM_TRIGGERED == state.state async def test_restore_armed_state(hass): """Ensure armed state is restored on startup.""" mock_restore_cache(hass, ( State('alarm_control_panel.test', STATE_ALARM_ARMED_AWAY), )) hass.state = CoreState.starting mock_component(hass, 'recorder') assert await async_setup_component(hass, alarm_control_panel.DOMAIN, { 'alarm_control_panel': { 'platform': 'manual', 'name': 'test', 'pending_time': 0, 'trigger_time': 0, 'disarm_after_trigger': False }}) state = hass.states.get('alarm_control_panel.test') assert state assert state.state == STATE_ALARM_ARMED_AWAY async def test_restore_disarmed_state(hass): """Ensure disarmed state is restored on startup.""" mock_restore_cache(hass, ( State('alarm_control_panel.test', STATE_ALARM_DISARMED), )) hass.state = CoreState.starting mock_component(hass, 'recorder') assert await async_setup_component(hass, alarm_control_panel.DOMAIN, { 'alarm_control_panel': { 'platform': 'manual', 'name': 'test', 'pending_time': 0, 'trigger_time': 0, 'disarm_after_trigger': False }}) state = hass.states.get('alarm_control_panel.test') assert state assert state.state == STATE_ALARM_DISARMED