jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-19 19:27:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

Improve type hints in purpleair tests (#123874)

Browse Source
This commit is contained in:
epenet 2024-08-14 13:05:52 +02:00 committed by GitHub
parent 1af6528f4f
commit 5f1d7e5566
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 16 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
tests/components/purpleair/conftest.py
View File

@ -1,5 +1,7 @@
"""Define fixtures for PurpleAir tests.""" """Define fixtures for PurpleAir tests."""
from collections.abc import Generator
from typing import Any
from unittest.mock import AsyncMock, Mock, patch from unittest.mock import AsyncMock, Mock, patch
from aiopurpleair.endpoints.sensors import NearbySensorResult from aiopurpleair.endpoints.sensors import NearbySensorResult
@ -7,6 +9,7 @@ from aiopurpleair.models.sensors import GetSensorsResponse
import pytest import pytest
from homeassistant.components.purpleair import DOMAIN from homeassistant.components.purpleair import DOMAIN
from homeassistant.core import HomeAssistant
from tests.common import MockConfigEntry, load_fixture from tests.common import MockConfigEntry, load_fixture
@ -16,7 +19,7 @@ TEST_SENSOR_INDEX2 = 567890
@pytest.fixture(name="api") @pytest.fixture(name="api")
def api_fixture(get_sensors_response): def api_fixture(get_sensors_response: GetSensorsResponse) -> Mock:
"""Define a fixture to return a mocked aiopurple API object.""" """Define a fixture to return a mocked aiopurple API object."""
return Mock( return Mock(
async_check_api_key=AsyncMock(), async_check_api_key=AsyncMock(),
@ -34,7 +37,11 @@ def api_fixture(get_sensors_response):
@pytest.fixture(name="config_entry") @pytest.fixture(name="config_entry")
def config_entry_fixture(hass, config_entry_data, config_entry_options): def config_entry_fixture(
hass: HomeAssistant,
config_entry_data: dict[str, Any],
config_entry_options: dict[str, Any],
) -> MockConfigEntry:
"""Define a config entry fixture.""" """Define a config entry fixture."""
entry = MockConfigEntry( entry = MockConfigEntry(
domain=DOMAIN, domain=DOMAIN,
@ -48,7 +55,7 @@ def config_entry_fixture(hass, config_entry_data, config_entry_options):
@pytest.fixture(name="config_entry_data") @pytest.fixture(name="config_entry_data")
def config_entry_data_fixture(): def config_entry_data_fixture() -> dict[str, Any]:
"""Define a config entry data fixture.""" """Define a config entry data fixture."""
return { return {
"api_key": TEST_API_KEY, "api_key": TEST_API_KEY,
@ -56,7 +63,7 @@ def config_entry_data_fixture():
@pytest.fixture(name="config_entry_options") @pytest.fixture(name="config_entry_options")
def config_entry_options_fixture(): def config_entry_options_fixture() -> dict[str, Any]:
"""Define a config entry options fixture.""" """Define a config entry options fixture."""
return { return {
"sensor_indices": [TEST_SENSOR_INDEX1], "sensor_indices": [TEST_SENSOR_INDEX1],
@ -64,7 +71,7 @@ def config_entry_options_fixture():
@pytest.fixture(name="get_sensors_response", scope="package") @pytest.fixture(name="get_sensors_response", scope="package")
def get_sensors_response_fixture(): def get_sensors_response_fixture() -> GetSensorsResponse:
"""Define a fixture to mock an aiopurpleair GetSensorsResponse object.""" """Define a fixture to mock an aiopurpleair GetSensorsResponse object."""
return GetSensorsResponse.parse_raw( return GetSensorsResponse.parse_raw(
load_fixture("get_sensors_response.json", "purpleair") load_fixture("get_sensors_response.json", "purpleair")
@ -72,7 +79,7 @@ def get_sensors_response_fixture():
@pytest.fixture(name="mock_aiopurpleair") @pytest.fixture(name="mock_aiopurpleair")
async def mock_aiopurpleair_fixture(api): def mock_aiopurpleair_fixture(api: Mock) -> Generator[Mock]:
"""Define a fixture to patch aiopurpleair.""" """Define a fixture to patch aiopurpleair."""
with ( with (
patch("homeassistant.components.purpleair.config_flow.API", return_value=api), patch("homeassistant.components.purpleair.config_flow.API", return_value=api),
@ -82,7 +89,9 @@ async def mock_aiopurpleair_fixture(api):
@pytest.fixture(name="setup_config_entry") @pytest.fixture(name="setup_config_entry")
async def setup_config_entry_fixture(hass, config_entry, mock_aiopurpleair): async def setup_config_entry_fixture(
hass: HomeAssistant, config_entry: MockConfigEntry, mock_aiopurpleair: Mock
) -> None:
"""Define a fixture to set up purpleair.""" """Define a fixture to set up purpleair."""
assert await hass.config_entries.async_setup(config_entry.entry_id) assert await hass.config_entries.async_setup(config_entry.entry_id)
await hass.async_block_till_done() await hass.async_block_till_done()