Add wall_display_set_screen method (#977)

Author: bieniu

aioshelly/rpc_device/device.py

@@ -575,6 +575,11 @@ async def update_cover_status(self, id_: int) -> None:
         cover_status = await self.cover_get_status(id_)
         self._status[key].update(cover_status)
 
+    async def wall_display_set_screen(self, value: bool) -> None:
+        """Set Wall Display screen on/off."""
+        params = {"on": value}
+        await self.call_rpc("Ui.Screen.Set", params)
+
     async def poll(self) -> None:
         """Poll device for calls that do not receive push updates."""
         calls: list[tuple[str, dict[str, Any] | None]] = [("Shelly.GetStatus", None)]

tests/rpc_device/test_device.py

@@ -1651,3 +1651,17 @@ async def test_rpc_device_initialize_ble() -> None:
     # Verify BLE connect was called
     assert mock_ble_rpc.connect.called
     assert rpc_device.initialized
+
+
+@pytest.mark.asyncio
+async def test_wall_display_set_screen(
+    rpc_device: RpcDevice,
+) -> None:
+    """Test RpcDevice wall_display_set_screen() method."""
+    await rpc_device.wall_display_set_screen(True)
+
+    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] == "Ui.Screen.Set"
+    assert call_args_list[0][0][0][0][1] == {"on": True}