ad86eb4be3
So far the Flo integration only supports shutoff valves. Add support for Flo leak detector pucks, which measure temperature and humidity in addition to providing leak alerts.
69 lines
2.1 KiB
Python
69 lines
2.1 KiB
Python
"""Test the services for the Flo by Moen integration."""
|
|
from homeassistant.components.flo.const import DOMAIN as FLO_DOMAIN
|
|
from homeassistant.components.flo.switch import (
|
|
ATTR_REVERT_TO_MODE,
|
|
ATTR_SLEEP_MINUTES,
|
|
SERVICE_RUN_HEALTH_TEST,
|
|
SERVICE_SET_AWAY_MODE,
|
|
SERVICE_SET_HOME_MODE,
|
|
SERVICE_SET_SLEEP_MODE,
|
|
SYSTEM_MODE_HOME,
|
|
)
|
|
from homeassistant.const import ATTR_ENTITY_ID, CONF_PASSWORD, CONF_USERNAME
|
|
from homeassistant.setup import async_setup_component
|
|
|
|
from .common import TEST_PASSWORD, TEST_USER_ID
|
|
|
|
SWITCH_ENTITY_ID = "switch.shutoff_valve"
|
|
|
|
|
|
async def test_services(hass, config_entry, aioclient_mock_fixture, aioclient_mock):
|
|
"""Test Flo services."""
|
|
config_entry.add_to_hass(hass)
|
|
assert await async_setup_component(
|
|
hass, FLO_DOMAIN, {CONF_USERNAME: TEST_USER_ID, CONF_PASSWORD: TEST_PASSWORD}
|
|
)
|
|
await hass.async_block_till_done()
|
|
|
|
assert len(hass.data[FLO_DOMAIN][config_entry.entry_id]["devices"]) == 2
|
|
assert aioclient_mock.call_count == 6
|
|
|
|
await hass.services.async_call(
|
|
FLO_DOMAIN,
|
|
SERVICE_RUN_HEALTH_TEST,
|
|
{ATTR_ENTITY_ID: SWITCH_ENTITY_ID},
|
|
blocking=True,
|
|
)
|
|
await hass.async_block_till_done()
|
|
assert aioclient_mock.call_count == 7
|
|
|
|
await hass.services.async_call(
|
|
FLO_DOMAIN,
|
|
SERVICE_SET_AWAY_MODE,
|
|
{ATTR_ENTITY_ID: SWITCH_ENTITY_ID},
|
|
blocking=True,
|
|
)
|
|
await hass.async_block_till_done()
|
|
assert aioclient_mock.call_count == 8
|
|
|
|
await hass.services.async_call(
|
|
FLO_DOMAIN,
|
|
SERVICE_SET_HOME_MODE,
|
|
{ATTR_ENTITY_ID: SWITCH_ENTITY_ID},
|
|
blocking=True,
|
|
)
|
|
await hass.async_block_till_done()
|
|
assert aioclient_mock.call_count == 9
|
|
|
|
await hass.services.async_call(
|
|
FLO_DOMAIN,
|
|
SERVICE_SET_SLEEP_MODE,
|
|
{
|
|
ATTR_ENTITY_ID: SWITCH_ENTITY_ID,
|
|
ATTR_REVERT_TO_MODE: SYSTEM_MODE_HOME,
|
|
ATTR_SLEEP_MINUTES: 120,
|
|
},
|
|
blocking=True,
|
|
)
|
|
await hass.async_block_till_done()
|
|
assert aioclient_mock.call_count == 10
|