From 6626c63bb516d2165c8cd7e8855c74add93cae1d Mon Sep 17 00:00:00 2001
From: epenet <6771947+epenet@users.noreply.github.com>
Date: Wed, 14 Aug 2024 12:51:52 +0200
Subject: [PATCH] Improve type hints in recollect_waste tests (#123882)

---
 tests/components/recollect_waste/conftest.py | 14 ++++++++++----
 1 file changed, 10 insertions(+), 4 deletions(-)

diff --git a/tests/components/recollect_waste/conftest.py b/tests/components/recollect_waste/conftest.py
index 360dd8aac98..8384da3f388 100644
--- a/tests/components/recollect_waste/conftest.py
+++ b/tests/components/recollect_waste/conftest.py
@@ -1,6 +1,7 @@
 """Define test fixtures for ReCollect Waste."""
 
 from datetime import date
+from typing import Any
 from unittest.mock import AsyncMock, Mock, patch
 
 from aiorecollect.client import PickupEvent, PickupType
@@ -11,6 +12,7 @@ from homeassistant.components.recollect_waste.const import (
     CONF_SERVICE_ID,
     DOMAIN,
 )
+from homeassistant.core import HomeAssistant
 
 from tests.common import MockConfigEntry
 
@@ -25,7 +27,9 @@ def client_fixture(pickup_events):
 
 
 @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, unique_id=f"{TEST_PLACE_ID}, {TEST_SERVICE_ID}", data=config
@@ -35,7 +39,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_PLACE_ID: TEST_PLACE_ID,
@@ -54,7 +58,7 @@ def pickup_events_fixture():
 
 
 @pytest.fixture(name="mock_aiorecollect")
-async def mock_aiorecollect_fixture(client):
+def mock_aiorecollect_fixture(client):
     """Define a fixture to patch aiorecollect."""
     with (
         patch(
@@ -70,7 +74,9 @@ async def mock_aiorecollect_fixture(client):
 
 
 @pytest.fixture(name="setup_config_entry")
-async def setup_config_entry_fixture(hass, config_entry, mock_aiorecollect):
+async def setup_config_entry_fixture(
+    hass: HomeAssistant, config_entry: MockConfigEntry, mock_aiorecollect: None
+) -> None:
     """Define a fixture to set up recollect_waste."""
     assert await hass.config_entries.async_setup(config_entry.entry_id)
     await hass.async_block_till_done()