From b3a71dcea31fb8c12bf718753a1c154729b547cd Mon Sep 17 00:00:00 2001
From: epenet <6771947+epenet@users.noreply.github.com>
Date: Fri, 7 Jun 2024 11:31:45 +0200
Subject: [PATCH] Improve type hints in homekit_controller tests (#119053)

---
 tests/components/homekit_controller/conftest.py | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/tests/components/homekit_controller/conftest.py b/tests/components/homekit_controller/conftest.py
index 8bfb78b9840..427c5285436 100644
--- a/tests/components/homekit_controller/conftest.py
+++ b/tests/components/homekit_controller/conftest.py
@@ -1,7 +1,7 @@
 """HomeKit controller session fixtures."""
 
 import datetime
-import unittest.mock
+from unittest.mock import MagicMock, patch
 
 from aiohomekit.testing import FakeController
 from freezegun import freeze_time
@@ -26,10 +26,10 @@ def freeze_time_in_future() -> Generator[FrozenDateTimeFactory]:
 
 
 @pytest.fixture
-def controller(hass):
+def controller() -> Generator[FakeController]:
     """Replace aiohomekit.Controller with an instance of aiohomekit.testing.FakeController."""
     instance = FakeController()
-    with unittest.mock.patch(
+    with patch(
         "homeassistant.components.homekit_controller.utils.Controller",
         return_value=instance,
     ):
@@ -37,10 +37,10 @@ def controller(hass):
 
 
 @pytest.fixture(autouse=True)
-def hk_mock_async_zeroconf(mock_async_zeroconf):
+def hk_mock_async_zeroconf(mock_async_zeroconf: MagicMock) -> None:
     """Auto mock zeroconf."""
 
 
 @pytest.fixture(autouse=True)
-def auto_mock_bluetooth(mock_bluetooth):
+def auto_mock_bluetooth(mock_bluetooth: None) -> None:
     """Auto mock bluetooth."""