mirror of
https://github.com/home-assistant/core.git
synced 2025-07-22 20:57:21 +00:00
Use ZeroconfServiceInfo in lutron_caseta (#59988)
Co-authored-by: epenet <epenet@users.noreply.github.com>
This commit is contained in:
parent
2b7bcd6aeb
commit
9aa41be8b7
@ -10,8 +10,10 @@ from pylutron_caseta.smartbridge import Smartbridge
|
||||
import voluptuous as vol
|
||||
|
||||
from homeassistant import config_entries
|
||||
from homeassistant.components import zeroconf
|
||||
from homeassistant.const import CONF_HOST, CONF_NAME
|
||||
from homeassistant.core import callback
|
||||
from homeassistant.data_entry_flow import FlowResult
|
||||
|
||||
from .const import (
|
||||
ABORT_REASON_CANNOT_CONNECT,
|
||||
@ -61,16 +63,18 @@ class LutronCasetaFlowHandler(config_entries.ConfigFlow, domain=DOMAIN):
|
||||
|
||||
return self.async_show_form(step_id="user", data_schema=DATA_SCHEMA_USER)
|
||||
|
||||
async def async_step_zeroconf(self, discovery_info):
|
||||
async def async_step_zeroconf(
|
||||
self, discovery_info: zeroconf.ZeroconfServiceInfo
|
||||
) -> FlowResult:
|
||||
"""Handle a flow initialized by zeroconf discovery."""
|
||||
hostname = discovery_info["hostname"]
|
||||
hostname = discovery_info[zeroconf.ATTR_HOSTNAME]
|
||||
if hostname is None or not hostname.startswith("lutron-"):
|
||||
return self.async_abort(reason="not_lutron_device")
|
||||
|
||||
self.lutron_id = hostname.split("-")[1].replace(".local.", "")
|
||||
|
||||
await self.async_set_unique_id(self.lutron_id)
|
||||
host = discovery_info[CONF_HOST]
|
||||
host = discovery_info[zeroconf.ATTR_HOST]
|
||||
self._abort_if_unique_id_configured({CONF_HOST: host})
|
||||
|
||||
self.data[CONF_HOST] = host
|
||||
@ -80,7 +84,9 @@ class LutronCasetaFlowHandler(config_entries.ConfigFlow, domain=DOMAIN):
|
||||
}
|
||||
return await self.async_step_link()
|
||||
|
||||
async def async_step_homekit(self, discovery_info):
|
||||
async def async_step_homekit(
|
||||
self, discovery_info: zeroconf.ZeroconfServiceInfo
|
||||
) -> FlowResult:
|
||||
"""Handle a flow initialized by homekit discovery."""
|
||||
return await self.async_step_zeroconf(discovery_info)
|
||||
|
||||
|
@ -8,6 +8,7 @@ from pylutron_caseta.smartbridge import Smartbridge
|
||||
import pytest
|
||||
|
||||
from homeassistant import config_entries, data_entry_flow
|
||||
from homeassistant.components import zeroconf
|
||||
from homeassistant.components.lutron_caseta import DOMAIN
|
||||
import homeassistant.components.lutron_caseta.config_flow as CasetaConfigFlow
|
||||
from homeassistant.components.lutron_caseta.const import (
|
||||
@ -424,10 +425,10 @@ async def test_zeroconf_host_already_configured(hass, tmpdir):
|
||||
result = await hass.config_entries.flow.async_init(
|
||||
DOMAIN,
|
||||
context={"source": config_entries.SOURCE_ZEROCONF},
|
||||
data={
|
||||
CONF_HOST: "1.1.1.1",
|
||||
ATTR_HOSTNAME: "lutron-abc.local.",
|
||||
},
|
||||
data=zeroconf.ZeroconfServiceInfo(
|
||||
host="1.1.1.1",
|
||||
hostname="lutron-abc.local.",
|
||||
),
|
||||
)
|
||||
await hass.async_block_till_done()
|
||||
|
||||
@ -447,10 +448,10 @@ async def test_zeroconf_lutron_id_already_configured(hass):
|
||||
result = await hass.config_entries.flow.async_init(
|
||||
DOMAIN,
|
||||
context={"source": config_entries.SOURCE_ZEROCONF},
|
||||
data={
|
||||
CONF_HOST: "1.1.1.1",
|
||||
ATTR_HOSTNAME: "lutron-abc.local.",
|
||||
},
|
||||
data=zeroconf.ZeroconfServiceInfo(
|
||||
host="1.1.1.1",
|
||||
hostname="lutron-abc.local.",
|
||||
),
|
||||
)
|
||||
await hass.async_block_till_done()
|
||||
|
||||
@ -465,10 +466,10 @@ async def test_zeroconf_not_lutron_device(hass):
|
||||
result = await hass.config_entries.flow.async_init(
|
||||
DOMAIN,
|
||||
context={"source": config_entries.SOURCE_ZEROCONF},
|
||||
data={
|
||||
CONF_HOST: "1.1.1.1",
|
||||
ATTR_HOSTNAME: "notlutron-abc.local.",
|
||||
},
|
||||
data=zeroconf.ZeroconfServiceInfo(
|
||||
host="1.1.1.1",
|
||||
hostname="notlutron-abc.local.",
|
||||
),
|
||||
)
|
||||
await hass.async_block_till_done()
|
||||
|
||||
@ -489,10 +490,10 @@ async def test_zeroconf(hass, source, tmpdir):
|
||||
result = await hass.config_entries.flow.async_init(
|
||||
DOMAIN,
|
||||
context={"source": source},
|
||||
data={
|
||||
CONF_HOST: "1.1.1.1",
|
||||
ATTR_HOSTNAME: "lutron-abc.local.",
|
||||
},
|
||||
data=zeroconf.ZeroconfServiceInfo(
|
||||
host="1.1.1.1",
|
||||
hostname="lutron-abc.local.",
|
||||
),
|
||||
)
|
||||
await hass.async_block_till_done()
|
||||
|
||||
|
Loading…
x
Reference in New Issue
Block a user