310 lines
10 KiB
Python
310 lines
10 KiB
Python
|
"""The tests for the Template lock platform."""
|
||
|
|
import logging
|
||
|
|
|
||
|
|
from homeassistant.core import callback
|
||
|
|
from homeassistant import setup
|
||
|
|
from homeassistant.components import lock
|
||
|
|
from homeassistant.const import STATE_ON, STATE_OFF
|
||
|
|
|
||
|
|
from tests.common import (get_test_home_assistant,
|
||
|
|
assert_setup_component)
|
||
|
|
|
||
|
|
_LOGGER = logging.getLogger(__name__)
|
||
|
|
|
||
|
|
|
||
|
|
class TestTemplateLock:
|
||
|
|
"""Test the Template lock."""
|
||
|
|
|
||
|
|
hass = None
|
||
|
|
calls = None
|
||
|
|
# pylint: disable=invalid-name
|
||
|
|
|
||
|
|
def setup_method(self, method):
|
||
|
|
"""Set up things to be run when tests are started."""
|
||
|
|
self.hass = get_test_home_assistant()
|
||
|
|
self.calls = []
|
||
|
|
|
||
|
|
@callback
|
||
|
|
def record_call(service):
|
||
|
|
"""Track function calls."""
|
||
|
|
self.calls.append(service)
|
||
|
|
|
||
|
|
self.hass.services.register('test', 'automation', record_call)
|
||
|
|
|
||
|
|
def teardown_method(self, method):
|
||
|
|
"""Stop everything that was started."""
|
||
|
|
self.hass.stop()
|
||
|
|
|
||
|
|
def test_template_state(self):
|
||
|
|
"""Test template."""
|
||
|
|
with assert_setup_component(1, 'lock'):
|
||
|
|
assert setup.setup_component(self.hass, 'lock', {
|
||
|
|
'lock': {
|
||
|
|
'platform': 'template',
|
||
|
|
'name': 'Test template lock',
|
||
|
|
'value_template':
|
||
|
|
"{{ states.switch.test_state.state }}",
|
||
|
|
'lock': {
|
||
|
|
'service': 'switch.turn_on',
|
||
|
|
'entity_id': 'switch.test_state'
|
||
|
|
},
|
||
|
|
'unlock': {
|
||
|
|
'service': 'switch.turn_off',
|
||
|
|
'entity_id': 'switch.test_state'
|
||
|
|
}
|
||
|
|
}
|
||
|
|
})
|
||
|
|
|
||
|
|
self.hass.start()
|
||
|
|
self.hass.block_till_done()
|
||
|
|
|
||
|
|
self.hass.states.set('switch.test_state', STATE_ON)
|
||
|
|
self.hass.block_till_done()
|
||
|
|
|
||
|
|
state = self.hass.states.get('lock.test_template_lock')
|
||
|
|
assert state.state == lock.STATE_LOCKED
|
||
|
|
|
||
|
|
self.hass.states.set('switch.test_state', STATE_OFF)
|
||
|
|
self.hass.block_till_done()
|
||
|
|
|
||
|
|
state = self.hass.states.get('lock.test_template_lock')
|
||
|
|
assert state.state == lock.STATE_UNLOCKED
|
||
|
|
|
||
|
|
def test_template_state_boolean_on(self):
|
||
|
|
"""Test the setting of the state with boolean on."""
|
||
|
|
with assert_setup_component(1, 'lock'):
|
||
|
|
assert setup.setup_component(self.hass, 'lock', {
|
||
|
|
'lock': {
|
||
|
|
'platform': 'template',
|
||
|
|
'value_template':
|
||
|
|
"{{ 1 == 1 }}",
|
||
|
|
'lock': {
|
||
|
|
'service': 'switch.turn_on',
|
||
|
|
'entity_id': 'switch.test_state'
|
||
|
|
},
|
||
|
|
'unlock': {
|
||
|
|
'service': 'switch.turn_off',
|
||
|
|
'entity_id': 'switch.test_state'
|
||
|
|
}
|
||
|
|
}
|
||
|
|
})
|
||
|
|
|
||
|
|
self.hass.start()
|
||
|
|
self.hass.block_till_done()
|
||
|
|
|
||
|
|
state = self.hass.states.get('lock.template_lock')
|
||
|
|
assert state.state == lock.STATE_LOCKED
|
||
|
|
|
||
|
|
def test_template_state_boolean_off(self):
|
||
|
|
"""Test the setting of the state with off."""
|
||
|
|
with assert_setup_component(1, 'lock'):
|
||
|
|
assert setup.setup_component(self.hass, 'lock', {
|
||
|
|
'lock': {
|
||
|
|
'platform': 'template',
|
||
|
|
'value_template':
|
||
|
|
"{{ 1 == 2 }}",
|
||
|
|
'lock': {
|
||
|
|
'service': 'switch.turn_on',
|
||
|
|
'entity_id': 'switch.test_state'
|
||
|
|
},
|
||
|
|
'unlock': {
|
||
|
|
'service': 'switch.turn_off',
|
||
|
|
'entity_id': 'switch.test_state'
|
||
|
|
}
|
||
|
|
}
|
||
|
|
})
|
||
|
|
|
||
|
|
self.hass.start()
|
||
|
|
self.hass.block_till_done()
|
||
|
|
|
||
|
|
state = self.hass.states.get('lock.template_lock')
|
||
|
|
assert state.state == lock.STATE_UNLOCKED
|
||
|
|
|
||
|
|
def test_template_syntax_error(self):
|
||
|
|
"""Test templating syntax error."""
|
||
|
|
with assert_setup_component(0, 'lock'):
|
||
|
|
assert setup.setup_component(self.hass, 'lock', {
|
||
|
|
'lock': {
|
||
|
|
'platform': 'template',
|
||
|
|
'value_template':
|
||
|
|
"{% if rubbish %}",
|
||
|
|
'lock': {
|
||
|
|
'service': 'switch.turn_on',
|
||
|
|
'entity_id': 'switch.test_state'
|
||
|
|
},
|
||
|
|
'unlock': {
|
||
|
|
'service': 'switch.turn_off',
|
||
|
|
'entity_id': 'switch.test_state'
|
||
|
|
}
|
||
|
|
}
|
||
|
|
})
|
||
|
|
|
||
|
|
self.hass.start()
|
||
|
|
self.hass.block_till_done()
|
||
|
|
|
||
|
|
assert self.hass.states.all() == []
|
||
|
|
|
||
|
|
def test_invalid_name_does_not_create(self):
|
||
|
|
"""Test invalid name."""
|
||
|
|
with assert_setup_component(0, 'lock'):
|
||
|
|
assert setup.setup_component(self.hass, 'lock', {
|
||
|
|
'switch': {
|
||
|
|
'platform': 'lock',
|
||
|
|
'name': '{{%}',
|
||
|
|
'value_template':
|
||
|
|
"{{ rubbish }",
|
||
|
|
'lock': {
|
||
|
|
'service': 'switch.turn_on',
|
||
|
|
'entity_id': 'switch.test_state'
|
||
|
|
},
|
||
|
|
'unlock': {
|
||
|
|
'service': 'switch.turn_off',
|
||
|
|
'entity_id': 'switch.test_state'
|
||
|
|
}
|
||
|
|
}
|
||
|
|
})
|
||
|
|
|
||
|
|
self.hass.start()
|
||
|
|
self.hass.block_till_done()
|
||
|
|
|
||
|
|
assert self.hass.states.all() == []
|
||
|
|
|
||
|
|
def test_invalid_lock_does_not_create(self):
|
||
|
|
"""Test invalid lock."""
|
||
|
|
with assert_setup_component(0, 'lock'):
|
||
|
|
assert setup.setup_component(self.hass, 'lock', {
|
||
|
|
'lock': {
|
||
|
|
'platform': 'template',
|
||
|
|
'value_template': "Invalid"
|
||
|
|
}
|
||
|
|
})
|
||
|
|
|
||
|
|
self.hass.start()
|
||
|
|
self.hass.block_till_done()
|
||
|
|
|
||
|
|
assert self.hass.states.all() == []
|
||
|
|
|
||
|
|
def test_missing_template_does_not_create(self):
|
||
|
|
"""Test missing template."""
|
||
|
|
with assert_setup_component(0, 'lock'):
|
||
|
|
assert setup.setup_component(self.hass, 'lock', {
|
||
|
|
'lock': {
|
||
|
|
'platform': 'template',
|
||
|
|
'not_value_template':
|
||
|
|
"{{ states.switch.test_state.state }}",
|
||
|
|
'lock': {
|
||
|
|
'service': 'switch.turn_on',
|
||
|
|
'entity_id': 'switch.test_state'
|
||
|
|
},
|
||
|
|
'unlock': {
|
||
|
|
'service': 'switch.turn_off',
|
||
|
|
'entity_id': 'switch.test_state'
|
||
|
|
}
|
||
|
|
}
|
||
|
|
})
|
||
|
|
|
||
|
|
self.hass.start()
|
||
|
|
self.hass.block_till_done()
|
||
|
|
|
||
|
|
assert self.hass.states.all() == []
|
||
|
|
|
||
|
|
def test_no_template_match_all(self, caplog):
|
||
|
|
"""Test that we do not allow locks that match on all."""
|
||
|
|
with assert_setup_component(1, 'lock'):
|
||
|
|
assert setup.setup_component(self.hass, 'lock', {
|
||
|
|
'lock': {
|
||
|
|
'platform': 'template',
|
||
|
|
'value_template': '{{ 1 + 1 }}',
|
||
|
|
'lock': {
|
||
|
|
'service': 'switch.turn_on',
|
||
|
|
'entity_id': 'switch.test_state'
|
||
|
|
},
|
||
|
|
'unlock': {
|
||
|
|
'service': 'switch.turn_off',
|
||
|
|
'entity_id': 'switch.test_state'
|
||
|
|
}
|
||
|
|
}
|
||
|
|
})
|
||
|
|
|
||
|
|
self.hass.start()
|
||
|
|
self.hass.block_till_done()
|
||
|
|
|
||
|
|
state = self.hass.states.get('lock.template_lock')
|
||
|
|
assert state.state == lock.STATE_UNLOCKED
|
||
|
|
|
||
|
|
assert ('Template lock Template Lock has no entity ids configured '
|
||
|
|
'to track nor were we able to extract the entities to track '
|
||
|
|
'from the value_template template. This entity will only '
|
||
|
|
'be able to be updated manually.') in caplog.text
|
||
|
|
|
||
|
|
self.hass.states.set('lock.template_lock', lock.STATE_LOCKED)
|
||
|
|
self.hass.block_till_done()
|
||
|
|
state = self.hass.states.get('lock.template_lock')
|
||
|
|
assert state.state == lock.STATE_LOCKED
|
||
|
|
|
||
|
|
def test_lock_action(self):
|
||
|
|
"""Test lock action."""
|
||
|
|
assert setup.setup_component(self.hass, 'lock', {
|
||
|
|
'lock': {
|
||
|
|
'platform': 'template',
|
||
|
|
'value_template':
|
||
|
|
"{{ states.switch.test_state.state }}",
|
||
|
|
'lock': {
|
||
|
|
'service': 'test.automation'
|
||
|
|
},
|
||
|
|
'unlock': {
|
||
|
|
'service': 'switch.turn_off',
|
||
|
|
'entity_id': 'switch.test_state'
|
||
|
|
}
|
||
|
|
}
|
||
|
|
})
|
||
|
|
|
||
|
|
self.hass.start()
|
||
|
|
self.hass.block_till_done()
|
||
|
|
|
||
|
|
self.hass.states.set('switch.test_state', STATE_OFF)
|
||
|
|
self.hass.block_till_done()
|
||
|
|
|
||
|
|
state = self.hass.states.get('lock.template_lock')
|
||
|
|
assert state.state == lock.STATE_UNLOCKED
|
||
|
|
|
||
|
|
self.hass.services.call(lock.DOMAIN, lock.SERVICE_LOCK, {
|
||
|
|
lock.ATTR_ENTITY_ID: 'lock.template_lock'
|
||
|
|
})
|
||
|
|
self.hass.block_till_done()
|
||
|
|
|
||
|
|
assert len(self.calls) == 1
|
||
|
|
|
||
|
|
def test_unlock_action(self):
|
||
|
|
"""Test unlock action."""
|
||
|
|
assert setup.setup_component(self.hass, 'lock', {
|
||
|
|
'lock': {
|
||
|
|
'platform': 'template',
|
||
|
|
'value_template':
|
||
|
|
"{{ states.switch.test_state.state }}",
|
||
|
|
'lock': {
|
||
|
|
'service': 'switch.turn_on',
|
||
|
|
'entity_id': 'switch.test_state'
|
||
|
|
},
|
||
|
|
'unlock': {
|
||
|
|
'service': 'test.automation'
|
||
|
|
}
|
||
|
|
}
|
||
|
|
})
|
||
|
|
|
||
|
|
self.hass.start()
|
||
|
|
self.hass.block_till_done()
|
||
|
|
|
||
|
|
self.hass.states.set('switch.test_state', STATE_ON)
|
||
|
|
self.hass.block_till_done()
|
||
|
|
|
||
|
|
state = self.hass.states.get('lock.template_lock')
|
||
|
|
assert state.state == lock.STATE_LOCKED
|
||
|
|
|
||
|
|
self.hass.services.call(lock.DOMAIN, lock.SERVICE_UNLOCK, {
|
||
|
|
lock.ATTR_ENTITY_ID: 'lock.template_lock'
|
||
|
|
})
|
||
|
|
self.hass.block_till_done()
|
||
|
|
|
||
|
|
assert len(self.calls) == 1
|