Improve type hints in sonos tests (#123891)

This commit is contained in:
epenet 2024-08-14 12:48:36 +02:00 committed by GitHub
parent 165ec62405
commit 24a8060f43
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -1,9 +1,10 @@
"""Configuration for Sonos tests."""
import asyncio
from collections.abc import Callable, Generator
from collections.abc import Callable, Coroutine, Generator
from copy import copy
from ipaddress import ip_address
from typing import Any
from unittest.mock import AsyncMock, MagicMock, Mock, patch
import pytest
@ -120,7 +121,9 @@ async def async_autosetup_sonos(async_setup_sonos):
@pytest.fixture
def async_setup_sonos(hass, config_entry, fire_zgs_event):
def async_setup_sonos(
hass: HomeAssistant, config_entry: MockConfigEntry, fire_zgs_event
) -> Callable[[], Coroutine[Any, Any, None]]:
"""Return a coroutine to set up a Sonos integration instance on demand."""
async def _wrapper():
@ -136,7 +139,7 @@ def async_setup_sonos(hass, config_entry, fire_zgs_event):
@pytest.fixture(name="config_entry")
def config_entry_fixture():
def config_entry_fixture() -> MockConfigEntry:
"""Create a mock Sonos config entry."""
return MockConfigEntry(domain=DOMAIN, title="Sonos")
@ -650,7 +653,9 @@ def zgs_discovery_fixture():
@pytest.fixture(name="fire_zgs_event")
def zgs_event_fixture(hass: HomeAssistant, soco: SoCo, zgs_discovery: str):
def zgs_event_fixture(
hass: HomeAssistant, soco: SoCo, zgs_discovery: str
) -> Callable[[], Coroutine[Any, Any, None]]:
"""Create alarm_event fixture."""
variables = {"ZoneGroupState": zgs_discovery}