Scripted demo-stand vacuum: the checklist needs a robot no hardware can provide

The public stand cannot run any Tier-A map integration (no radios, no robots), which left the whole vacuum section of docs/TESTING.md untestable outside the owner's home. demo_robot fakes exactly the surface the card consumes: a Tasshack-shaped map sensor (dict vacuum_position, rooms named after the plan rooms so auto-calibration has something to match, flipped Y so the mirror default is meaningful) and a vacuum that drives a serpentine route through every room in ~3.5 minutes and docks itself.
2026-08-01 00:48:29 +00:00 · 2026-07-31 12:48:53 +03:00
parent 9870b2fb98
commit 18b8589438
5 changed files with 325 additions and 0 deletions
@@ -0,0 +1,65 @@
+"""The vacuum entity of the stand simulator."""
+from __future__ import annotations
+
+from typing import Any
+
+from homeassistant.components.vacuum import (
+    StateVacuumEntity,
+    VacuumActivity,
+    VacuumEntityFeature,
+)
+from homeassistant.core import HomeAssistant
+
+from . import DOMAIN, RobotSim
+
+ACTIVITY = {
+    "docked": VacuumActivity.DOCKED,
+    "cleaning": VacuumActivity.CLEANING,
+    "paused": VacuumActivity.PAUSED,
+    "returning": VacuumActivity.RETURNING,
+}
+
+
+async def async_setup_platform(hass: HomeAssistant, config, add_entities, discovery_info=None) -> None:
+    add_entities([DemoRobotVacuum(hass.data[DOMAIN])])
+
+
+class DemoRobotVacuum(StateVacuumEntity):
+    _attr_should_poll = False
+    _attr_name = "Робот-пылесос"
+    _attr_unique_id = "demo_robot_vacuum"
+    _attr_supported_features = (
+        VacuumEntityFeature.START
+        | VacuumEntityFeature.STOP
+        | VacuumEntityFeature.PAUSE
+        | VacuumEntityFeature.RETURN_HOME
+        | VacuumEntityFeature.STATE
+    )
+
+    def __init__(self, sim: RobotSim) -> None:
+        self._sim = sim
+        self.entity_id = "vacuum.demo_robot"
+
+    async def async_added_to_hass(self) -> None:
+        self.async_on_remove(self._sim.add_listener(self.async_write_ha_state))
+
+    @property
+    def activity(self) -> VacuumActivity:
+        return ACTIVITY.get(self._sim.state, VacuumActivity.IDLE)
+
+    @property
+    def extra_state_attributes(self) -> dict[str, Any]:
+        # the card resolves the active map name through the vacuum entity
+        return {"selected_map": "1"}
+
+    async def async_start(self) -> None:
+        self._sim.start()
+
+    async def async_pause(self) -> None:
+        self._sim.pause()
+
+    async def async_stop(self, **kwargs: Any) -> None:
+        self._sim.stop()
+
+    async def async_return_to_base(self, **kwargs: Any) -> None:
+        self._sim.return_to_base()