137 lines
4.2 KiB
Python
137 lines
4.2 KiB
Python
"""HomeKit session fixtures."""
|
|
|
|
import asyncio
|
|
from collections.abc import Generator
|
|
from contextlib import suppress
|
|
import os
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
import pytest
|
|
|
|
from homeassistant.components.device_tracker.legacy import YAML_DEVICES
|
|
from homeassistant.components.homekit.accessories import HomeDriver
|
|
from homeassistant.components.homekit.const import BRIDGE_NAME, EVENT_HOMEKIT_CHANGED
|
|
from homeassistant.components.homekit.iidmanager import AccessoryIIDStorage
|
|
from homeassistant.core import Event, HomeAssistant
|
|
|
|
from tests.common import async_capture_events
|
|
|
|
|
|
@pytest.fixture
|
|
def iid_storage(hass: HomeAssistant) -> Generator[AccessoryIIDStorage]:
|
|
"""Mock the iid storage."""
|
|
with patch.object(AccessoryIIDStorage, "_async_schedule_save"):
|
|
yield AccessoryIIDStorage(hass, "")
|
|
|
|
|
|
@pytest.fixture
|
|
def run_driver(
|
|
hass: HomeAssistant, iid_storage: AccessoryIIDStorage
|
|
) -> Generator[HomeDriver]:
|
|
"""Return a custom AccessoryDriver instance for HomeKit accessory init.
|
|
|
|
This mock does not mock async_stop, so the driver will not be stopped
|
|
"""
|
|
event_loop = asyncio.get_event_loop()
|
|
with (
|
|
patch("pyhap.accessory_driver.AsyncZeroconf"),
|
|
patch("pyhap.accessory_driver.AccessoryEncoder"),
|
|
patch("pyhap.accessory_driver.HAPServer"),
|
|
patch("pyhap.accessory_driver.AccessoryDriver.publish"),
|
|
patch(
|
|
"pyhap.accessory_driver.AccessoryDriver.persist",
|
|
),
|
|
):
|
|
yield HomeDriver(
|
|
hass,
|
|
pincode=b"123-45-678",
|
|
entry_id="",
|
|
entry_title="mock entry",
|
|
bridge_name=BRIDGE_NAME,
|
|
iid_storage=iid_storage,
|
|
address="127.0.0.1",
|
|
loop=event_loop,
|
|
)
|
|
|
|
|
|
@pytest.fixture
|
|
def hk_driver(
|
|
hass: HomeAssistant, iid_storage: AccessoryIIDStorage
|
|
) -> Generator[HomeDriver]:
|
|
"""Return a custom AccessoryDriver instance for HomeKit accessory init."""
|
|
event_loop = asyncio.get_event_loop()
|
|
with (
|
|
patch("pyhap.accessory_driver.AsyncZeroconf"),
|
|
patch("pyhap.accessory_driver.AccessoryEncoder"),
|
|
patch("pyhap.accessory_driver.HAPServer.async_stop"),
|
|
patch("pyhap.accessory_driver.HAPServer.async_start"),
|
|
patch(
|
|
"pyhap.accessory_driver.AccessoryDriver.publish",
|
|
),
|
|
patch(
|
|
"pyhap.accessory_driver.AccessoryDriver.persist",
|
|
),
|
|
):
|
|
yield HomeDriver(
|
|
hass,
|
|
pincode=b"123-45-678",
|
|
entry_id="",
|
|
entry_title="mock entry",
|
|
bridge_name=BRIDGE_NAME,
|
|
iid_storage=iid_storage,
|
|
address="127.0.0.1",
|
|
loop=event_loop,
|
|
)
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_hap(
|
|
hass: HomeAssistant,
|
|
iid_storage: AccessoryIIDStorage,
|
|
mock_zeroconf: MagicMock,
|
|
) -> Generator[HomeDriver]:
|
|
"""Return a custom AccessoryDriver instance for HomeKit accessory init."""
|
|
event_loop = asyncio.get_event_loop()
|
|
with (
|
|
patch("pyhap.accessory_driver.AsyncZeroconf"),
|
|
patch("pyhap.accessory_driver.AccessoryEncoder"),
|
|
patch("pyhap.accessory_driver.HAPServer.async_stop"),
|
|
patch("pyhap.accessory_driver.HAPServer.async_start"),
|
|
patch(
|
|
"pyhap.accessory_driver.AccessoryDriver.publish",
|
|
),
|
|
patch(
|
|
"pyhap.accessory_driver.AccessoryDriver.async_start",
|
|
),
|
|
patch(
|
|
"pyhap.accessory_driver.AccessoryDriver.async_stop",
|
|
),
|
|
patch(
|
|
"pyhap.accessory_driver.AccessoryDriver.persist",
|
|
),
|
|
):
|
|
yield HomeDriver(
|
|
hass,
|
|
pincode=b"123-45-678",
|
|
entry_id="",
|
|
entry_title="mock entry",
|
|
bridge_name=BRIDGE_NAME,
|
|
iid_storage=iid_storage,
|
|
address="127.0.0.1",
|
|
loop=event_loop,
|
|
)
|
|
|
|
|
|
@pytest.fixture
|
|
def events(hass: HomeAssistant) -> list[Event]:
|
|
"""Yield caught homekit_changed events."""
|
|
return async_capture_events(hass, EVENT_HOMEKIT_CHANGED)
|
|
|
|
|
|
@pytest.fixture
|
|
def demo_cleanup(hass: HomeAssistant) -> Generator[None]:
|
|
"""Clean up device tracker demo file."""
|
|
yield
|
|
with suppress(FileNotFoundError):
|
|
os.remove(hass.config.path(YAML_DEVICES))
|