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 async_upnp_client.search import SsdpSearchListener
from homeassistant import config_entries from homeassistant import config_entries
from homeassistant.components import network from homeassistant.components import network, ssdp
from homeassistant.core import HomeAssistant, callback from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.event import async_call_later, async_track_time_interval 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( self._hass.config_entries.flow.async_init(
DOMAIN, DOMAIN,
context={"source": config_entries.SOURCE_SSDP}, 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 info = None
if not no_device: if not no_device:
info = ssdp.SsdpServiceInfo( info = ssdp.SsdpServiceInfo(
ssdp_usn="mock_usn", ssdp_usn="",
ssdp_st="mock_st", ssdp_st=scanner.SSDP_ST,
upnp={}, upnp={},
ssdp_headers=capabilities or CAPABILITIES, ssdp_headers=capabilities or CAPABILITIES,
) )