From 4a7b0d105fb9880b4ee30018d4a8c3633ce40ff7 Mon Sep 17 00:00:00 2001 From: "J. Nick Koston" Date: Wed, 1 Dec 2021 09:21:04 -1000 Subject: [PATCH] Fix yeelight discovery Regressed in #60640 --- homeassistant/components/yeelight/scanner.py | 9 +++++++-- tests/components/yeelight/__init__.py | 4 ++-- 2 files changed, 9 insertions(+), 4 deletions(-) diff --git a/homeassistant/components/yeelight/scanner.py b/homeassistant/components/yeelight/scanner.py index a331db8d4edbf3..4b372df3744ffd 100644 --- a/homeassistant/components/yeelight/scanner.py +++ b/homeassistant/components/yeelight/scanner.py @@ -10,7 +10,7 @@ from async_upnp_client.search import SsdpSearchListener from homeassistant import config_entries -from homeassistant.components import network +from homeassistant.components import network, ssdp from homeassistant.core import HomeAssistant, callback from homeassistant.helpers.event import async_call_later, async_track_time_interval @@ -161,7 +161,12 @@ def _async_start_flow(*_): self._hass.config_entries.flow.async_init( DOMAIN, context={"source": config_entries.SOURCE_SSDP}, - data=response, + data=ssdp.SsdpServiceInfo( + ssdp_usn="", + ssdp_st=SSDP_ST, + ssdp_headers=response, + upnp={}, + ), ) ) diff --git a/tests/components/yeelight/__init__.py b/tests/components/yeelight/__init__.py index 6f7ae807c9dcdc..b6bf0b10d67510 100644 --- a/tests/components/yeelight/__init__.py +++ b/tests/components/yeelight/__init__.py @@ -182,8 +182,8 @@ def _generate_fake_ssdp_listener(*args, **kwargs): info = None if not no_device: info = ssdp.SsdpServiceInfo( - ssdp_usn="mock_usn", - ssdp_st="mock_st", + ssdp_usn="", + ssdp_st=scanner.SSDP_ST, upnp={}, ssdp_headers=capabilities or CAPABILITIES, )