From 4b8a8dda8da880eecbe08cb1a181a0a3e57801cf Mon Sep 17 00:00:00 2001 From: "J. Nick Koston" Date: Wed, 1 Dec 2021 10:13:59 -1000 Subject: [PATCH] Fix yeelight discovery (#60783) 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 a331db8d4ed..4b372df3744 100644 --- a/homeassistant/components/yeelight/scanner.py +++ b/homeassistant/components/yeelight/scanner.py @@ -10,7 +10,7 @@ from urllib.parse import urlparse 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 @@ class YeelightScanner: 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 6f7ae807c9d..b6bf0b10d67 100644 --- a/tests/components/yeelight/__init__.py +++ b/tests/components/yeelight/__init__.py @@ -182,8 +182,8 @@ def _patch_discovery(no_device=False, capabilities=None): 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, )