From 7fe2f175aae3dba7e66bc1af0a322548313756bd Mon Sep 17 00:00:00 2001
From: epenet <6771947+epenet@users.noreply.github.com>
Date: Wed, 14 Aug 2024 12:50:51 +0200
Subject: [PATCH] Improve type hints in ridwell tests (#123886)

---
 tests/components/ridwell/conftest.py | 19 +++++++++++++------
 1 file changed, 13 insertions(+), 6 deletions(-)

diff --git a/tests/components/ridwell/conftest.py b/tests/components/ridwell/conftest.py
index 32907ac8037..6ea9d91f8e9 100644
--- a/tests/components/ridwell/conftest.py
+++ b/tests/components/ridwell/conftest.py
@@ -1,6 +1,8 @@
 """Define test fixtures for Ridwell."""
 
+from collections.abc import Generator
 from datetime import date
+from typing import Any
 from unittest.mock import AsyncMock, Mock, patch
 
 from aioridwell.model import EventState, RidwellPickup, RidwellPickupEvent
@@ -8,6 +10,7 @@ import pytest
 
 from homeassistant.components.ridwell.const import DOMAIN
 from homeassistant.const import CONF_PASSWORD, CONF_USERNAME
+from homeassistant.core import HomeAssistant
 
 from tests.common import MockConfigEntry
 
@@ -19,7 +22,7 @@ TEST_USER_ID = "12345"
 
 
 @pytest.fixture(name="account")
-def account_fixture():
+def account_fixture() -> Mock:
     """Define a Ridwell account."""
     return Mock(
         account_id=TEST_ACCOUNT_ID,
@@ -44,7 +47,7 @@ def account_fixture():
 
 
 @pytest.fixture(name="client")
-def client_fixture(account):
+def client_fixture(account: Mock) -> Mock:
     """Define an aioridwell client."""
     return Mock(
         async_authenticate=AsyncMock(),
@@ -55,7 +58,9 @@ def client_fixture(account):
 
 
 @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,
@@ -68,7 +73,7 @@ def config_entry_fixture(hass, config):
 
 
 @pytest.fixture(name="config")
-def config_fixture(hass):
+def config_fixture() -> dict[str, Any]:
     """Define a config entry data fixture."""
     return {
         CONF_USERNAME: TEST_USERNAME,
@@ -77,7 +82,7 @@ def config_fixture(hass):
 
 
 @pytest.fixture(name="mock_aioridwell")
-async def mock_aioridwell_fixture(hass, client, config):
+def mock_aioridwell_fixture(client: Mock, config: dict[str, Any]) -> Generator[None]:
     """Define a fixture to patch aioridwell."""
     with (
         patch(
@@ -93,7 +98,9 @@ async def mock_aioridwell_fixture(hass, client, config):
 
 
 @pytest.fixture(name="setup_config_entry")
-async def setup_config_entry_fixture(hass, config_entry, mock_aioridwell):
+async def setup_config_entry_fixture(
+    hass: HomeAssistant, config_entry: MockConfigEntry, mock_aioridwell: None
+) -> None:
     """Define a fixture to set up ridwell."""
     assert await hass.config_entries.async_setup(config_entry.entry_id)
     await hass.async_block_till_done()