Add enum_set and text_set methods (#851)

Author: bieniu

aioshelly/rpc_device/device.py

@@ -333,6 +333,14 @@ async def blu_trv_clear_boost(self, trv_id: int) -> None:
         }
         await self.call_rpc("BluTRV.Call", params=params, timeout=BLU_TRV_TIMEOUT)
 
+    async def enum_set(self, id_: int, value: str) -> None:
+        """Set the value for the enum component."""
+        params = {
+            "id": id_,
+            "value": value,
+        }
+        await self.call_rpc("Enum.Set", params=params)
+
     async def number_set(self, id_: int, value: float) -> None:
         """Set the value for the number component."""
         params = {
@@ -341,6 +349,14 @@ async def number_set(self, id_: int, value: float) -> None:
         }
         await self.call_rpc("Number.Set", params=params)
 
+    async def text_set(self, id_: int, value: str) -> None:
+        """Set the value for the text component."""
+        params = {
+            "id": id_,
+            "value": value,
+        }
+        await self.call_rpc("Text.Set", params=params)
+
     async def update_status(self) -> None:
         """Get device status from 'Shelly.GetStatus'."""
         self._status = await self.call_rpc("Shelly.GetStatus")

tests/rpc_device/test_device.py

@@ -978,3 +978,35 @@ async def test_number_set(
         "id": 12,
         "value": 33.2,
     }
+
+
+@pytest.mark.asyncio
+async def test_enum_set(
+    rpc_device: RpcDevice,
+) -> None:
+    """Test RpcDevice enum_set() method."""
+    await rpc_device.enum_set(12, "option 1")
+
+    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] == "Enum.Set"
+    assert call_args_list[0][0][0][0][1] == {
+        "id": 12,
+        "value": "option 1",
+    }
+
+
+@pytest.mark.asyncio
+async def test_text_set(
+    rpc_device: RpcDevice,
+) -> None:
+    """Test RpcDevice text_set() method."""
+    await rpc_device.text_set(12, "lorem ipsum")
+
+    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] == "Text.Set"
+    assert call_args_list[0][0][0][0][1] == {
+        "id": 12,
+        "value": "lorem ipsum",
+    }