From 8e0dfbcd1339fa0c178d2058037d3bfebbee011d Mon Sep 17 00:00:00 2001
From: epenet <6771947+epenet@users.noreply.github.com>
Date: Tue, 13 Aug 2024 13:15:35 +0200
Subject: [PATCH] Improve type hints in modbus tests (#123795)

---
 tests/components/modbus/conftest.py    | 4 +++-
 tests/components/modbus/test_sensor.py | 2 +-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/tests/components/modbus/conftest.py b/tests/components/modbus/conftest.py
index 6741504585a..28f8eae5a0b 100644
--- a/tests/components/modbus/conftest.py
+++ b/tests/components/modbus/conftest.py
@@ -192,7 +192,9 @@ async def mock_test_state_fixture(
 
 
 @pytest.fixture(name="mock_modbus_ha")
-async def mock_modbus_ha_fixture(hass, mock_modbus):
+async def mock_modbus_ha_fixture(
+    hass: HomeAssistant, mock_modbus: mock.AsyncMock
+) -> mock.AsyncMock:
     """Load homeassistant to allow service calls."""
     assert await async_setup_component(hass, "homeassistant", {})
     await hass.async_block_till_done()
diff --git a/tests/components/modbus/test_sensor.py b/tests/components/modbus/test_sensor.py
index 20ff558fce6..87015fa634c 100644
--- a/tests/components/modbus/test_sensor.py
+++ b/tests/components/modbus/test_sensor.py
@@ -1335,7 +1335,7 @@ async def test_wrap_sensor(hass: HomeAssistant, mock_do_cycle, expected) -> None
 
 
 @pytest.fixture(name="mock_restore")
-async def mock_restore(hass):
+async def mock_restore(hass: HomeAssistant) -> None:
     """Mock restore cache."""
     mock_restore_cache_with_extra_data(
         hass,