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

Use DeviceAutomationType in tests/components/[m-r]* (#62443)

Browse source
This commit is contained in:
Ville Skyttä 2021-12-20 23:29:22 +02:00 • committed by GitHub
parent 5926961ed5
commit f913961d63
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
14 changed files with 121 additions and 40 deletions
Download patch file Download diff file Expand all files Collapse all files

21
tests/components/nest/test_device_trigger.py
View file

@ -4,6 +4,7 @@ from google_nest_sdm.event import EventMessage
import pytest
import homeassistant.components.automation as automation
from homeassistant.components.device_automation import DeviceAutomationType
from homeassistant.components.device_automation.exceptions import (
InvalidDeviceAutomationConfig,
)
@ -119,7 +120,9 @@ async def test_get_triggers(hass):
"device_id": device_entry.id,
},
]
triggers = await async_get_device_automations(hass, "trigger", device_entry.id)
triggers = await async_get_device_automations(
hass, DeviceAutomationType.TRIGGER, device_entry.id
)
assert_lists_same(triggers, expected_triggers)
@ -147,7 +150,9 @@ async def test_multiple_devices(hass):
entry2 = registry.async_get("camera.camera_2")
assert entry2.unique_id == "device-id-2-camera"
triggers = await async_get_device_automations(hass, "trigger", entry1.device_id)
triggers = await async_get_device_automations(
hass, DeviceAutomationType.TRIGGER, entry1.device_id
)
assert len(triggers) == 1
assert triggers[0] == {
"platform": "device",
@ -156,7 +161,9 @@ async def test_multiple_devices(hass):
"device_id": entry1.device_id,
}
triggers = await async_get_device_automations(hass, "trigger", entry2.device_id)
triggers = await async_get_device_automations(
hass, DeviceAutomationType.TRIGGER, entry2.device_id
)
assert len(triggers) == 1
assert triggers[0] == {
"platform": "device",
@ -191,7 +198,9 @@ async def test_triggers_for_invalid_device_id(hass):
assert device_entry_2 is not None
with pytest.raises(InvalidDeviceAutomationConfig):
await async_get_device_automations(hass, "trigger", device_entry_2.id)
await async_get_device_automations(
hass, DeviceAutomationType.TRIGGER, device_entry_2.id
)
async def test_no_triggers(hass):
@ -203,7 +212,9 @@ async def test_no_triggers(hass):
entry = registry.async_get("camera.my_camera")
assert entry.unique_id == "some-device-id-camera"
triggers = await async_get_device_automations(hass, "trigger", entry.device_id)
triggers = await async_get_device_automations(
hass, DeviceAutomationType.TRIGGER, entry.device_id
)
assert triggers == []