Add method to configure CoIoT protocol (gen1) (#1026)

* Add method to configure CoIoT protocol (gen1)

* add tests

* comment

Author: chemelli74

aioshelly/block_device/device.py

@@ -443,6 +443,11 @@ async def set_thermostat_state(self, channel: int = 0, **kwargs: Any) -> None:
         """Set thermostat state (Shelly TRV)."""
         await self.http_request("get", f"thermostat/{channel}", kwargs)
 
+    async def configure_coiot_protocol(self, address: str, port: int) -> None:
+        """Configure CoIoT protocol."""
+        params = {"coiot_enable": "true", "coiot_peer": f"{address}:{port}"}
+        await self.http_request("post", "settings/advanced", params)
+
     @property
     def requires_auth(self) -> bool:
         """Device check for authentication."""

tests/block_device/conftest.py

@@ -6,8 +6,19 @@
 import pytest_asyncio
 from aiohttp.client import ClientSession
 
+import aioshelly.block_device.device as device_module
+from aioshelly.block_device import BlockDevice
+
 
 @pytest_asyncio.fixture
 async def client_session() -> AsyncGenerator[ClientSession, None]:
     """Fixture for a ClientSession."""
     return MagicMock(spec=ClientSession)
+
+
+@pytest_asyncio.fixture
+async def mock_block_device() -> AsyncGenerator[BlockDevice, None]:
+    """Fixture for a mock block device."""
+    # Create a fake BlockDevice instance
+    fake_block = device_module.BlockDevice.__new__(device_module.BlockDevice)
+    yield fake_block

tests/block_device/test_device.py

@@ -2,7 +2,7 @@
 
 import asyncio
 import socket
-from unittest.mock import MagicMock, Mock
+from unittest.mock import AsyncMock, MagicMock, Mock
 
 import pytest
 from aiohttp.client import ClientSession
@@ -65,3 +65,36 @@ async def test_block_device_ip_address_property_guard(
 
     with pytest.raises(RuntimeError, match="Block device ip_address is None"):
         _ = block_device.ip_address
+
+
+@pytest.mark.asyncio
+async def test_configure_coiot_protocol_sends_correct_parameters(
+    mock_block_device: BlockDevice,
+) -> None:
+    """Test that configure_coiot_protocol calls http_request with expected params."""
+    mock_block_device.http_request = AsyncMock()
+
+    test_address = "10.10.10.10"
+    test_port = 5683
+
+    await mock_block_device.configure_coiot_protocol(test_address, test_port)
+
+    mock_block_device.http_request.assert_awaited_once_with(
+        "post",
+        "settings/advanced",
+        {
+            "coiot_enable": "true",
+            "coiot_peer": f"{test_address}:{test_port}",
+        },
+    )
+
+
+@pytest.mark.asyncio
+async def test_configure_coiot_protocol_propagates_exceptions(
+    mock_block_device: BlockDevice,
+) -> None:
+    """Test that exceptions from http_request bubble up (no silent swallow)."""
+    mock_block_device.http_request = AsyncMock(side_effect=RuntimeError("HTTP failure"))
+
+    with pytest.raises(RuntimeError):
+        await mock_block_device.configure_coiot_protocol("10.10.10.10", 5683)