add scenes support

Author: regevbr

roborock/web_api.py

@@ -269,7 +269,7 @@ async def get_rooms(self, user_data: UserData, home_id: int | None = None) -> li
         else:
             raise RoborockException("home_response result was an unexpected type")
 
-    async def get_scenes(self, user_data: UserData, home_id: int) -> list[HomeDataRoom]:
+    async def get_scenes(self, user_data: UserData, home_id: int) -> list[HomeDataScene]:
         rriot = user_data.rriot
         if rriot is None:
             raise RoborockException("rriot is none")

tests/conftest.py

@@ -15,7 +15,7 @@
 from roborock.containers import DeviceData
 from roborock.version_1_apis.roborock_local_client_v1 import RoborockLocalClientV1
 from roborock.version_1_apis.roborock_mqtt_client_v1 import RoborockMqttClientV1
-from tests.mock_data import HOME_DATA_RAW, TEST_LOCAL_API_HOST, USER_DATA
+from tests.mock_data import HOME_DATA_RAW, HOME_DATA_SCENES_RAW, TEST_LOCAL_API_HOST, USER_DATA
 
 _LOGGER = logging.getLogger(__name__)
 
@@ -200,6 +200,11 @@ def mock_rest() -> aioresponses:
             status=200,
             payload={"api": None, "code": 200, "result": HOME_DATA_RAW, "status": "ok", "success": True},
         )
+        mocked.get(
+            re.compile(r"https://api-.*\.roborock\.com/user/scene/device/123456"),
+            status=200,
+            payload={"api": None, "code": 200, "result": HOME_DATA_SCENES_RAW, "status": "ok", "success": True},
+        )
         yield mocked
 
 

tests/mock_data.py

@@ -35,6 +35,16 @@
 }
 LOCAL_KEY = "key123key123key1"  # 16 bytes / 128 bits
 PRODUCT_ID = "product-id-123"
+HOME_DATA_SCENES_RAW = [
+    {
+        "id": 5925425,
+        "name": "My plan",
+        "param": "not used",
+        "enabled": True,
+        "extra": None,
+        "type": "WORKFLOW",
+    }
+]
 HOME_DATA_RAW = {
     "id": 123456,
     "name": "My Home",

tests/test_web_api.py

@@ -1,4 +1,4 @@
-from roborock import HomeData, UserData
+from roborock import HomeData, HomeDataScene, UserData
 from roborock.web_api import RoborockApiClient
 from tests.mock_data import HOME_DATA_RAW, USER_DATA
 
@@ -26,3 +26,18 @@ async def test_get_home_data_v2():
     ud = await api.code_login(4123)
     hd = await api.get_home_data_v2(ud)
     assert hd == HomeData.from_dict(HOME_DATA_RAW)
+
+
+async def test_get_scenes():
+    """Test that we can get scenes"""
+    api = RoborockApiClient(username="[email protected]")
+    ud = await api.pass_login("password")
+    sc = await api.get_scenes(ud, 123456)
+    assert sc == [
+        HomeDataScene.from_dict(
+            {
+                "id": 5925425,
+                "name": "My plan",
+            }
+        )
+    ]