Add tests for LCN integration setup (#48070)

tests/components/lcn/conftest.py

@@ -0,0 +1,97 @@
+"""Test configuration and mocks for LCN component."""
+import json
+from unittest.mock import AsyncMock, patch
+
+import pypck
+from pypck.connection import PchkConnectionManager
+import pypck.module
+from pypck.module import GroupConnection, ModuleConnection
+import pytest
+
+from homeassistant.components.lcn.const import DOMAIN
+from homeassistant.const import CONF_HOST
+from homeassistant.setup import async_setup_component
+
+from tests.common import MockConfigEntry, load_fixture
+
+
+class MockModuleConnection(ModuleConnection):
+    """Fake a LCN module connection."""
+
+    status_request_handler = AsyncMock()
+    activate_status_request_handler = AsyncMock()
+    cancel_status_request_handler = AsyncMock()
+    send_command = AsyncMock(return_value=True)
+
+
+class MockGroupConnection(GroupConnection):
+    """Fake a LCN group connection."""
+
+    send_command = AsyncMock(return_value=True)
+
+
+class MockPchkConnectionManager(PchkConnectionManager):
+    """Fake connection handler."""
+
+    async def async_connect(self, timeout=30):
+        """Mock establishing a connection to PCHK."""
+        self.authentication_completed_future.set_result(True)
+        self.license_error_future.set_result(True)
+        self.segment_scan_completed_event.set()
+
+    async def async_close(self):
+        """Mock closing a connection to PCHK."""
+
+    @patch.object(pypck.connection, "ModuleConnection", MockModuleConnection)
+    @patch.object(pypck.connection, "GroupConnection", MockGroupConnection)
+    def get_address_conn(self, addr):
+        """Get LCN address connection."""
+        return super().get_address_conn(addr, request_serials=False)
+
+    send_command = AsyncMock()
+
+
+def create_config_entry(name):
+    """Set up config entries with configuration data."""
+    fixture_filename = f"lcn/config_entry_{name}.json"
+    entry_data = json.loads(load_fixture(fixture_filename))
+    options = {}
+
+    title = entry_data[CONF_HOST]
+    unique_id = fixture_filename
+    entry = MockConfigEntry(
+        domain=DOMAIN,
+        title=title,
+        unique_id=unique_id,
+        data=entry_data,
+        options=options,
+    )
+    return entry
+
+
+@pytest.fixture(name="entry")
+def create_config_entry_pchk():
+    """Return one specific config entry."""
+    return create_config_entry("pchk")
+
+
+@pytest.fixture(name="entry2")
+def create_config_entry_myhome():
+    """Return one specific config entry."""
+    return create_config_entry("myhome")
+
+
+async def init_integration(hass, entry):
+    """Set up the LCN integration in Home Assistant."""
+    entry.add_to_hass(hass)
+    await hass.config_entries.async_setup(entry.entry_id)
+    await hass.async_block_till_done()
+
+
+async def setup_component(hass):
+    """Set up the LCN component."""
+    fixture_filename = "lcn/config.json"
+    config_data = json.loads(load_fixture(fixture_filename))
+
+    await async_setup_component(hass, DOMAIN, config_data)
+    await hass.async_block_till_done()