Improve type hints in homekit_controller tests (#119053)

This commit is contained in:
epenet 2024-06-07 11:31:45 +02:00 committed by GitHub
parent af65da3875
commit b3a71dcea3
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -1,7 +1,7 @@
"""HomeKit controller session fixtures.""" """HomeKit controller session fixtures."""
import datetime import datetime
import unittest.mock from unittest.mock import MagicMock, patch
from aiohomekit.testing import FakeController from aiohomekit.testing import FakeController
from freezegun import freeze_time from freezegun import freeze_time
@ -26,10 +26,10 @@ def freeze_time_in_future() -> Generator[FrozenDateTimeFactory]:
@pytest.fixture @pytest.fixture
def controller(hass): def controller() -> Generator[FakeController]:
"""Replace aiohomekit.Controller with an instance of aiohomekit.testing.FakeController.""" """Replace aiohomekit.Controller with an instance of aiohomekit.testing.FakeController."""
instance = FakeController() instance = FakeController()
with unittest.mock.patch( with patch(
"homeassistant.components.homekit_controller.utils.Controller", "homeassistant.components.homekit_controller.utils.Controller",
return_value=instance, return_value=instance,
): ):
@ -37,10 +37,10 @@ def controller(hass):
@pytest.fixture(autouse=True) @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.""" """Auto mock zeroconf."""
@pytest.fixture(autouse=True) @pytest.fixture(autouse=True)
def auto_mock_bluetooth(mock_bluetooth): def auto_mock_bluetooth(mock_bluetooth: None) -> None:
"""Auto mock bluetooth.""" """Auto mock bluetooth."""