Fix test fixture annotations (#122180)

This commit is contained in:
Marc Mueller 2024-07-19 14:46:30 +02:00 committed by GitHub
parent 281c66b6c2
commit 53c85a5c9b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
11 changed files with 20 additions and 19 deletions

View File

@ -14,7 +14,7 @@ from tests.common import MockConfigEntry, snapshot_platform
@pytest.fixture(autouse=True) @pytest.fixture(autouse=True)
async def platforms() -> AsyncGenerator[list[str]]: async def platforms() -> AsyncGenerator[None]:
"""Return the platforms to be loaded for this test.""" """Return the platforms to be loaded for this test."""
with patch("homeassistant.components.aosmith.PLATFORMS", [Platform.SENSOR]): with patch("homeassistant.components.aosmith.PLATFORMS", [Platform.SENSOR]):
yield yield

View File

@ -29,7 +29,7 @@ from tests.common import MockConfigEntry, snapshot_platform
@pytest.fixture(autouse=True) @pytest.fixture(autouse=True)
async def platforms() -> AsyncGenerator[list[str]]: async def platforms() -> AsyncGenerator[None]:
"""Return the platforms to be loaded for this test.""" """Return the platforms to be loaded for this test."""
with patch("homeassistant.components.aosmith.PLATFORMS", [Platform.WATER_HEATER]): with patch("homeassistant.components.aosmith.PLATFORMS", [Platform.WATER_HEATER]):
yield yield

View File

@ -17,7 +17,7 @@ def mock_setup_entry() -> Generator[AsyncMock]:
@pytest.fixture @pytest.fixture
def ccm15_device() -> Generator[AsyncMock]: def ccm15_device() -> Generator[None]:
"""Mock ccm15 device.""" """Mock ccm15 device."""
ccm15_devices = { ccm15_devices = {
0: CCM15SlaveDevice(bytes.fromhex("000000b0b8001b")), 0: CCM15SlaveDevice(bytes.fromhex("000000b0b8001b")),
@ -32,7 +32,7 @@ def ccm15_device() -> Generator[AsyncMock]:
@pytest.fixture @pytest.fixture
def network_failure_ccm15_device() -> Generator[AsyncMock]: def network_failure_ccm15_device() -> Generator[None]:
"""Mock empty set of ccm15 device.""" """Mock empty set of ccm15 device."""
device_state = CCM15DeviceState(devices={}) device_state = CCM15DeviceState(devices={})
with patch( with patch(

View File

@ -1,10 +1,11 @@
"""Unit test for CCM15 coordinator component.""" """Unit test for CCM15 coordinator component."""
from datetime import timedelta from datetime import timedelta
from unittest.mock import AsyncMock, patch from unittest.mock import patch
from ccm15 import CCM15DeviceState from ccm15 import CCM15DeviceState
from freezegun.api import FrozenDateTimeFactory from freezegun.api import FrozenDateTimeFactory
import pytest
from syrupy.assertion import SnapshotAssertion from syrupy.assertion import SnapshotAssertion
from homeassistant.components.ccm15.const import DOMAIN from homeassistant.components.ccm15.const import DOMAIN
@ -27,11 +28,11 @@ from homeassistant.helpers import entity_registry as er
from tests.common import MockConfigEntry, async_fire_time_changed from tests.common import MockConfigEntry, async_fire_time_changed
@pytest.mark.usefixtures("ccm15_device")
async def test_climate_state( async def test_climate_state(
hass: HomeAssistant, hass: HomeAssistant,
snapshot: SnapshotAssertion, snapshot: SnapshotAssertion,
entity_registry: er.EntityRegistry, entity_registry: er.EntityRegistry,
ccm15_device: AsyncMock,
freezer: FrozenDateTimeFactory, freezer: FrozenDateTimeFactory,
) -> None: ) -> None:
"""Test the coordinator.""" """Test the coordinator."""

View File

@ -1,7 +1,6 @@
"""Test CCM15 diagnostics.""" """Test CCM15 diagnostics."""
from unittest.mock import AsyncMock import pytest
from syrupy import SnapshotAssertion from syrupy import SnapshotAssertion
from homeassistant.components.ccm15.const import DOMAIN from homeassistant.components.ccm15.const import DOMAIN
@ -13,10 +12,10 @@ from tests.components.diagnostics import get_diagnostics_for_config_entry
from tests.typing import ClientSessionGenerator from tests.typing import ClientSessionGenerator
@pytest.mark.usefixtures("ccm15_device")
async def test_entry_diagnostics( async def test_entry_diagnostics(
hass: HomeAssistant, hass: HomeAssistant,
hass_client: ClientSessionGenerator, hass_client: ClientSessionGenerator,
ccm15_device: AsyncMock,
snapshot: SnapshotAssertion, snapshot: SnapshotAssertion,
) -> None: ) -> None:
"""Test config entry diagnostics.""" """Test config entry diagnostics."""

View File

@ -1,6 +1,6 @@
"""Tests for the ccm15 component.""" """Tests for the ccm15 component."""
from unittest.mock import AsyncMock import pytest
from homeassistant.components.ccm15.const import DOMAIN from homeassistant.components.ccm15.const import DOMAIN
from homeassistant.config_entries import ConfigEntryState from homeassistant.config_entries import ConfigEntryState
@ -10,7 +10,8 @@ from homeassistant.core import HomeAssistant
from tests.common import MockConfigEntry from tests.common import MockConfigEntry
async def test_load_unload(hass: HomeAssistant, ccm15_device: AsyncMock) -> None: @pytest.mark.usefixtures("ccm15_device")
async def test_load_unload(hass: HomeAssistant) -> None:
"""Test options flow.""" """Test options flow."""
entry = MockConfigEntry( entry = MockConfigEntry(
domain=DOMAIN, domain=DOMAIN,

View File

@ -17,7 +17,7 @@ from tests.common import MockConfigEntry
@pytest.fixture(autouse=True) @pytest.fixture(autouse=True)
def platforms_fixture() -> Generator[list[str]]: def platforms_fixture() -> Generator[None]:
"""Return the platforms to be loaded for this test.""" """Return the platforms to be loaded for this test."""
# Arbitrary platform to ensure notifications are loaded # Arbitrary platform to ensure notifications are loaded
with patch("homeassistant.components.flume.PLATFORMS", [Platform.BINARY_SENSOR]): with patch("homeassistant.components.flume.PLATFORMS", [Platform.BINARY_SENSOR]):

View File

@ -1,7 +1,7 @@
"""Common fixtures for Hunter Douglas Powerview tests.""" """Common fixtures for Hunter Douglas Powerview tests."""
from collections.abc import Generator from collections.abc import Generator
from unittest.mock import AsyncMock, MagicMock, PropertyMock, patch from unittest.mock import AsyncMock, PropertyMock, patch
from aiopvapi.resources.shade import ShadePosition from aiopvapi.resources.shade import ShadePosition
import pytest import pytest
@ -29,7 +29,7 @@ def mock_hunterdouglas_hub(
rooms_json: str, rooms_json: str,
scenes_json: str, scenes_json: str,
shades_json: str, shades_json: str,
) -> Generator[MagicMock]: ) -> Generator[None]:
"""Return a mocked Powerview Hub with all data populated.""" """Return a mocked Powerview Hub with all data populated."""
with ( with (
patch( patch(

View File

@ -14,10 +14,10 @@ from .const import MOCK_MAC
from tests.common import MockConfigEntry from tests.common import MockConfigEntry
@pytest.mark.usefixtures("mock_hunterdouglas_hub")
@pytest.mark.parametrize("api_version", [1, 2, 3]) @pytest.mark.parametrize("api_version", [1, 2, 3])
async def test_scenes( async def test_scenes(
hass: HomeAssistant, hass: HomeAssistant,
mock_hunterdouglas_hub: None,
api_version: int, api_version: int,
) -> None: ) -> None:
"""Test the scenes.""" """Test the scenes."""

View File

@ -3,7 +3,7 @@
from __future__ import annotations from __future__ import annotations
from collections.abc import Generator from collections.abc import Generator
from unittest.mock import AsyncMock, MagicMock, patch from unittest.mock import MagicMock, patch
from gotailwind import TailwindDeviceStatus from gotailwind import TailwindDeviceStatus
import pytest import pytest
@ -36,7 +36,7 @@ def mock_config_entry() -> MockConfigEntry:
@pytest.fixture @pytest.fixture
def mock_setup_entry() -> Generator[AsyncMock]: def mock_setup_entry() -> Generator[None]:
"""Mock setting up a config entry.""" """Mock setting up a config entry."""
with patch( with patch(
"homeassistant.components.tailwind.async_setup_entry", return_value=True "homeassistant.components.tailwind.async_setup_entry", return_value=True

View File

@ -3,7 +3,7 @@
from __future__ import annotations from __future__ import annotations
from collections.abc import Generator from collections.abc import Generator
from unittest.mock import AsyncMock, MagicMock, patch from unittest.mock import MagicMock, patch
import pytest import pytest
@ -35,7 +35,7 @@ def mock_config_entry() -> MockConfigEntry:
@pytest.fixture @pytest.fixture
def mock_setup_entry() -> Generator[AsyncMock]: def mock_setup_entry() -> Generator[None]:
"""Mock setting up a config entry.""" """Mock setting up a config entry."""
with patch("homeassistant.components.tuya.async_setup_entry", return_value=True): with patch("homeassistant.components.tuya.async_setup_entry", return_value=True):
yield yield