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

Black

Browse source
This commit is contained in:
Paulus Schoutsen 2019-07-31 12:25:30 -07:00
parent da05dfe708
commit 4de97abc3a
2676 changed files with 163166 additions and 140084 deletions
Download patch file Download diff file Expand all files Collapse all files

56
tests/components/device_automation/test_init.py
View file

@ -6,8 +6,7 @@ from homeassistant.components.websocket_api.const import TYPE_RESULT
from homeassistant.helpers import device_registry
from tests.common import (
MockConfigEntry, mock_device_registry, mock_registry)
from tests.common import MockConfigEntry, mock_device_registry, mock_registry
@pytest.fixture
@ -32,36 +31,45 @@ def _same_triggers(a, b):
return True
async def test_websocket_get_triggers(
hass, hass_ws_client, device_reg, entity_reg):
async def test_websocket_get_triggers(hass, hass_ws_client, device_reg, entity_reg):
"""Test we get the expected triggers from a light through websocket."""
await async_setup_component(hass, 'device_automation', {})
config_entry = MockConfigEntry(domain='test', data={})
await async_setup_component(hass, "device_automation", {})
config_entry = MockConfigEntry(domain="test", data={})
config_entry.add_to_hass(hass)
device_entry = device_reg.async_get_or_create(
config_entry_id=config_entry.entry_id,
connections={
(device_registry.CONNECTION_NETWORK_MAC, '12:34:56:AB:CD:EF')
})
entity_reg.async_get_or_create(
'light', 'test', '5678', device_id=device_entry.id)
connections={(device_registry.CONNECTION_NETWORK_MAC, "12:34:56:AB:CD:EF")},
)
entity_reg.async_get_or_create("light", "test", "5678", device_id=device_entry.id)
expected_triggers = [
{'platform': 'device', 'domain': 'light', 'type': 'turn_off',
'device_id': device_entry.id, 'entity_id': 'light.test_5678'},
{'platform': 'device', 'domain': 'light', 'type': 'turn_on',
'device_id': device_entry.id, 'entity_id': 'light.test_5678'},
{
"platform": "device",
"domain": "light",
"type": "turn_off",
"device_id": device_entry.id,
"entity_id": "light.test_5678",
},
{
"platform": "device",
"domain": "light",
"type": "turn_on",
"device_id": device_entry.id,
"entity_id": "light.test_5678",
},
]
client = await hass_ws_client(hass)
await client.send_json({
'id': 1,
'type': 'device_automation/list_triggers',
'device_id': device_entry.id
})
await client.send_json(
{
"id": 1,
"type": "device_automation/list_triggers",
"device_id": device_entry.id,
}
)
msg = await client.receive_json()
assert msg['id'] == 1
assert msg['type'] == TYPE_RESULT
assert msg['success']
triggers = msg['result']['triggers']
assert msg["id"] == 1
assert msg["type"] == TYPE_RESULT
assert msg["success"]
triggers = msg["result"]["triggers"]
assert _same_triggers(triggers, expected_triggers)