302 lines
8.2 KiB
Python
302 lines
8.2 KiB
Python
"""Tests for rainbird sensor platform."""
|
|
|
|
|
|
from http import HTTPStatus
|
|
import logging
|
|
|
|
import pytest
|
|
|
|
from homeassistant.components.rainbird import DOMAIN
|
|
from homeassistant.const import ATTR_ENTITY_ID, Platform
|
|
from homeassistant.core import HomeAssistant
|
|
|
|
from .conftest import (
|
|
ACK_ECHO,
|
|
AVAILABLE_STATIONS_RESPONSE,
|
|
EMPTY_STATIONS_RESPONSE,
|
|
HOST,
|
|
PASSWORD,
|
|
URL,
|
|
ZONE_3_ON_RESPONSE,
|
|
ZONE_5_ON_RESPONSE,
|
|
ZONE_OFF_RESPONSE,
|
|
ComponentSetup,
|
|
mock_response,
|
|
)
|
|
|
|
from tests.components.switch import common as switch_common
|
|
from tests.test_util.aiohttp import AiohttpClientMocker, AiohttpClientMockResponse
|
|
|
|
|
|
@pytest.fixture
|
|
def platforms() -> list[str]:
|
|
"""Fixture to specify platforms to test."""
|
|
return [Platform.SWITCH]
|
|
|
|
|
|
async def test_no_zones(
|
|
hass: HomeAssistant,
|
|
setup_integration: ComponentSetup,
|
|
responses: list[AiohttpClientMockResponse],
|
|
) -> None:
|
|
"""Test case where listing stations returns no stations."""
|
|
|
|
responses.append(mock_response(EMPTY_STATIONS_RESPONSE))
|
|
assert await setup_integration()
|
|
|
|
zone = hass.states.get("switch.sprinkler_1")
|
|
assert zone is None
|
|
|
|
|
|
async def test_zones(
|
|
hass: HomeAssistant,
|
|
setup_integration: ComponentSetup,
|
|
responses: list[AiohttpClientMockResponse],
|
|
) -> None:
|
|
"""Test switch platform with fake data that creates 7 zones with one enabled."""
|
|
|
|
responses.extend(
|
|
[mock_response(AVAILABLE_STATIONS_RESPONSE), mock_response(ZONE_5_ON_RESPONSE)]
|
|
)
|
|
|
|
assert await setup_integration()
|
|
|
|
zone = hass.states.get("switch.sprinkler_1")
|
|
assert zone is not None
|
|
assert zone.state == "off"
|
|
|
|
zone = hass.states.get("switch.sprinkler_2")
|
|
assert zone is not None
|
|
assert zone.state == "off"
|
|
|
|
zone = hass.states.get("switch.sprinkler_3")
|
|
assert zone is not None
|
|
assert zone.state == "off"
|
|
|
|
zone = hass.states.get("switch.sprinkler_4")
|
|
assert zone is not None
|
|
assert zone.state == "off"
|
|
|
|
zone = hass.states.get("switch.sprinkler_5")
|
|
assert zone is not None
|
|
assert zone.state == "on"
|
|
|
|
zone = hass.states.get("switch.sprinkler_6")
|
|
assert zone is not None
|
|
assert zone.state == "off"
|
|
|
|
zone = hass.states.get("switch.sprinkler_7")
|
|
assert zone is not None
|
|
assert zone.state == "off"
|
|
|
|
assert not hass.states.get("switch.sprinkler_8")
|
|
|
|
|
|
async def test_switch_on(
|
|
hass: HomeAssistant,
|
|
setup_integration: ComponentSetup,
|
|
aioclient_mock: AiohttpClientMocker,
|
|
responses: list[AiohttpClientMockResponse],
|
|
) -> None:
|
|
"""Test turning on irrigation switch."""
|
|
|
|
responses.extend(
|
|
[mock_response(AVAILABLE_STATIONS_RESPONSE), mock_response(ZONE_OFF_RESPONSE)]
|
|
)
|
|
assert await setup_integration()
|
|
|
|
# Initially all zones are off. Pick zone3 as an arbitrary to assert
|
|
# state, then update below as a switch.
|
|
zone = hass.states.get("switch.sprinkler_3")
|
|
assert zone is not None
|
|
assert zone.state == "off"
|
|
|
|
aioclient_mock.mock_calls.clear()
|
|
responses.extend(
|
|
[
|
|
mock_response(ACK_ECHO), # Switch on response
|
|
mock_response(ZONE_3_ON_RESPONSE), # Updated zone state
|
|
]
|
|
)
|
|
await switch_common.async_turn_on(hass, "switch.sprinkler_3")
|
|
await hass.async_block_till_done()
|
|
assert len(aioclient_mock.mock_calls) == 2
|
|
aioclient_mock.mock_calls.clear()
|
|
|
|
# Verify switch state is updated
|
|
zone = hass.states.get("switch.sprinkler_3")
|
|
assert zone is not None
|
|
assert zone.state == "on"
|
|
|
|
|
|
async def test_switch_off(
|
|
hass: HomeAssistant,
|
|
setup_integration: ComponentSetup,
|
|
aioclient_mock: AiohttpClientMocker,
|
|
responses: list[AiohttpClientMockResponse],
|
|
) -> None:
|
|
"""Test turning off irrigation switch."""
|
|
|
|
responses.extend(
|
|
[mock_response(AVAILABLE_STATIONS_RESPONSE), mock_response(ZONE_3_ON_RESPONSE)]
|
|
)
|
|
assert await setup_integration()
|
|
|
|
# Initially the test zone is on
|
|
zone = hass.states.get("switch.sprinkler_3")
|
|
assert zone is not None
|
|
assert zone.state == "on"
|
|
|
|
aioclient_mock.mock_calls.clear()
|
|
responses.extend(
|
|
[
|
|
mock_response(ACK_ECHO), # Switch off response
|
|
mock_response(ZONE_OFF_RESPONSE), # Updated zone state
|
|
]
|
|
)
|
|
await switch_common.async_turn_off(hass, "switch.sprinkler_3")
|
|
await hass.async_block_till_done()
|
|
|
|
# One call to change the service and one to refresh state
|
|
assert len(aioclient_mock.mock_calls) == 2
|
|
|
|
# Verify switch state is updated
|
|
zone = hass.states.get("switch.sprinkler_3")
|
|
assert zone is not None
|
|
assert zone.state == "off"
|
|
|
|
|
|
async def test_irrigation_service(
|
|
hass: HomeAssistant,
|
|
setup_integration: ComponentSetup,
|
|
aioclient_mock: AiohttpClientMocker,
|
|
responses: list[AiohttpClientMockResponse],
|
|
) -> None:
|
|
"""Test calling the irrigation service."""
|
|
|
|
responses.extend(
|
|
[mock_response(AVAILABLE_STATIONS_RESPONSE), mock_response(ZONE_3_ON_RESPONSE)]
|
|
)
|
|
assert await setup_integration()
|
|
|
|
aioclient_mock.mock_calls.clear()
|
|
responses.extend([mock_response(ACK_ECHO), mock_response(ZONE_OFF_RESPONSE)])
|
|
|
|
await hass.services.async_call(
|
|
DOMAIN,
|
|
"start_irrigation",
|
|
{ATTR_ENTITY_ID: "switch.sprinkler_5", "duration": 30},
|
|
blocking=True,
|
|
)
|
|
|
|
# One call to change the service and one to refresh state
|
|
assert len(aioclient_mock.mock_calls) == 2
|
|
|
|
|
|
async def test_rain_delay_service(
|
|
hass: HomeAssistant,
|
|
setup_integration: ComponentSetup,
|
|
aioclient_mock: AiohttpClientMocker,
|
|
responses: list[AiohttpClientMockResponse],
|
|
) -> None:
|
|
"""Test calling the rain delay service."""
|
|
|
|
responses.extend(
|
|
[mock_response(AVAILABLE_STATIONS_RESPONSE), mock_response(ZONE_3_ON_RESPONSE)]
|
|
)
|
|
assert await setup_integration()
|
|
|
|
aioclient_mock.mock_calls.clear()
|
|
responses.extend(
|
|
[
|
|
mock_response(ACK_ECHO),
|
|
]
|
|
)
|
|
|
|
await hass.services.async_call(
|
|
DOMAIN, "set_rain_delay", {"duration": 30}, blocking=True
|
|
)
|
|
|
|
assert len(aioclient_mock.mock_calls) == 1
|
|
|
|
|
|
async def test_platform_unavailable(
|
|
hass: HomeAssistant,
|
|
setup_integration: ComponentSetup,
|
|
responses: list[AiohttpClientMockResponse],
|
|
caplog: pytest.LogCaptureFixture,
|
|
) -> None:
|
|
"""Test failure while listing the stations when setting up the platform."""
|
|
|
|
responses.append(
|
|
AiohttpClientMockResponse("POST", URL, status=HTTPStatus.SERVICE_UNAVAILABLE)
|
|
)
|
|
|
|
with caplog.at_level(logging.WARNING):
|
|
assert await setup_integration()
|
|
|
|
assert "Failed to get stations" in caplog.text
|
|
|
|
|
|
async def test_coordinator_unavailable(
|
|
hass: HomeAssistant,
|
|
setup_integration: ComponentSetup,
|
|
responses: list[AiohttpClientMockResponse],
|
|
caplog: pytest.LogCaptureFixture,
|
|
) -> None:
|
|
"""Test failure to refresh the update coordinator."""
|
|
|
|
responses.extend(
|
|
[
|
|
mock_response(AVAILABLE_STATIONS_RESPONSE),
|
|
AiohttpClientMockResponse(
|
|
"POST", URL, status=HTTPStatus.SERVICE_UNAVAILABLE
|
|
),
|
|
],
|
|
)
|
|
|
|
with caplog.at_level(logging.WARNING):
|
|
assert await setup_integration()
|
|
|
|
assert "Failed to load zone state" in caplog.text
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"yaml_config",
|
|
[
|
|
{
|
|
DOMAIN: {
|
|
"host": HOST,
|
|
"password": PASSWORD,
|
|
"trigger_time": 360,
|
|
"zones": {
|
|
1: {
|
|
"friendly_name": "Garden Sprinkler",
|
|
},
|
|
2: {
|
|
"friendly_name": "Back Yard",
|
|
},
|
|
},
|
|
}
|
|
},
|
|
],
|
|
)
|
|
async def test_yaml_config(
|
|
hass: HomeAssistant,
|
|
setup_integration: ComponentSetup,
|
|
responses: list[AiohttpClientMockResponse],
|
|
) -> None:
|
|
"""Test switch platform with fake data that creates 7 zones with one enabled."""
|
|
|
|
responses.extend(
|
|
[mock_response(AVAILABLE_STATIONS_RESPONSE), mock_response(ZONE_5_ON_RESPONSE)]
|
|
)
|
|
|
|
assert await setup_integration()
|
|
|
|
assert hass.states.get("switch.garden_sprinkler")
|
|
assert not hass.states.get("switch.sprinkler_1")
|
|
assert hass.states.get("switch.back_yard")
|
|
assert not hass.states.get("switch.sprinkler_2")
|
|
assert hass.states.get("switch.sprinkler_3")
|