95 lines
2.9 KiB
Python
95 lines
2.9 KiB
Python
"""Tests for select platform."""
|
|
|
|
from unittest.mock import AsyncMock
|
|
|
|
from aioautomower.exceptions import ApiError
|
|
from aioautomower.model import HeadlightModes, MowerAttributes
|
|
from freezegun.api import FrozenDateTimeFactory
|
|
import pytest
|
|
|
|
from homeassistant.components.husqvarna_automower.coordinator import SCAN_INTERVAL
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.exceptions import HomeAssistantError
|
|
|
|
from . import setup_integration
|
|
from .const import TEST_MOWER_ID
|
|
|
|
from tests.common import MockConfigEntry, async_fire_time_changed
|
|
|
|
|
|
async def test_select_states(
|
|
hass: HomeAssistant,
|
|
mock_automower_client: AsyncMock,
|
|
mock_config_entry: MockConfigEntry,
|
|
freezer: FrozenDateTimeFactory,
|
|
values: dict[str, MowerAttributes],
|
|
) -> None:
|
|
"""Test states of headlight mode select."""
|
|
await setup_integration(hass, mock_config_entry)
|
|
state = hass.states.get("select.test_mower_1_headlight_mode")
|
|
assert state is not None
|
|
assert state.state == "evening_only"
|
|
|
|
for state, expected_state in (
|
|
(
|
|
HeadlightModes.ALWAYS_OFF,
|
|
"always_off",
|
|
),
|
|
(HeadlightModes.ALWAYS_ON, "always_on"),
|
|
(HeadlightModes.EVENING_AND_NIGHT, "evening_and_night"),
|
|
):
|
|
values[TEST_MOWER_ID].settings.headlight.mode = state
|
|
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("select.test_mower_1_headlight_mode")
|
|
assert state.state == expected_state
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("service"),
|
|
[
|
|
("always_on"),
|
|
("always_off"),
|
|
("evening_only"),
|
|
("evening_and_night"),
|
|
],
|
|
)
|
|
async def test_select_commands(
|
|
hass: HomeAssistant,
|
|
service: str,
|
|
mock_automower_client: AsyncMock,
|
|
mock_config_entry: MockConfigEntry,
|
|
) -> None:
|
|
"""Test select commands for headlight mode."""
|
|
await setup_integration(hass, mock_config_entry)
|
|
await hass.services.async_call(
|
|
domain="select",
|
|
service="select_option",
|
|
service_data={
|
|
"entity_id": "select.test_mower_1_headlight_mode",
|
|
"option": service,
|
|
},
|
|
blocking=True,
|
|
)
|
|
mocked_method = mock_automower_client.commands.set_headlight_mode
|
|
mocked_method.assert_called_once_with(TEST_MOWER_ID, service)
|
|
assert len(mocked_method.mock_calls) == 1
|
|
|
|
mocked_method.side_effect = ApiError("Test error")
|
|
with pytest.raises(
|
|
HomeAssistantError,
|
|
match="Failed to send command: Test error",
|
|
):
|
|
await hass.services.async_call(
|
|
domain="select",
|
|
service="select_option",
|
|
service_data={
|
|
"entity_id": "select.test_mower_1_headlight_mode",
|
|
"option": service,
|
|
},
|
|
blocking=True,
|
|
)
|
|
assert len(mocked_method.mock_calls) == 2
|