Litterrobot tests

home-assistant · Oct 13, 2024 · 179c0c0 · 179c0c0
1 parent bd7cc34
commit 179c0c0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/tests/components/litterrobot/test_init.py b/tests/components/litterrobot/test_init.py
@@ -9,7 +9,7 @@
 from homeassistant.components.vacuum import (
     DOMAIN as VACUUM_DOMAIN,
     SERVICE_START,
-    STATE_DOCKED,
+    VacuumEntityState,
 )
 from homeassistant.config_entries import ConfigEntryState
 from homeassistant.const import ATTR_ENTITY_ID
@@ -30,7 +30,7 @@ async def test_unload_entry(hass: HomeAssistant, mock_account: MagicMock) -> Non
 
     vacuum = hass.states.get(VACUUM_ENTITY_ID)
     assert vacuum
-    assert vacuum.state == STATE_DOCKED
+    assert vacuum.state == VacuumEntityState.DOCKED
 
     await hass.services.async_call(
         VACUUM_DOMAIN,

diff --git a/tests/components/litterrobot/test_vacuum.py b/tests/components/litterrobot/test_vacuum.py
@@ -102,7 +102,7 @@ async def test_vacuum_with_error(
         ({"displayCode": "DC_CAT_DETECT"}, VacuumEntityState.DOCKED),
         ({"isDFIFull": True}, VacuumEntityState.ERROR),
         (
-            {"robotCycleState": "CYCLE_VacuumEntityState.CAT_DETECT"},
+            {"robotCycleState": "CYCLE_STATE_CAT_DETECT"},
             VacuumEntityState.PAUSED,
         ),
     ],