Add cury_set_mode method (#972)

* Add cury_set_mode method

* Handle "none" value

Author: bieniu

aioshelly/rpc_device/device.py

@@ -508,6 +508,16 @@ async def cury_set_away_mode(
         }
         await self.call_rpc("Cury.SetAwayMode", params=params)
 
+    async def cury_set_mode(
+        self,
+        id_: int,
+        value: str,
+    ) -> None:
+        """Set mode for Cury."""
+        params: dict[str, Any] = {"id": id_}
+        params["mode"] = None if value == "none" else value
+        await self.call_rpc("Cury.SetMode", params=params)
+
     async def enum_set(self, id_: int, value: str) -> None:
         """Set the value for the enum component."""
         params = {

tests/rpc_device/test_device.py

@@ -1457,6 +1457,33 @@ async def test_cury_set_away_mode(
     }
 
 
+@pytest.mark.asyncio
+async def test_cury_set_mode(
+    rpc_device: RpcDevice,
+) -> None:
+    """Test RpcDevice cury_set_mode() method."""
+    await rpc_device.cury_set_mode(9, "living_room")
+
+    assert rpc_device.call_rpc_multiple.call_count == 1
+    call_args_list = rpc_device.call_rpc_multiple.call_args_list
+
+    assert call_args_list[0][0][0][0][0] == "Cury.SetMode"
+    assert call_args_list[0][0][0][0][1] == {
+        "id": 9,
+        "mode": "living_room",
+    }
+
+    await rpc_device.cury_set_mode(9, "none")
+
+    assert rpc_device.call_rpc_multiple.call_count == 2
+    call_args_list = rpc_device.call_rpc_multiple.call_args_list
+
+    assert call_args_list[1][0][0][0][1] == {
+        "id": 9,
+        "mode": None,
+    }
+
+
 @pytest.mark.asyncio
 async def test_rpc_device_init_with_ble(ws_context: WsServer) -> None:
     """Test RpcDevice initialization with BLE device."""