Add methods for Blu Trv boost mode (#839)

* Add methods for Blu Trv boost mode

* Typo

* Cleaning after rebase

Author: bieniu

aioshelly/rpc_device/device.py

@@ -302,6 +302,26 @@ async def blu_trv_set_valve_position(self, trv_id: int, position: float) -> None
         }
         await self.call_rpc("BluTRV.Call", params=params, timeout=BLU_TRV_TIMEOUT)
 
+    async def blu_trv_set_boost(self, trv_id: int, duration: int | None = None) -> None:
+        """Start boost mode for BLU TRV."""
+        params = {
+            "id": trv_id,
+            "method": "Trv.SetBoost",
+        }
+        params["params"] = (
+            {"id": 0} if duration is None else {"id": 0, "duration": duration}
+        )
+        await self.call_rpc("BluTRV.Call", params=params, timeout=BLU_TRV_TIMEOUT)
+
+    async def blu_trv_clear_boost(self, trv_id: int) -> None:
+        """Clear boost mode for BLU TRV."""
+        params = {
+            "id": trv_id,
+            "method": "Trv.ClearBoost",
+            "params": {"id": 0},
+        }
+        await self.call_rpc("BluTRV.Call", params=params, timeout=BLU_TRV_TIMEOUT)
+
     async def number_set(self, id_: int, value: float) -> None:
         """Set the value for the number component."""
         params = {

tests/rpc_device/test_device.py

@@ -892,6 +892,60 @@ async def test_blu_trv_set_valve_position(
     assert call_args_list[0][0][1] == 60
 
 
+@pytest.mark.asyncio
+async def test_blu_trv_set_boost(
+    rpc_device: RpcDevice,
+) -> None:
+    """Test RpcDevice blu_trv_set_boost() method."""
+    await rpc_device.blu_trv_set_boost(200)
+
+    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] == "BluTRV.Call"
+    assert call_args_list[0][0][0][0][1] == {
+        "id": 200,
+        "method": "Trv.SetBoost",
+        "params": {"id": 0},
+    }
+    assert call_args_list[0][0][1] == 60
+
+
+@pytest.mark.asyncio
+async def test_blu_trv_set_boost_duration(
+    rpc_device: RpcDevice,
+) -> None:
+    """Test RpcDevice blu_trv_set_boost() method with duration."""
+    await rpc_device.blu_trv_set_boost(200, 33)
+
+    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] == "BluTRV.Call"
+    assert call_args_list[0][0][0][0][1] == {
+        "id": 200,
+        "method": "Trv.SetBoost",
+        "params": {"id": 0, "duration": 33},
+    }
+    assert call_args_list[0][0][1] == 60
+
+
+@pytest.mark.asyncio
+async def test_blu_trv_clear_boost(
+    rpc_device: RpcDevice,
+) -> None:
+    """Test RpcDevice blu_trv_clear_boost() method."""
+    await rpc_device.blu_trv_clear_boost(200)
+
+    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] == "BluTRV.Call"
+    assert call_args_list[0][0][0][0][1] == {
+        "id": 200,
+        "method": "Trv.ClearBoost",
+        "params": {"id": 0},
+    }
+    assert call_args_list[0][0][1] == 60
+
+
 @pytest.mark.asyncio
 async def test_number_set(
     rpc_device: RpcDevice,