mirror of
https://github.com/home-assistant/core.git
synced 2025-06-12 09:07:08 +00:00
337 lines
11 KiB
Python
337 lines
11 KiB
Python
"""The tests for Network UPS Tools (NUT) device actions."""
|
|
|
|
from unittest.mock import AsyncMock
|
|
|
|
from aionut import NUTError
|
|
import pytest
|
|
from pytest_unordered import unordered
|
|
|
|
from homeassistant.components import automation, device_automation
|
|
from homeassistant.components.device_automation import (
|
|
DeviceAutomationType,
|
|
InvalidDeviceAutomationConfig,
|
|
)
|
|
from homeassistant.components.nut import DOMAIN
|
|
from homeassistant.components.nut.const import INTEGRATION_SUPPORTED_COMMANDS
|
|
from homeassistant.const import CONF_DEVICE_ID, CONF_TYPE
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.exceptions import HomeAssistantError
|
|
from homeassistant.helpers import device_registry as dr
|
|
from homeassistant.setup import async_setup_component
|
|
|
|
from .util import async_init_integration
|
|
|
|
from tests.common import MockConfigEntry, async_get_device_automations
|
|
|
|
|
|
async def test_get_all_actions_for_specified_user(
|
|
hass: HomeAssistant,
|
|
device_registry: dr.DeviceRegistry,
|
|
) -> None:
|
|
"""Test we get all the expected actions from a nut if user is specified."""
|
|
list_commands_return_value = {
|
|
supported_command: supported_command
|
|
for supported_command in INTEGRATION_SUPPORTED_COMMANDS
|
|
}
|
|
|
|
await async_init_integration(
|
|
hass,
|
|
username="someuser",
|
|
password="somepassword",
|
|
list_vars={"ups.status": "OL"},
|
|
list_commands_return_value=list_commands_return_value,
|
|
)
|
|
device_entry = next(device for device in device_registry.devices.values())
|
|
expected_actions = [
|
|
{
|
|
"domain": DOMAIN,
|
|
"type": action.replace(".", "_"),
|
|
"device_id": device_entry.id,
|
|
"metadata": {},
|
|
}
|
|
for action in INTEGRATION_SUPPORTED_COMMANDS
|
|
]
|
|
actions = await async_get_device_automations(
|
|
hass, DeviceAutomationType.ACTION, device_entry.id
|
|
)
|
|
assert actions == unordered(expected_actions)
|
|
|
|
|
|
async def test_no_actions_for_anonymous_user(
|
|
hass: HomeAssistant,
|
|
device_registry: dr.DeviceRegistry,
|
|
) -> None:
|
|
"""Test we get no actions if user is not specified."""
|
|
list_commands_return_value = {"some action": "some description"}
|
|
|
|
await async_init_integration(
|
|
hass,
|
|
username=None,
|
|
password=None,
|
|
list_vars={"ups.status": "OL"},
|
|
list_commands_return_value=list_commands_return_value,
|
|
)
|
|
device_entry = next(device for device in device_registry.devices.values())
|
|
actions = await async_get_device_automations(
|
|
hass, DeviceAutomationType.ACTION, device_entry.id
|
|
)
|
|
|
|
assert len(actions) == 0
|
|
|
|
|
|
async def test_no_actions_device_not_found(
|
|
hass: HomeAssistant,
|
|
) -> None:
|
|
"""Test we get no actions for a device that cannot be found."""
|
|
list_commands_return_value = {"beeper.enable": None}
|
|
await async_init_integration(
|
|
hass,
|
|
list_vars={"ups.status": "OL"},
|
|
list_commands_return_value=list_commands_return_value,
|
|
)
|
|
|
|
device_id = "invalid_device_id"
|
|
platform = await device_automation.async_get_device_automation_platform(
|
|
hass, DOMAIN, DeviceAutomationType.ACTION
|
|
)
|
|
actions = await platform.async_get_actions(hass, device_id)
|
|
|
|
assert len(actions) == 0
|
|
|
|
|
|
async def test_no_actions_device_invalid(
|
|
hass: HomeAssistant,
|
|
device_registry: dr.DeviceRegistry,
|
|
) -> None:
|
|
"""Test we get no actions for a device that is invalid."""
|
|
list_commands_return_value = {"beeper.enable": None}
|
|
entry = await async_init_integration(
|
|
hass,
|
|
list_vars={"ups.status": "OL"},
|
|
list_commands_return_value=list_commands_return_value,
|
|
)
|
|
device_entry = next(device for device in device_registry.devices.values())
|
|
|
|
assert await hass.config_entries.async_unload(entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
platform = await device_automation.async_get_device_automation_platform(
|
|
hass, DOMAIN, DeviceAutomationType.ACTION
|
|
)
|
|
actions = await platform.async_get_actions(hass, device_entry.id)
|
|
|
|
assert len(actions) == 0
|
|
|
|
|
|
async def test_list_commands_exception(
|
|
hass: HomeAssistant, device_registry: dr.DeviceRegistry
|
|
) -> None:
|
|
"""Test there are no actions if list_commands raises exception."""
|
|
await async_init_integration(
|
|
hass, list_vars={"ups.status": "OL"}, list_commands_side_effect=NUTError
|
|
)
|
|
|
|
device_entry = next(device for device in device_registry.devices.values())
|
|
actions = await async_get_device_automations(
|
|
hass, DeviceAutomationType.ACTION, device_entry.id
|
|
)
|
|
assert len(actions) == 0
|
|
|
|
|
|
async def test_unsupported_command(
|
|
hass: HomeAssistant, device_registry: dr.DeviceRegistry
|
|
) -> None:
|
|
"""Test unsupported command is excluded."""
|
|
|
|
list_commands_return_value = {
|
|
"beeper.enable": None,
|
|
"device.something": "Does something unsupported",
|
|
}
|
|
await async_init_integration(
|
|
hass,
|
|
list_vars={"ups.status": "OL"},
|
|
list_commands_return_value=list_commands_return_value,
|
|
)
|
|
device_entry = next(device for device in device_registry.devices.values())
|
|
actions = await async_get_device_automations(
|
|
hass, DeviceAutomationType.ACTION, device_entry.id
|
|
)
|
|
assert len(actions) == 1
|
|
|
|
|
|
async def test_action(hass: HomeAssistant, device_registry: dr.DeviceRegistry) -> None:
|
|
"""Test actions are executed."""
|
|
|
|
list_commands_return_value = {
|
|
"beeper.enable": None,
|
|
"beeper.disable": None,
|
|
}
|
|
run_command = AsyncMock()
|
|
await async_init_integration(
|
|
hass,
|
|
list_ups={"someUps": "Some UPS"},
|
|
list_vars={"ups.status": "OL"},
|
|
list_commands_return_value=list_commands_return_value,
|
|
run_command=run_command,
|
|
)
|
|
device_entry = next(device for device in device_registry.devices.values())
|
|
|
|
assert await async_setup_component(
|
|
hass,
|
|
automation.DOMAIN,
|
|
{
|
|
automation.DOMAIN: [
|
|
{
|
|
"trigger": {
|
|
"platform": "event",
|
|
"event_type": "test_some_event",
|
|
},
|
|
"action": {
|
|
"domain": DOMAIN,
|
|
"device_id": device_entry.id,
|
|
"type": "beeper_enable",
|
|
},
|
|
},
|
|
{
|
|
"trigger": {
|
|
"platform": "event",
|
|
"event_type": "test_another_event",
|
|
},
|
|
"action": {
|
|
"domain": DOMAIN,
|
|
"device_id": device_entry.id,
|
|
"type": "beeper_disable",
|
|
},
|
|
},
|
|
]
|
|
},
|
|
)
|
|
|
|
hass.bus.async_fire("test_some_event")
|
|
await hass.async_block_till_done()
|
|
run_command.assert_called_with("someUps", "beeper.enable")
|
|
|
|
hass.bus.async_fire("test_another_event")
|
|
await hass.async_block_till_done()
|
|
run_command.assert_called_with("someUps", "beeper.disable")
|
|
|
|
|
|
async def test_run_command_exception(
|
|
hass: HomeAssistant,
|
|
device_registry: dr.DeviceRegistry,
|
|
) -> None:
|
|
"""Test if run command raises exception with translation."""
|
|
|
|
command_name = "beeper.enable"
|
|
nut_error_message = "Something wrong happened"
|
|
run_command = AsyncMock(side_effect=NUTError(nut_error_message))
|
|
await async_init_integration(
|
|
hass,
|
|
list_vars={"ups.status": "OL"},
|
|
list_ups={"ups1": "UPS 1"},
|
|
list_commands_return_value={command_name: None},
|
|
run_command=run_command,
|
|
)
|
|
device_entry = next(device for device in device_registry.devices.values())
|
|
|
|
platform = await device_automation.async_get_device_automation_platform(
|
|
hass, DOMAIN, DeviceAutomationType.ACTION
|
|
)
|
|
|
|
error_message = f"Error running command {command_name}, {nut_error_message}"
|
|
with pytest.raises(HomeAssistantError, match=error_message):
|
|
await platform.async_call_action_from_config(
|
|
hass,
|
|
{
|
|
CONF_TYPE: command_name,
|
|
CONF_DEVICE_ID: device_entry.id,
|
|
},
|
|
{},
|
|
None,
|
|
)
|
|
|
|
|
|
async def test_action_exception_device_not_found(hass: HomeAssistant) -> None:
|
|
"""Test raises exception if device not found."""
|
|
list_commands_return_value = {"beeper.enable": None}
|
|
await async_init_integration(
|
|
hass,
|
|
list_vars={"ups.status": "OL"},
|
|
list_commands_return_value=list_commands_return_value,
|
|
)
|
|
|
|
platform = await device_automation.async_get_device_automation_platform(
|
|
hass, DOMAIN, DeviceAutomationType.ACTION
|
|
)
|
|
|
|
device_id = "invalid_device_id"
|
|
error_message = f"Unable to find a NUT device with ID {device_id}"
|
|
with pytest.raises(InvalidDeviceAutomationConfig, match=error_message):
|
|
await platform.async_call_action_from_config(
|
|
hass,
|
|
{CONF_TYPE: "beeper.enable", CONF_DEVICE_ID: device_id},
|
|
{},
|
|
None,
|
|
)
|
|
|
|
|
|
async def test_action_exception_invalid_config(
|
|
hass: HomeAssistant,
|
|
device_registry: dr.DeviceRegistry,
|
|
) -> None:
|
|
"""Test raises exception if no NUT config entry found."""
|
|
|
|
config_entry = MockConfigEntry()
|
|
config_entry.add_to_hass(hass)
|
|
await hass.config_entries.async_setup(config_entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
device_entry = device_registry.async_get_or_create(
|
|
config_entry_id=config_entry.entry_id,
|
|
identifiers={(DOMAIN, "mock-identifier")},
|
|
)
|
|
|
|
platform = await device_automation.async_get_device_automation_platform(
|
|
hass, DOMAIN, DeviceAutomationType.ACTION
|
|
)
|
|
|
|
with pytest.raises(InvalidDeviceAutomationConfig):
|
|
await platform.async_call_action_from_config(
|
|
hass,
|
|
{CONF_TYPE: "beeper.enable", CONF_DEVICE_ID: device_entry.id},
|
|
{},
|
|
None,
|
|
)
|
|
|
|
|
|
async def test_action_exception_device_invalid(
|
|
hass: HomeAssistant,
|
|
device_registry: dr.DeviceRegistry,
|
|
) -> None:
|
|
"""Test raises exception if config entry for device is invalid."""
|
|
list_commands_return_value = {"beeper.enable": None}
|
|
entry = await async_init_integration(
|
|
hass,
|
|
list_vars={"ups.status": "OL"},
|
|
list_commands_return_value=list_commands_return_value,
|
|
)
|
|
device_entry = next(device for device in device_registry.devices.values())
|
|
|
|
assert await hass.config_entries.async_unload(entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
platform = await device_automation.async_get_device_automation_platform(
|
|
hass, DOMAIN, DeviceAutomationType.ACTION
|
|
)
|
|
|
|
error_message = (
|
|
f"Invalid configuration entries for NUT device with ID {device_entry.id}"
|
|
)
|
|
with pytest.raises(InvalidDeviceAutomationConfig, match=error_message):
|
|
await platform.async_call_action_from_config(
|
|
hass,
|
|
{CONF_TYPE: "beeper.enable", CONF_DEVICE_ID: device_entry.id},
|
|
{},
|
|
None,
|
|
)
|