mirror of
https://github.com/home-assistant/core.git
synced 2026-07-14 18:14:35 +01:00
Justin Mutter
104 lines
2.8 KiB
Python
104 lines
2.8 KiB
Python
"""Tests for the AquaLogic switch platform."""
|
|
|
|
from unittest.mock import MagicMock
|
|
|
|
from aqualogic.core import States
|
|
import pytest
|
|
from syrupy.assertion import SnapshotAssertion
|
|
|
|
from homeassistant.components.aqualogic.const import UPDATE_TOPIC
|
|
from homeassistant.components.switch import DOMAIN as SWITCH_DOMAIN
|
|
from homeassistant.const import (
|
|
ATTR_ENTITY_ID,
|
|
SERVICE_TURN_OFF,
|
|
SERVICE_TURN_ON,
|
|
STATE_OFF,
|
|
Platform,
|
|
)
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers.dispatcher import async_dispatcher_send
|
|
|
|
|
|
@pytest.fixture
|
|
def platforms() -> list[Platform]:
|
|
"""Fixture to specify platforms to test."""
|
|
return [Platform.SWITCH]
|
|
|
|
|
|
@pytest.mark.usefixtures("init_integration")
|
|
async def test_switches(
|
|
hass: HomeAssistant,
|
|
snapshot: SnapshotAssertion,
|
|
) -> None:
|
|
"""Test switch entities are created and report correct state."""
|
|
async_dispatcher_send(hass, UPDATE_TOPIC)
|
|
await hass.async_block_till_done()
|
|
|
|
states = {
|
|
state.entity_id: state
|
|
for state in sorted(hass.states.async_all("switch"), key=lambda s: s.entity_id)
|
|
}
|
|
assert states == snapshot
|
|
|
|
|
|
@pytest.mark.usefixtures("init_integration")
|
|
@pytest.mark.parametrize(
|
|
("service", "expected_state"),
|
|
[
|
|
pytest.param(SERVICE_TURN_ON, True, id="turn_on"),
|
|
pytest.param(SERVICE_TURN_OFF, False, id="turn_off"),
|
|
],
|
|
)
|
|
async def test_turn(
|
|
hass: HomeAssistant,
|
|
mock_panel: MagicMock,
|
|
service: str,
|
|
expected_state: bool,
|
|
) -> None:
|
|
"""Test turning a switch on/off calls set_state on the panel."""
|
|
await hass.services.async_call(
|
|
SWITCH_DOMAIN,
|
|
service,
|
|
{ATTR_ENTITY_ID: "switch.aqualogic_lights"},
|
|
blocking=True,
|
|
)
|
|
mock_panel.set_state.assert_called_once_with(States.LIGHTS, expected_state)
|
|
|
|
|
|
@pytest.mark.usefixtures("init_integration")
|
|
@pytest.mark.parametrize(
|
|
"service",
|
|
[
|
|
pytest.param(SERVICE_TURN_ON, id="turn_on"),
|
|
pytest.param(SERVICE_TURN_OFF, id="turn_off"),
|
|
],
|
|
)
|
|
async def test_turn_no_panel(
|
|
hass: HomeAssistant,
|
|
mock_processor: MagicMock,
|
|
service: str,
|
|
) -> None:
|
|
"""Test that turning a switch does nothing when the panel is unavailable."""
|
|
panel = mock_processor.panel
|
|
mock_processor.panel = None
|
|
await hass.services.async_call(
|
|
SWITCH_DOMAIN,
|
|
service,
|
|
{ATTR_ENTITY_ID: "switch.aqualogic_lights"},
|
|
blocking=True,
|
|
)
|
|
panel.set_state.assert_not_called()
|
|
|
|
|
|
@pytest.mark.usefixtures("init_integration")
|
|
async def test_is_on_no_panel(
|
|
hass: HomeAssistant,
|
|
mock_processor: MagicMock,
|
|
) -> None:
|
|
"""Test switch reports off when panel is unavailable."""
|
|
mock_processor.panel = None
|
|
async_dispatcher_send(hass, UPDATE_TOPIC)
|
|
await hass.async_block_till_done()
|
|
|
|
assert hass.states.get("switch.aqualogic_lights").state == STATE_OFF
|