mirror of
https://github.com/home-assistant/core.git
synced 2026-07-03 12:46:09 +01:00
Jonathan Laliberte
55 lines
1.8 KiB
Python
55 lines
1.8 KiB
Python
"""Tests for the Roomba vacuum platform."""
|
|
|
|
from unittest.mock import AsyncMock, patch
|
|
|
|
import pytest
|
|
|
|
from homeassistant.components.vacuum import VacuumActivity
|
|
from homeassistant.const import Platform
|
|
from homeassistant.core import HomeAssistant
|
|
|
|
from tests.common import MockConfigEntry
|
|
|
|
ENTITY_ID = "vacuum.test_roomba"
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("phase", "cycle", "expected"),
|
|
[
|
|
("charge", "none", VacuumActivity.DOCKED),
|
|
# Docked to recharge in the middle of a mission stays docked instead of
|
|
# being reported as paused (regression test for #148287).
|
|
("charge", "clean", VacuumActivity.DOCKED),
|
|
("hmMidMsn", "clean", VacuumActivity.CLEANING),
|
|
("hmPostMsn", "clean", VacuumActivity.RETURNING),
|
|
("run", "clean", VacuumActivity.CLEANING),
|
|
("pause", "clean", VacuumActivity.PAUSED),
|
|
# Stopped on the floor mid-mission is a paused state.
|
|
("stop", "clean", VacuumActivity.PAUSED),
|
|
("stop", "none", VacuumActivity.IDLE),
|
|
("stuck", "clean", VacuumActivity.ERROR),
|
|
],
|
|
)
|
|
async def test_vacuum_activity(
|
|
hass: HomeAssistant,
|
|
mock_config_entry: MockConfigEntry,
|
|
mock_roomba: AsyncMock,
|
|
phase: str,
|
|
cycle: str,
|
|
expected: VacuumActivity,
|
|
) -> None:
|
|
"""Test the vacuum activity mapping from the reported mission status."""
|
|
mock_roomba.master_state["state"]["reported"]["cleanMissionStatus"] = {
|
|
"cycle": cycle,
|
|
"phase": phase,
|
|
}
|
|
|
|
with patch("homeassistant.components.roomba.PLATFORMS", [Platform.VACUUM]):
|
|
mock_config_entry.add_to_hass(hass)
|
|
await hass.config_entries.async_setup(mock_config_entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
state = hass.states.get(ENTITY_ID)
|
|
assert state is not None
|
|
assert state.state == expected
|