mirror of
https://github.com/home-assistant/core.git
synced 2025-05-12 01:49:16 +00:00
6f02550ac3
* Include HKC BLE MAC in device info when available * update tests * cover * dry * dry * dry
90 lines
2.4 KiB
Python
90 lines
2.4 KiB
Python
"""HomeKit controller session fixtures."""
|
|
|
|
from collections.abc import Callable, Generator
|
|
import datetime
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
from aiohomekit.model import Transport
|
|
from aiohomekit.testing import FakeController, FakeDiscovery, FakePairing
|
|
from bleak.backends.device import BLEDevice
|
|
from freezegun import freeze_time
|
|
from freezegun.api import FrozenDateTimeFactory
|
|
import pytest
|
|
|
|
from homeassistant.util import dt as dt_util
|
|
|
|
from tests.components.light.conftest import mock_light_profiles # noqa: F401
|
|
|
|
pytest.register_assert_rewrite("tests.components.homekit_controller.common")
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def freeze_time_in_future() -> Generator[FrozenDateTimeFactory]:
|
|
"""Freeze time at a known point."""
|
|
now = dt_util.utcnow()
|
|
start_dt = datetime.datetime(now.year + 1, 1, 1, 0, 0, 0, tzinfo=now.tzinfo)
|
|
with freeze_time(start_dt) as frozen_time:
|
|
yield frozen_time
|
|
|
|
|
|
@pytest.fixture
|
|
def controller() -> Generator[FakeController]:
|
|
"""Replace aiohomekit.Controller with an instance of aiohomekit.testing.FakeController."""
|
|
instance = FakeController()
|
|
with patch(
|
|
"homeassistant.components.homekit_controller.utils.Controller",
|
|
return_value=instance,
|
|
):
|
|
yield instance
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def hk_mock_async_zeroconf(mock_async_zeroconf: MagicMock) -> None:
|
|
"""Auto mock zeroconf."""
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def auto_mock_bluetooth(mock_bluetooth: None) -> None:
|
|
"""Auto mock bluetooth."""
|
|
|
|
|
|
@pytest.fixture
|
|
def get_next_aid() -> Generator[Callable[[], int]]:
|
|
"""Generate a function that returns increasing accessory ids."""
|
|
id_counter = 0
|
|
|
|
def _get_id():
|
|
nonlocal id_counter
|
|
id_counter += 1
|
|
return id_counter
|
|
|
|
return _get_id
|
|
|
|
|
|
@pytest.fixture
|
|
def fake_ble_discovery() -> Generator[None]:
|
|
"""Fake BLE discovery."""
|
|
|
|
class FakeBLEDiscovery(FakeDiscovery):
|
|
device = BLEDevice(
|
|
address="AA:BB:CC:DD:EE:FF", name="TestDevice", rssi=-50, details=()
|
|
)
|
|
|
|
with patch("aiohomekit.testing.FakeDiscovery", FakeBLEDiscovery):
|
|
yield
|
|
|
|
|
|
@pytest.fixture
|
|
def fake_ble_pairing() -> Generator[None]:
|
|
"""Fake BLE pairing."""
|
|
|
|
class FakeBLEPairing(FakePairing):
|
|
"""Fake BLE pairing."""
|
|
|
|
@property
|
|
def transport(self):
|
|
return Transport.BLE
|
|
|
|
with patch("aiohomekit.testing.FakePairing", FakeBLEPairing):
|
|
yield
|