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
import json
from typing import Any
from unittest.mock import AsyncMock, Mock, patch
import pytest
@ -13,6 +14,7 @@ from homeassistant.const import (
CONF_LATITUDE,
CONF_LONGITUDE,
)
from homeassistant.core import HomeAssistant
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")
def config_entry_fixture(hass, config):
def config_entry_fixture(
hass: HomeAssistant, config: dict[str, Any]
) -> MockConfigEntry:
"""Define a config entry fixture."""
entry = MockConfigEntry(
domain=DOMAIN,
@ -54,7 +58,7 @@ def config_entry_fixture(hass, config):
@pytest.fixture(name="config")
def config_fixture():
def config_fixture() -> dict[str, Any]:
"""Define a config entry data fixture."""
return {
CONF_API_KEY: TEST_API_KEY,
@ -89,7 +93,9 @@ async def mock_pyopenuv_fixture(client):
@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."""
assert await hass.config_entries.async_setup(config_entry.entry_id)
await hass.async_block_till_done()