Improve type hints in openuv tests (#123813)

This commit is contained in:
epenet 2024-08-13 16:07:17 +02:00 committed by GitHub
parent 135f15fdc3
commit 4cc3f7211b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -2,6 +2,7 @@
from collections.abc import Generator from collections.abc import Generator
import json import json
from typing import Any
from unittest.mock import AsyncMock, Mock, patch from unittest.mock import AsyncMock, Mock, patch
import pytest import pytest
@ -13,6 +14,7 @@ from homeassistant.const import (
CONF_LATITUDE, CONF_LATITUDE,
CONF_LONGITUDE, CONF_LONGITUDE,
) )
from homeassistant.core import HomeAssistant
from tests.common import MockConfigEntry, load_fixture from tests.common import MockConfigEntry, load_fixture
@ -41,7 +43,9 @@ def client_fixture(data_protection_window, data_uv_index):
@pytest.fixture(name="config_entry") @pytest.fixture(name="config_entry")
def config_entry_fixture(hass, config): def config_entry_fixture(
hass: HomeAssistant, config: dict[str, Any]
) -> MockConfigEntry:
"""Define a config entry fixture.""" """Define a config entry fixture."""
entry = MockConfigEntry( entry = MockConfigEntry(
domain=DOMAIN, domain=DOMAIN,
@ -54,7 +58,7 @@ def config_entry_fixture(hass, config):
@pytest.fixture(name="config") @pytest.fixture(name="config")
def config_fixture(): def config_fixture() -> dict[str, Any]:
"""Define a config entry data fixture.""" """Define a config entry data fixture."""
return { return {
CONF_API_KEY: TEST_API_KEY, CONF_API_KEY: TEST_API_KEY,
@ -89,7 +93,9 @@ async def mock_pyopenuv_fixture(client):
@pytest.fixture(name="setup_config_entry") @pytest.fixture(name="setup_config_entry")
async def setup_config_entry_fixture(hass, config_entry, mock_pyopenuv): async def setup_config_entry_fixture(
hass: HomeAssistant, config_entry: MockConfigEntry, mock_pyopenuv: None
) -> None:
"""Define a fixture to set up openuv.""" """Define a fixture to set up openuv."""
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()