From 13b071fd72cee862fd665da22bdc37289d5c324c Mon Sep 17 00:00:00 2001
From: epenet <6771947+epenet@users.noreply.github.com>
Date: Wed, 14 Aug 2024 12:50:29 +0200
Subject: [PATCH] Improve type hints in risco tests (#123887)

---
 tests/components/risco/conftest.py    | 20 ++++++++++++++------
 tests/components/risco/test_sensor.py |  2 +-
 2 files changed, 15 insertions(+), 7 deletions(-)

diff --git a/tests/components/risco/conftest.py b/tests/components/risco/conftest.py
index ab3b64b245d..3961d85d694 100644
--- a/tests/components/risco/conftest.py
+++ b/tests/components/risco/conftest.py
@@ -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(
diff --git a/tests/components/risco/test_sensor.py b/tests/components/risco/test_sensor.py
index 4c8f7bb4180..2b1094554ae 100644
--- a/tests/components/risco/test_sensor.py
+++ b/tests/components/risco/test_sensor.py
@@ -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")