Discover aidot wifi lights over dhcp (#177437)

Co-authored-by: s1eedz <ldxapp2016@gmail.com>
Co-authored-by: Josef Zweck <josef@zweck.dev>

homeassistant/components/aidot/manifest.json

@@ -3,6 +3,11 @@
   "name": "AiDot",
   "codeowners": ["@s1eedz", "@HongBryan"],
   "config_flow": true,
+  "dhcp": [
+    {
+      "hostname": "aidot"
+    }
+  ],
   "documentation": "https://www.home-assistant.io/integrations/aidot",
   "integration_type": "hub",
   "iot_class": "local_polling",

homeassistant/components/aidot/quality_scale.yaml

@@ -49,7 +49,7 @@ rules:
   devices: done
   diagnostics: todo
   discovery-update-info: todo
-  discovery: todo
+  discovery: done
   docs-data-update: todo
   docs-examples: todo
   docs-known-limitations: todo

homeassistant/components/aidot/strings.json

@@ -1,7 +1,8 @@
 {
   "config": {
     "abort": {
-      "already_configured": "[%key:common::config_flow::abort::already_configured_account%]"
+      "already_configured": "[%key:common::config_flow::abort::already_configured_account%]",
+      "already_in_progress": "[%key:common::config_flow::abort::already_in_progress%]"
     },
     "error": {
       "cannot_connect": "[%key:common::config_flow::error::cannot_connect%]",

homeassistant/generated/dhcp.py

@@ -11,6 +11,10 @@ DHCP: Final[list[dict[str, str | bool]]] = [
         "hostname": "neo-*",
         "macaddress": "FC0FE7*",
     },
+    {
+        "domain": "aidot",
+        "hostname": "aidot",
+    },
     {
         "domain": "airobot",
         "hostname": "airobot-thermostat-*",

tests/components/aidot/test_config_flow.py

@@ -7,15 +7,22 @@ from aiohttp import ClientError
 import pytest
 
 from homeassistant.components.aidot.const import DOMAIN
-from homeassistant.config_entries import SOURCE_USER
+from homeassistant.config_entries import SOURCE_DHCP, SOURCE_USER
 from homeassistant.const import CONF_COUNTRY_CODE, CONF_PASSWORD, CONF_USERNAME
 from homeassistant.core import HomeAssistant
 from homeassistant.data_entry_flow import FlowResultType
+from homeassistant.helpers.service_info.dhcp import DhcpServiceInfo
 
 from .const import TEST_COUNTRY, TEST_EMAIL, TEST_LOGIN_RESP, TEST_PASSWORD
 
 from tests.common import MockConfigEntry
 
+DHCP_SERVICE_INFO = DhcpServiceInfo(
+    hostname="aidot",
+    ip="192.168.1.100",
+    macaddress="001122334455",
+)
+
 
 async def test_config_flow_cloud_login_success(
     hass: HomeAssistant, mock_setup_entry: AsyncMock
@@ -44,6 +51,56 @@ async def test_config_flow_cloud_login_success(
     assert result["result"].unique_id == TEST_LOGIN_RESP["id"]
 
 
+async def test_dhcp_discovery(hass: HomeAssistant) -> None:
+    """Test DHCP discovery shows the user form."""
+    result = await hass.config_entries.flow.async_init(
+        DOMAIN,
+        context={"source": SOURCE_DHCP},
+        data=DHCP_SERVICE_INFO,
+    )
+
+    assert result["type"] is FlowResultType.FORM
+    assert result["step_id"] == "user"
+    assert result["errors"] == {}
+
+
+async def test_dhcp_discovery_aborts_if_already_configured(
+    hass: HomeAssistant, mock_config_entry: MockConfigEntry
+) -> None:
+    """Test DHCP discovery aborts when the integration is configured."""
+    mock_config_entry.add_to_hass(hass)
+
+    result = await hass.config_entries.flow.async_init(
+        DOMAIN,
+        context={"source": SOURCE_DHCP},
+        data=DHCP_SERVICE_INFO,
+    )
+
+    assert result["type"] is FlowResultType.ABORT
+    assert result["reason"] == "already_configured"
+
+
+async def test_dhcp_discovery_aborts_if_already_in_progress(
+    hass: HomeAssistant,
+) -> None:
+    """Test a duplicate DHCP discovery flow aborts."""
+    first_result = await hass.config_entries.flow.async_init(
+        DOMAIN,
+        context={"source": SOURCE_DHCP},
+        data=DHCP_SERVICE_INFO,
+    )
+    assert first_result["type"] is FlowResultType.FORM
+
+    result = await hass.config_entries.flow.async_init(
+        DOMAIN,
+        context={"source": SOURCE_DHCP},
+        data=DHCP_SERVICE_INFO,
+    )
+
+    assert result["type"] is FlowResultType.ABORT
+    assert result["reason"] == "already_in_progress"
+
+
 @pytest.mark.parametrize(
     ("exception", "error"),
     [