mirror of
https://github.com/home-assistant/core.git
synced 2026-02-15 07:36:16 +00:00
97 lines
3.1 KiB
Python
97 lines
3.1 KiB
Python
"""Test the Whirlpool select domain."""
|
|
|
|
from unittest.mock import MagicMock
|
|
|
|
import pytest
|
|
from syrupy.assertion import SnapshotAssertion
|
|
|
|
from homeassistant.components.select import ATTR_OPTION, DOMAIN as SELECT_DOMAIN
|
|
from homeassistant.const import ATTR_ENTITY_ID, SERVICE_SELECT_OPTION, Platform
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.exceptions import ServiceValidationError
|
|
from homeassistant.helpers import entity_registry as er
|
|
|
|
from . import init_integration, snapshot_whirlpool_entities, trigger_attr_callback
|
|
|
|
|
|
async def test_all_entities(
|
|
hass: HomeAssistant, snapshot: SnapshotAssertion, entity_registry: er.EntityRegistry
|
|
) -> None:
|
|
"""Test all entities."""
|
|
await init_integration(hass)
|
|
snapshot_whirlpool_entities(hass, entity_registry, snapshot, Platform.SELECT)
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
(
|
|
"entity_id",
|
|
"mock_fixture",
|
|
"mock_getter_method_name",
|
|
"mock_setter_method_name",
|
|
"values",
|
|
),
|
|
[
|
|
(
|
|
"select.beer_fridge_temperature",
|
|
"mock_refrigerator_api",
|
|
"get_offset_temp",
|
|
"set_offset_temp",
|
|
[(-4, "-4"), (-2, "-2"), (0, "0"), (3, "3"), (5, "5")],
|
|
),
|
|
],
|
|
)
|
|
async def test_select_entities(
|
|
hass: HomeAssistant,
|
|
entity_id: str,
|
|
mock_fixture: str,
|
|
mock_getter_method_name: str,
|
|
mock_setter_method_name: str,
|
|
values: list[tuple[int, str]],
|
|
request: pytest.FixtureRequest,
|
|
) -> None:
|
|
"""Test reading and setting select options."""
|
|
await init_integration(hass)
|
|
mock_instance = request.getfixturevalue(mock_fixture)
|
|
|
|
# Test reading current option
|
|
mock_getter_method = getattr(mock_instance, mock_getter_method_name)
|
|
for raw_value, expected_state in values:
|
|
mock_getter_method.return_value = raw_value
|
|
|
|
await trigger_attr_callback(hass, mock_instance)
|
|
state = hass.states.get(entity_id)
|
|
assert state is not None
|
|
assert state.state == expected_state
|
|
|
|
# Test changing option
|
|
mock_setter_method = getattr(mock_instance, mock_setter_method_name)
|
|
for raw_value, selected_option in values:
|
|
await hass.services.async_call(
|
|
SELECT_DOMAIN,
|
|
SERVICE_SELECT_OPTION,
|
|
{ATTR_ENTITY_ID: entity_id, ATTR_OPTION: selected_option},
|
|
blocking=True,
|
|
)
|
|
assert mock_setter_method.call_count == 1
|
|
mock_setter_method.assert_called_with(raw_value)
|
|
mock_setter_method.reset_mock()
|
|
|
|
|
|
async def test_select_option_value_error(
|
|
hass: HomeAssistant, mock_refrigerator_api: MagicMock
|
|
) -> None:
|
|
"""Test handling of ValueError exception when selecting an option."""
|
|
await init_integration(hass)
|
|
mock_refrigerator_api.set_offset_temp.side_effect = ValueError
|
|
|
|
with pytest.raises(ServiceValidationError):
|
|
await hass.services.async_call(
|
|
SELECT_DOMAIN,
|
|
SERVICE_SELECT_OPTION,
|
|
{
|
|
ATTR_ENTITY_ID: "select.beer_fridge_temperature",
|
|
ATTR_OPTION: "something",
|
|
},
|
|
blocking=True,
|
|
)
|