Fix yeelight discovery (#60783)

Regressed in #60640
This commit is contained in:
J. Nick Koston 2021-12-01 10:13:59 -10:00 committed by GitHub
parent 2b49694a4c
commit 4b8a8dda8d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 9 additions and 4 deletions

View File

@ -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={},
),
)
)

View File

@ -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,
)