From 45ab9cae1a76c362f13dbf6a6fa5724a6fe95761 Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Fri, 5 Jul 2024 16:32:10 +0200 Subject: [PATCH] Improve type hints for silent_ssdp_scanner in tests (#121309) --- tests/components/samsungtv/conftest.py | 2 +- tests/components/sonos/conftest.py | 4 ++-- tests/components/upnp/conftest.py | 3 ++- tests/components/yamaha_musiccast/test_config_flow.py | 3 ++- 4 files changed, 7 insertions(+), 5 deletions(-) diff --git a/tests/components/samsungtv/conftest.py b/tests/components/samsungtv/conftest.py index 60cefa4959c..752bce3b960 100644 --- a/tests/components/samsungtv/conftest.py +++ b/tests/components/samsungtv/conftest.py @@ -36,7 +36,7 @@ def mock_setup_entry() -> Generator[AsyncMock]: @pytest.fixture(autouse=True) -async def silent_ssdp_scanner(hass): +def silent_ssdp_scanner() -> Generator[None]: """Start SSDP component and get Scanner, prevent actual SSDP traffic.""" with ( patch("homeassistant.components.ssdp.Scanner._async_start_ssdp_listeners"), diff --git a/tests/components/sonos/conftest.py b/tests/components/sonos/conftest.py index 51dd2b9047c..26666d98ced 100644 --- a/tests/components/sonos/conftest.py +++ b/tests/components/sonos/conftest.py @@ -1,7 +1,7 @@ """Configuration for Sonos tests.""" import asyncio -from collections.abc import Callable +from collections.abc import Callable, Generator from copy import copy from ipaddress import ip_address from unittest.mock import AsyncMock, MagicMock, Mock, patch @@ -271,7 +271,7 @@ def soco_fixture(soco_factory): @pytest.fixture(autouse=True) -async def silent_ssdp_scanner(hass): +def silent_ssdp_scanner() -> Generator[None]: """Start SSDP component and get Scanner, prevent actual SSDP traffic.""" with ( patch("homeassistant.components.ssdp.Scanner._async_start_ssdp_listeners"), diff --git a/tests/components/upnp/conftest.py b/tests/components/upnp/conftest.py index 8f5de1fa824..35e96ff7284 100644 --- a/tests/components/upnp/conftest.py +++ b/tests/components/upnp/conftest.py @@ -2,6 +2,7 @@ from __future__ import annotations +from collections.abc import Generator import copy from datetime import datetime import socket @@ -153,7 +154,7 @@ def mock_setup_entry(): @pytest.fixture(autouse=True) -async def silent_ssdp_scanner(hass): +def silent_ssdp_scanner() -> Generator[None]: """Start SSDP component and get Scanner, prevent actual SSDP traffic.""" with ( patch("homeassistant.components.ssdp.Scanner._async_start_ssdp_listeners"), diff --git a/tests/components/yamaha_musiccast/test_config_flow.py b/tests/components/yamaha_musiccast/test_config_flow.py index 321e7250e5a..7629d2401c2 100644 --- a/tests/components/yamaha_musiccast/test_config_flow.py +++ b/tests/components/yamaha_musiccast/test_config_flow.py @@ -1,5 +1,6 @@ """Test config flow.""" +from collections.abc import Generator from unittest.mock import patch from aiomusiccast import MusicCastConnectionException @@ -17,7 +18,7 @@ from tests.common import MockConfigEntry @pytest.fixture(autouse=True) -async def silent_ssdp_scanner(hass): +def silent_ssdp_scanner() -> Generator[None]: """Start SSDP component and get Scanner, prevent actual SSDP traffic.""" with ( patch("homeassistant.components.ssdp.Scanner._async_start_ssdp_listeners"),