Improve editing of device automation referring non added select entity (#52047)

* Improve editing of device automation referring non added select entity

* Update tests
This commit is contained in:
Erik Montnemery 2021-06-21 14:49:51 +02:00 committed by GitHub
parent f29bcf7ff7
commit b916247e8e
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
8 changed files with 79 additions and 31 deletions

View file

@ -94,7 +94,7 @@ async def test_action(hass: HomeAssistant) -> None:
assert select_calls[0].data == {"entity_id": "select.entity", "option": "option1"}
async def test_get_trigger_capabilities(hass: HomeAssistant) -> None:
async def test_get_action_capabilities(hass: HomeAssistant) -> None:
"""Test we get the expected capabilities from a select action."""
config = {
"platform": "device",
@ -106,7 +106,18 @@ async def test_get_trigger_capabilities(hass: HomeAssistant) -> None:
# Test when entity doesn't exists
capabilities = await async_get_action_capabilities(hass, config)
assert capabilities == {}
assert capabilities
assert "extra_fields" in capabilities
assert voluptuous_serialize.convert(
capabilities["extra_fields"], custom_serializer=cv.custom_serializer
) == [
{
"name": "option",
"required": True,
"type": "select",
"options": [],
},
]
# Mock an entity
hass.states.async_set("select.test", "option1", {"options": ["option1", "option2"]})