Implement bridge-backed Home Assistant integration

2026-09-04 12:57:32 +02:00
parent 42050010b2
commit 9766f7f9b2
25 changed files with 2154 additions and 7 deletions
@@ -0,0 +1,148 @@
+"""Tests for the MB-Secure config flow."""
+
+from unittest.mock import patch
+
+from homeassistant.config_entries import SOURCE_HASSIO, SOURCE_USER
+from homeassistant.const import CONF_HOST, CONF_PORT
+from homeassistant.core import HomeAssistant
+from homeassistant.data_entry_flow import FlowResultType
+from homeassistant.helpers.service_info.hassio import HassioServiceInfo
+
+from custom_components.mb_secure.api import BridgeAuthenticationError
+from custom_components.mb_secure.const import CONF_TOKEN, DOMAIN
+from custom_components.mb_secure.models import BridgeInfo
+
+USER_INPUT = {
+    CONF_HOST: "bridge.local",
+    CONF_PORT: 8099,
+    CONF_TOKEN: "x" * 32,
+}
+
+
+async def test_user_flow_creates_entry_with_stable_bridge_id(
+    hass: HomeAssistant,
+) -> None:
+    """Successful validation uses the installation ID as config entry ID."""
+    result = await hass.config_entries.flow.async_init(
+        DOMAIN, context={"source": SOURCE_USER}
+    )
+    assert result["type"] is FlowResultType.FORM
+
+    info = BridgeInfo(
+        bridge_id="bridge-installation-1",
+        bridge_version="0.1.0",
+        api_version=1,
+        connected=False,
+        capabilities=frozenset({"snapshot", "events"}),
+    )
+    with patch(
+        "custom_components.mb_secure.api.BridgeClient.async_get_info",
+        return_value=info,
+    ):
+        result = await hass.config_entries.flow.async_configure(
+            result["flow_id"], USER_INPUT
+        )
+
+    assert result["type"] is FlowResultType.CREATE_ENTRY
+    assert result["data"] == USER_INPUT
+    assert result["result"].unique_id == "bridge-installation-1"
+
+
+async def test_user_flow_reports_invalid_token(hass: HomeAssistant) -> None:
+    """A rejected local bridge token remains inside the form flow."""
+    result = await hass.config_entries.flow.async_init(
+        DOMAIN, context={"source": SOURCE_USER}
+    )
+
+    with patch(
+        "custom_components.mb_secure.api.BridgeClient.async_get_info",
+        side_effect=BridgeAuthenticationError,
+    ):
+        result = await hass.config_entries.flow.async_configure(
+            result["flow_id"], USER_INPUT
+        )
+
+    assert result["type"] is FlowResultType.FORM
+    assert result["errors"] == {"base": "invalid_auth"}
+
+
+async def test_user_flow_rejects_bridge_without_stable_id(
+    hass: HomeAssistant,
+) -> None:
+    """A bridge without an installation ID cannot create stable entities."""
+    result = await hass.config_entries.flow.async_init(
+        DOMAIN, context={"source": SOURCE_USER}
+    )
+    info = BridgeInfo(
+        bridge_id=None,
+        bridge_version="0.1.0",
+        api_version=1,
+        connected=False,
+        capabilities=frozenset(),
+    )
+    with patch(
+        "custom_components.mb_secure.api.BridgeClient.async_get_info",
+        return_value=info,
+    ):
+        result = await hass.config_entries.flow.async_configure(
+            result["flow_id"], USER_INPUT
+        )
+
+    assert result["type"] is FlowResultType.FORM
+    assert result["errors"] == {"base": "missing_bridge_id"}
+
+
+async def test_hassio_discovery_requires_confirmation_and_uses_bridge_id(
+    hass: HomeAssistant,
+) -> None:
+    """Supervisor discovery remains user-confirmed and adopts the bridge ID."""
+    discovery = HassioServiceInfo(
+        config=USER_INPUT,
+        name="MB-Secure Bridge",
+        slug="mb_secure_bridge",
+        uuid="discovery-installation-1",
+    )
+    result = await hass.config_entries.flow.async_init(
+        DOMAIN,
+        context={"source": SOURCE_HASSIO},
+        data=discovery,
+    )
+    assert result["type"] is FlowResultType.FORM
+    assert result["step_id"] == "hassio_confirm"
+
+    info = BridgeInfo(
+        bridge_id="bridge-installation-1",
+        bridge_version="0.1.0",
+        api_version=1,
+        connected=True,
+        capabilities=frozenset({"snapshot", "events"}),
+    )
+    with patch(
+        "custom_components.mb_secure.api.BridgeClient.async_get_info",
+        return_value=info,
+    ):
+        result = await hass.config_entries.flow.async_configure(result["flow_id"], {})
+
+    assert result["type"] is FlowResultType.CREATE_ENTRY
+    assert result["result"].unique_id == "bridge-installation-1"
+    assert result["data"] == USER_INPUT
+
+
+async def test_hassio_discovery_rejects_incomplete_data(
+    hass: HomeAssistant,
+) -> None:
+    """Supervisor discovery never guesses missing connection credentials."""
+    discovery = HassioServiceInfo(
+        config={"host": "bridge.local"},
+        name="MB-Secure Bridge",
+        slug="mb_secure_bridge",
+        uuid="discovery-installation-1",
+    )
+    result = await hass.config_entries.flow.async_init(
+        DOMAIN,
+        context={"source": SOURCE_HASSIO},
+        data=discovery,
+    )
+
+    assert result["type"] is FlowResultType.ABORT
+    assert result["reason"] == "invalid_discovery"