evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Adjust yeelight ssdp_listener tests (#61065)

Browse source
This commit is contained in:
epenet 2021-12-06 02:51:45 +01:00 • committed by GitHub
parent 4e9fd56b8c
commit fb5cc13061
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
tests/components/yeelight/__init__.py
View file

@ -157,7 +157,7 @@ def _mocked_bulb(cannot_connect=False):
return bulb
def _patched_ssdp_listener(info, *args, **kwargs):
def _patched_ssdp_listener(info: ssdp.SsdpHeaders, *args, **kwargs):
listener = SsdpSearchListener(*args, **kwargs)
async def _async_callback(*_):
@ -181,12 +181,7 @@ def _patch_discovery(no_device=False, capabilities=None):
def _generate_fake_ssdp_listener(*args, **kwargs):
info = None
if not no_device:
info = ssdp.SsdpServiceInfo(
ssdp_usn="",
ssdp_st=scanner.SSDP_ST,
upnp={},
ssdp_headers=capabilities or CAPABILITIES,
)
info = capabilities or CAPABILITIES
return _patched_ssdp_listener(info, *args, **kwargs)
return patch(