From 165ec62405de083fde4d5eb193347e1b8d585f0d Mon Sep 17 00:00:00 2001
From: epenet <6771947+epenet@users.noreply.github.com>
Date: Wed, 14 Aug 2024 12:48:10 +0200
Subject: [PATCH] Improve type hints in ssdp tests (#123892)

---
 tests/components/ssdp/conftest.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/tests/components/ssdp/conftest.py b/tests/components/ssdp/conftest.py
index 8b06163cd95..ac0ac7298a8 100644
--- a/tests/components/ssdp/conftest.py
+++ b/tests/components/ssdp/conftest.py
@@ -1,11 +1,14 @@
 """Configuration for SSDP tests."""
 
+from collections.abc import Generator
 from unittest.mock import AsyncMock, patch
 
 from async_upnp_client.server import UpnpServer
 from async_upnp_client.ssdp_listener import SsdpListener
 import pytest
 
+from homeassistant.core import HomeAssistant
+
 
 @pytest.fixture(autouse=True)
 async def silent_ssdp_listener():
@@ -32,7 +35,7 @@ async def disabled_upnp_server():
 
 
 @pytest.fixture
-def mock_flow_init(hass):
+def mock_flow_init(hass: HomeAssistant) -> Generator[AsyncMock]:
     """Mock hass.config_entries.flow.async_init."""
     with patch.object(
         hass.config_entries.flow, "async_init", return_value=AsyncMock()