evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update lock entity to support locking, unlocking, jammed (#51455)

Browse source
This commit is contained in:
J. Nick Koston 2021-07-20 06:12:56 -10:00 • committed by GitHub
parent 0cc4231ac2
commit 9b705ad6df
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
10 changed files with 418 additions and 27 deletions
Download patch file Download diff file Expand all files Collapse all files

101
tests/components/lock/test_device_condition.py
View file

@ -3,7 +3,13 @@ import pytest
import homeassistant.components.automation as automation
from homeassistant.components.lock import DOMAIN
from homeassistant.const import STATE_LOCKED, STATE_UNLOCKED
from homeassistant.const import (
STATE_JAMMED,
STATE_LOCKED,
STATE_LOCKING,
STATE_UNLOCKED,
STATE_UNLOCKING,
)
from homeassistant.helpers import device_registry
from homeassistant.setup import async_setup_component
@ -60,6 +66,27 @@ async def test_get_conditions(hass, device_reg, entity_reg):
"device_id": device_entry.id,
"entity_id": f"{DOMAIN}.test_5678",
},
{
"condition": "device",
"domain": DOMAIN,
"type": "is_unlocking",
"device_id": device_entry.id,
"entity_id": f"{DOMAIN}.test_5678",
},
{
"condition": "device",
"domain": DOMAIN,
"type": "is_locking",
"device_id": device_entry.id,
"entity_id": f"{DOMAIN}.test_5678",
},
{
"condition": "device",
"domain": DOMAIN,
"type": "is_jammed",
"device_id": device_entry.id,
"entity_id": f"{DOMAIN}.test_5678",
},
]
conditions = await async_get_device_automations(hass, "condition", device_entry.id)
assert_lists_same(conditions, expected_conditions)
@ -110,6 +137,60 @@ async def test_if_state(hass, calls):
},
},
},
{
"trigger": {"platform": "event", "event_type": "test_event3"},
"condition": [
{
"condition": "device",
"domain": DOMAIN,
"device_id": "",
"entity_id": "lock.entity",
"type": "is_unlocking",
}
],
"action": {
"service": "test.automation",
"data_template": {
"some": "is_unlocking - {{ trigger.platform }} - {{ trigger.event.event_type }}"
},
},
},
{
"trigger": {"platform": "event", "event_type": "test_event4"},
"condition": [
{
"condition": "device",
"domain": DOMAIN,
"device_id": "",
"entity_id": "lock.entity",
"type": "is_locking",
}
],
"action": {
"service": "test.automation",
"data_template": {
"some": "is_locking - {{ trigger.platform }} - {{ trigger.event.event_type }}"
},
},
},
{
"trigger": {"platform": "event", "event_type": "test_event5"},
"condition": [
{
"condition": "device",
"domain": DOMAIN,
"device_id": "",
"entity_id": "lock.entity",
"type": "is_jammed",
}
],
"action": {
"service": "test.automation",
"data_template": {
"some": "is_jammed - {{ trigger.platform }} - {{ trigger.event.event_type }}"
},
},
},
]
},
)
@ -125,3 +206,21 @@ async def test_if_state(hass, calls):
await hass.async_block_till_done()
assert len(calls) == 2
assert calls[1].data["some"] == "is_unlocked - event - test_event2"
hass.states.async_set("lock.entity", STATE_UNLOCKING)
hass.bus.async_fire("test_event3")
await hass.async_block_till_done()
assert len(calls) == 3
assert calls[2].data["some"] == "is_unlocking - event - test_event3"
hass.states.async_set("lock.entity", STATE_LOCKING)
hass.bus.async_fire("test_event4")
await hass.async_block_till_done()
assert len(calls) == 4
assert calls[3].data["some"] == "is_locking - event - test_event4"
hass.states.async_set("lock.entity", STATE_JAMMED)
hass.bus.async_fire("test_event5")
await hass.async_block_till_done()
assert len(calls) == 5
assert calls[4].data["some"] == "is_jammed - event - test_event5"