add scenes support

Author: regevbr

roborock/web_api.py

@@ -372,10 +372,7 @@ async def get_scenes(self, user_data: UserData, device_id: int) -> list[HomeData
             raise RoborockException(scenes_response)
         scenes = scenes_response.get("result")
         if isinstance(scenes, list):
-            output_list = []
-            for scene in scenes:
-                output_list.append(HomeDataScene.from_dict(scene))
-            return output_list
+            return [HomeDataScene.from_dict(scene) for scene in scenes]
         else:
             raise RoborockException("home_response result was an unexpected type")
 

tests/conftest.py

@@ -252,10 +252,15 @@ def mock_rest() -> aioresponses:
             },
         )
         mocked.get(
-            re.compile(r"https://api-.*\.roborock\.com/user/scene/device/123456"),
+            re.compile(r"https://api-.*\.roborock\.com/user/scene/device/.*"),
             status=200,
             payload={"api": None, "code": 200, "result": HOME_DATA_SCENES_RAW, "status": "ok", "success": True},
         )
+        mocked.post(
+            re.compile(r"https://api-.*\.roborock\.com/user/scene/.*/execute"),
+            status=200,
+            payload={"api": None, "code": 200, "result": None, "status": "ok", "success": True},
+        )
         yield mocked
 
 

tests/test_web_api.py

@@ -1,3 +1,5 @@
+from yarl import URL
+
 from roborock import HomeData, HomeDataScene, UserData
 from roborock.web_api import RoborockApiClient
 from tests.mock_data import HOME_DATA_RAW, USER_DATA
@@ -51,3 +53,11 @@ async def test_get_scenes():
             }
         )
     ]
+
+
+async def test_execute_scene(mock_rest):
+    """Test that we can execute a scene"""
+    api = RoborockApiClient(username="test_user@gmail.com")
+    ud = await api.pass_login("password")
+    await api.execute_scene(ud, 123456)
+    assert ("post", URL("https://api-us.roborock.com/user/scene/123456/execute")) in mock_rest.requests