diff --git a/homeassistant/components/nanoleaf/config_flow.py b/homeassistant/components/nanoleaf/config_flow.py index e54d0646d21..6ae70b32d8e 100644 --- a/homeassistant/components/nanoleaf/config_flow.py +++ b/homeassistant/components/nanoleaf/config_flow.py @@ -106,20 +106,18 @@ class ConfigFlow(config_entries.ConfigFlow, domain=DOMAIN): ) -> FlowResult: """Handle Nanoleaf Homekit and Zeroconf discovery.""" return await self._async_discovery_handler( - discovery_info[zeroconf.ATTR_HOST], - discovery_info[zeroconf.ATTR_NAME].replace( - f".{discovery_info[zeroconf.ATTR_TYPE]}", "" - ), - discovery_info[zeroconf.ATTR_PROPERTIES][zeroconf.ATTR_PROPERTIES_ID], + discovery_info.host, + discovery_info.name.replace(f".{discovery_info.type}", ""), + discovery_info.properties[zeroconf.ATTR_PROPERTIES_ID], ) async def async_step_ssdp(self, discovery_info: ssdp.SsdpServiceInfo) -> FlowResult: """Handle Nanoleaf SSDP discovery.""" _LOGGER.debug("SSDP discovered: %s", discovery_info) return await self._async_discovery_handler( - discovery_info["_host"], - discovery_info["nl-devicename"], - discovery_info["nl-deviceid"], + discovery_info.ssdp_headers["_host"], + discovery_info.ssdp_headers["nl-devicename"], + discovery_info.ssdp_headers["nl-deviceid"], ) async def _async_discovery_handler( diff --git a/tests/components/nanoleaf/test_config_flow.py b/tests/components/nanoleaf/test_config_flow.py index 3b2cd3777cb..4e4a48e9bfe 100644 --- a/tests/components/nanoleaf/test_config_flow.py +++ b/tests/components/nanoleaf/test_config_flow.py @@ -465,7 +465,8 @@ async def test_ssdp_discovery(hass: HomeAssistant) -> None: data=ssdp.SsdpServiceInfo( ssdp_usn="mock_usn", ssdp_st="mock_st", - upnp={ + upnp={}, + ssdp_headers={ "_host": TEST_HOST, "nl-devicename": TEST_NAME, "nl-deviceid": TEST_DEVICE_ID,