home-assistant/core
1
0
Fork 1
mirror of https://github.com/home-assistant/core.git synced 2025-07-31 19:25:12 +02:00
Code Issues Packages Projects Releases Wiki Activity

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
1 changed files with 2 additions and 7 deletions
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(