Improve type hints in risco tests (#123887)

This commit is contained in:
epenet 2024-08-14 12:50:29 +02:00 committed by GitHub
parent 7ff368fe0d
commit 13b071fd72
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 15 additions and 7 deletions

View file

@ -1,7 +1,10 @@
"""Fixtures for Risco tests."""
from collections.abc import AsyncGenerator
from typing import Any
from unittest.mock import MagicMock, PropertyMock, patch
from pyrisco.cloud.event import Event
import pytest
from homeassistant.components.risco.const import DOMAIN, TYPE_LOCAL
@ -13,6 +16,7 @@ from homeassistant.const import (
CONF_TYPE,
CONF_USERNAME,
)
from homeassistant.core import HomeAssistant
from .util import TEST_SITE_NAME, TEST_SITE_UUID, system_mock, zone_mock
@ -116,19 +120,19 @@ def two_zone_local():
@pytest.fixture
def options():
def options() -> dict[str, Any]:
"""Fixture for default (empty) options."""
return {}
@pytest.fixture
def events():
def events() -> list[Event]:
"""Fixture for default (empty) events."""
return []
@pytest.fixture
def cloud_config_entry(hass, options):
def cloud_config_entry(hass: HomeAssistant, options: dict[str, Any]) -> MockConfigEntry:
"""Fixture for a cloud config entry."""
config_entry = MockConfigEntry(
domain=DOMAIN,
@ -151,7 +155,9 @@ def login_with_error(exception):
@pytest.fixture
async def setup_risco_cloud(hass, cloud_config_entry, events):
async def setup_risco_cloud(
hass: HomeAssistant, cloud_config_entry: MockConfigEntry, events: list[Event]
) -> AsyncGenerator[MockConfigEntry]:
"""Set up a Risco integration for testing."""
with (
patch(
@ -181,7 +187,7 @@ async def setup_risco_cloud(hass, cloud_config_entry, events):
@pytest.fixture
def local_config_entry(hass, options):
def local_config_entry(hass: HomeAssistant, options: dict[str, Any]) -> MockConfigEntry:
"""Fixture for a local config entry."""
config_entry = MockConfigEntry(
domain=DOMAIN, data=TEST_LOCAL_CONFIG, options=options
@ -201,7 +207,9 @@ def connect_with_error(exception):
@pytest.fixture
async def setup_risco_local(hass, local_config_entry):
async def setup_risco_local(
hass: HomeAssistant, local_config_entry: MockConfigEntry
) -> AsyncGenerator[MockConfigEntry]:
"""Set up a local Risco integration for testing."""
with (
patch(

View file

@ -160,7 +160,7 @@ def _check_state(hass, category, entity_id):
@pytest.fixture
async def _set_utc_time_zone(hass):
async def _set_utc_time_zone(hass: HomeAssistant) -> None:
await hass.config.async_set_time_zone("UTC")