From fb5cc13061656b631deee8e78bde7f7c995c109f Mon Sep 17 00:00:00 2001
From: epenet <6771947+epenet@users.noreply.github.com>
Date: Mon, 6 Dec 2021 02:51:45 +0100
Subject: [PATCH] Adjust yeelight ssdp_listener tests (#61065)

---
 tests/components/yeelight/__init__.py | 9 ++-------
 1 file changed, 2 insertions(+), 7 deletions(-)

diff --git a/tests/components/yeelight/__init__.py b/tests/components/yeelight/__init__.py
index b6bf0b10d67..b48cfc5402a 100644
--- a/tests/components/yeelight/__init__.py
+++ b/tests/components/yeelight/__init__.py
@@ -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(