119 lines
3.9 KiB
Python
119 lines
3.9 KiB
Python
"""Tests for switch platform."""
|
|
|
|
from unittest.mock import AsyncMock, patch
|
|
|
|
from aioautomower.exceptions import ApiException
|
|
from aioautomower.model import MowerStates, RestrictedReasons
|
|
from aioautomower.utils import mower_list_to_dictionary_dataclass
|
|
from freezegun.api import FrozenDateTimeFactory
|
|
import pytest
|
|
from syrupy import SnapshotAssertion
|
|
|
|
from homeassistant.components.husqvarna_automower.const import DOMAIN
|
|
from homeassistant.components.husqvarna_automower.coordinator import SCAN_INTERVAL
|
|
from homeassistant.const import Platform
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.exceptions import HomeAssistantError
|
|
from homeassistant.helpers import entity_registry as er
|
|
|
|
from . import setup_integration
|
|
from .const import TEST_MOWER_ID
|
|
|
|
from tests.common import (
|
|
MockConfigEntry,
|
|
async_fire_time_changed,
|
|
load_json_value_fixture,
|
|
)
|
|
|
|
|
|
async def test_switch_states(
|
|
hass: HomeAssistant,
|
|
mock_automower_client: AsyncMock,
|
|
mock_config_entry: MockConfigEntry,
|
|
freezer: FrozenDateTimeFactory,
|
|
) -> None:
|
|
"""Test switch state."""
|
|
values = mower_list_to_dictionary_dataclass(
|
|
load_json_value_fixture("mower.json", DOMAIN)
|
|
)
|
|
await setup_integration(hass, mock_config_entry)
|
|
|
|
for state, restricted_reson, expected_state in [
|
|
(MowerStates.RESTRICTED, RestrictedReasons.NOT_APPLICABLE, "off"),
|
|
(MowerStates.IN_OPERATION, RestrictedReasons.NONE, "on"),
|
|
]:
|
|
values[TEST_MOWER_ID].mower.state = state
|
|
values[TEST_MOWER_ID].planner.restricted_reason = restricted_reson
|
|
mock_automower_client.get_status.return_value = values
|
|
freezer.tick(SCAN_INTERVAL)
|
|
async_fire_time_changed(hass)
|
|
await hass.async_block_till_done()
|
|
state = hass.states.get("switch.test_mower_1_enable_schedule")
|
|
assert state.state == expected_state
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("service", "aioautomower_command"),
|
|
[
|
|
("turn_off", "park_until_further_notice"),
|
|
("turn_on", "resume_schedule"),
|
|
("toggle", "park_until_further_notice"),
|
|
],
|
|
)
|
|
async def test_switch_commands(
|
|
hass: HomeAssistant,
|
|
aioautomower_command: str,
|
|
service: str,
|
|
mock_automower_client: AsyncMock,
|
|
mock_config_entry: MockConfigEntry,
|
|
) -> None:
|
|
"""Test switch commands."""
|
|
await setup_integration(hass, mock_config_entry)
|
|
await hass.services.async_call(
|
|
domain="switch",
|
|
service=service,
|
|
service_data={"entity_id": "switch.test_mower_1_enable_schedule"},
|
|
blocking=True,
|
|
)
|
|
mocked_method = getattr(mock_automower_client, aioautomower_command)
|
|
assert len(mocked_method.mock_calls) == 1
|
|
|
|
mocked_method.side_effect = ApiException("Test error")
|
|
with pytest.raises(HomeAssistantError) as exc_info:
|
|
await hass.services.async_call(
|
|
domain="switch",
|
|
service=service,
|
|
service_data={"entity_id": "switch.test_mower_1_enable_schedule"},
|
|
blocking=True,
|
|
)
|
|
assert (
|
|
str(exc_info.value)
|
|
== "Command couldn't be sent to the command queue: Test error"
|
|
)
|
|
assert len(mocked_method.mock_calls) == 2
|
|
|
|
|
|
async def test_switch(
|
|
hass: HomeAssistant,
|
|
entity_registry: er.EntityRegistry,
|
|
mock_automower_client: AsyncMock,
|
|
mock_config_entry: MockConfigEntry,
|
|
snapshot: SnapshotAssertion,
|
|
) -> None:
|
|
"""Test states of the switch."""
|
|
with patch(
|
|
"homeassistant.components.husqvarna_automower.PLATFORMS",
|
|
[Platform.SWITCH],
|
|
):
|
|
await setup_integration(hass, mock_config_entry)
|
|
entity_entries = er.async_entries_for_config_entry(
|
|
entity_registry, mock_config_entry.entry_id
|
|
)
|
|
|
|
assert entity_entries
|
|
for entity_entry in entity_entries:
|
|
assert hass.states.get(entity_entry.entity_id) == snapshot(
|
|
name=f"{entity_entry.entity_id}-state"
|
|
)
|
|
assert entity_entry == snapshot(name=f"{entity_entry.entity_id}-entry")
|