93 lines
2.6 KiB
Python
93 lines
2.6 KiB
Python
"""Test the Litter-Robot vacuum entity."""
|
|
from datetime import timedelta
|
|
from unittest.mock import patch
|
|
|
|
import pytest
|
|
|
|
from homeassistant.components import litterrobot
|
|
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,
|
|
)
|
|
from homeassistant.const import ATTR_COMMAND, ATTR_ENTITY_ID
|
|
from homeassistant.util.dt import utcnow
|
|
|
|
from .common import CONFIG
|
|
|
|
from tests.common import MockConfigEntry, async_fire_time_changed
|
|
|
|
ENTITY_ID = "vacuum.test_litter_box"
|
|
|
|
|
|
async def setup_hub(hass, mock_hub):
|
|
"""Load the Litter-Robot vacuum platform with the provided hub."""
|
|
hass.config.components.add(litterrobot.DOMAIN)
|
|
entry = MockConfigEntry(
|
|
domain=litterrobot.DOMAIN,
|
|
data=CONFIG[litterrobot.DOMAIN],
|
|
)
|
|
|
|
with patch.dict(hass.data, {litterrobot.DOMAIN: {entry.entry_id: mock_hub}}):
|
|
await hass.config_entries.async_forward_entry_setup(entry, PLATFORM_DOMAIN)
|
|
await hass.async_block_till_done()
|
|
|
|
|
|
async def test_vacuum(hass, mock_hub):
|
|
"""Tests the vacuum entity was set up."""
|
|
await setup_hub(hass, mock_hub)
|
|
|
|
vacuum = hass.states.get(ENTITY_ID)
|
|
assert vacuum is not None
|
|
assert vacuum.state == STATE_DOCKED
|
|
assert vacuum.attributes["is_sleeping"] is False
|
|
|
|
|
|
@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"},
|
|
},
|
|
),
|
|
],
|
|
)
|
|
async def test_commands(hass, mock_hub, service, command, extra):
|
|
"""Test sending commands to the vacuum."""
|
|
await setup_hub(hass, mock_hub)
|
|
|
|
vacuum = hass.states.get(ENTITY_ID)
|
|
assert vacuum is not None
|
|
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_hub.account.robots[0], command).assert_called_once()
|