Adjust conftest type hints (#117900)

This commit is contained in:
epenet 2024-05-22 09:34:17 +02:00 committed by GitHub
parent 39b4e890a0
commit cddb057eae
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -1,5 +1,7 @@
"""Fixtures for component testing."""
from __future__ import annotations
from collections.abc import Callable, Generator
from typing import TYPE_CHECKING, Any
from unittest.mock import MagicMock, patch
@ -9,13 +11,12 @@ import pytest
from homeassistant.const import STATE_OFF, STATE_ON
from homeassistant.core import HomeAssistant
from tests.components.conversation import MockAgent
if TYPE_CHECKING:
from tests.components.device_tracker.common import MockScanner
from tests.components.light.common import MockLight
from tests.components.sensor.common import MockSensor
from tests.components.switch.common import MockSwitch
from .conversation import MockAgent
from .device_tracker.common import MockScanner
from .light.common import MockLight
from .sensor.common import MockSensor
from .switch.common import MockSwitch
@pytest.fixture(scope="session", autouse=True)
@ -125,7 +126,7 @@ def prevent_ffmpeg_subprocess() -> Generator[None, None, None]:
@pytest.fixture
def mock_light_entities() -> list["MockLight"]:
def mock_light_entities() -> list[MockLight]:
"""Return mocked light entities."""
from tests.components.light.common import MockLight
@ -137,7 +138,7 @@ def mock_light_entities() -> list["MockLight"]:
@pytest.fixture
def mock_sensor_entities() -> dict[str, "MockSensor"]:
def mock_sensor_entities() -> dict[str, MockSensor]:
"""Return mocked sensor entities."""
from tests.components.sensor.common import get_mock_sensor_entities
@ -145,7 +146,7 @@ def mock_sensor_entities() -> dict[str, "MockSensor"]:
@pytest.fixture
def mock_switch_entities() -> list["MockSwitch"]:
def mock_switch_entities() -> list[MockSwitch]:
"""Return mocked toggle entities."""
from tests.components.switch.common import get_mock_switch_entities
@ -153,7 +154,7 @@ def mock_switch_entities() -> list["MockSwitch"]:
@pytest.fixture
def mock_legacy_device_scanner() -> "MockScanner":
def mock_legacy_device_scanner() -> MockScanner:
"""Return mocked legacy device scanner entity."""
from tests.components.device_tracker.common import MockScanner
@ -161,9 +162,7 @@ def mock_legacy_device_scanner() -> "MockScanner":
@pytest.fixture
def mock_legacy_device_tracker_setup() -> (
Callable[[HomeAssistant, "MockScanner"], None]
):
def mock_legacy_device_tracker_setup() -> Callable[[HomeAssistant, MockScanner], None]:
"""Return setup callable for legacy device tracker setup."""
from tests.components.device_tracker.common import mock_legacy_device_tracker_setup