96 lines
2.6 KiB
Python
96 lines
2.6 KiB
Python
"""Test the Litter-Robot vacuum entity."""
|
|
from datetime import timedelta
|
|
|
|
import pytest
|
|
|
|
from homeassistant.components.litterrobot.hub import REFRESH_WAIT_TIME
|
|
from homeassistant.components.vacuum import (
|
|
ATTR_PARAMS,
|
|
DOMAIN as PLATFORM_DOMAIN,
|
|
SERVICE_SEND_COMMAND,
|
|
SERVICE_START,
|
|
SERVICE_TURN_OFF,
|
|
SERVICE_TURN_ON,
|
|
STATE_DOCKED,
|
|
STATE_ERROR,
|
|
)
|
|
from homeassistant.const import ATTR_COMMAND, ATTR_ENTITY_ID
|
|
from homeassistant.util.dt import utcnow
|
|
|
|
from .conftest import setup_integration
|
|
|
|
from tests.common import async_fire_time_changed
|
|
|
|
ENTITY_ID = "vacuum.test_litter_box"
|
|
|
|
|
|
async def test_vacuum(hass, mock_account):
|
|
"""Tests the vacuum entity was set up."""
|
|
await setup_integration(hass, mock_account, PLATFORM_DOMAIN)
|
|
|
|
vacuum = hass.states.get(ENTITY_ID)
|
|
assert vacuum
|
|
assert vacuum.state == STATE_DOCKED
|
|
assert vacuum.attributes["is_sleeping"] is False
|
|
|
|
|
|
async def test_vacuum_with_error(hass, mock_account_with_error):
|
|
"""Tests a vacuum entity with an error."""
|
|
await setup_integration(hass, mock_account_with_error, PLATFORM_DOMAIN)
|
|
|
|
vacuum = hass.states.get(ENTITY_ID)
|
|
assert vacuum
|
|
assert vacuum.state == STATE_ERROR
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"service,command,extra",
|
|
[
|
|
(SERVICE_START, "start_cleaning", None),
|
|
(SERVICE_TURN_OFF, "set_power_status", None),
|
|
(SERVICE_TURN_ON, "set_power_status", None),
|
|
(
|
|
SERVICE_SEND_COMMAND,
|
|
"reset_waste_drawer",
|
|
{ATTR_COMMAND: "reset_waste_drawer"},
|
|
),
|
|
(
|
|
SERVICE_SEND_COMMAND,
|
|
"set_sleep_mode",
|
|
{
|
|
ATTR_COMMAND: "set_sleep_mode",
|
|
ATTR_PARAMS: {"enabled": True, "sleep_time": "22:30"},
|
|
},
|
|
),
|
|
(
|
|
SERVICE_SEND_COMMAND,
|
|
"set_sleep_mode",
|
|
{
|
|
ATTR_COMMAND: "set_sleep_mode",
|
|
ATTR_PARAMS: {"enabled": True, "sleep_time": None},
|
|
},
|
|
),
|
|
],
|
|
)
|
|
async def test_commands(hass, mock_account, service, command, extra):
|
|
"""Test sending commands to the vacuum."""
|
|
await setup_integration(hass, mock_account, PLATFORM_DOMAIN)
|
|
|
|
vacuum = hass.states.get(ENTITY_ID)
|
|
assert vacuum
|
|
assert vacuum.state == STATE_DOCKED
|
|
|
|
data = {ATTR_ENTITY_ID: ENTITY_ID}
|
|
if extra:
|
|
data.update(extra)
|
|
|
|
await hass.services.async_call(
|
|
PLATFORM_DOMAIN,
|
|
service,
|
|
data,
|
|
blocking=True,
|
|
)
|
|
future = utcnow() + timedelta(seconds=REFRESH_WAIT_TIME)
|
|
async_fire_time_changed(hass, future)
|
|
getattr(mock_account.robots[0], command).assert_called_once()
|