From 8117532cc7be91fea7801dfdb678f24bb851fec4 Mon Sep 17 00:00:00 2001
From: epenet <6771947+epenet@users.noreply.github.com>
Date: Wed, 14 Aug 2024 12:58:49 +0200
Subject: [PATCH] Improve type hints in rainforest_eagle tests (#123880)

---
 tests/components/rainforest_eagle/conftest.py | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/tests/components/rainforest_eagle/conftest.py b/tests/components/rainforest_eagle/conftest.py
index 1aff693e61f..c3790a12e86 100644
--- a/tests/components/rainforest_eagle/conftest.py
+++ b/tests/components/rainforest_eagle/conftest.py
@@ -1,6 +1,7 @@
 """Conftest for rainforest_eagle."""
 
-from unittest.mock import AsyncMock, Mock, patch
+from collections.abc import AsyncGenerator
+from unittest.mock import AsyncMock, MagicMock, Mock, patch
 
 import pytest
 
@@ -13,6 +14,7 @@ from homeassistant.components.rainforest_eagle.const import (
     TYPE_EAGLE_200,
 )
 from homeassistant.const import CONF_HOST, CONF_TYPE
+from homeassistant.core import HomeAssistant
 from homeassistant.setup import async_setup_component
 
 from . import MOCK_200_RESPONSE_WITHOUT_PRICE, MOCK_CLOUD_ID
@@ -21,7 +23,7 @@ from tests.common import MockConfigEntry
 
 
 @pytest.fixture
-def config_entry_200(hass):
+def config_entry_200(hass: HomeAssistant) -> MockConfigEntry:
     """Return a config entry."""
     entry = MockConfigEntry(
         domain="rainforest_eagle",
@@ -38,7 +40,9 @@ def config_entry_200(hass):
 
 
 @pytest.fixture
-async def setup_rainforest_200(hass, config_entry_200):
+async def setup_rainforest_200(
+    hass: HomeAssistant, config_entry_200: MockConfigEntry
+) -> AsyncGenerator[Mock]:
     """Set up rainforest."""
     with patch(
         "aioeagle.ElectricMeter.create_instance",
@@ -53,7 +57,7 @@ async def setup_rainforest_200(hass, config_entry_200):
 
 
 @pytest.fixture
-async def setup_rainforest_100(hass):
+async def setup_rainforest_100(hass: HomeAssistant) -> AsyncGenerator[MagicMock]:
     """Set up rainforest."""
     MockConfigEntry(
         domain="rainforest_eagle",