99 lines
3.0 KiB
Python
99 lines
3.0 KiB
Python
"""Provide common fixtures."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import asyncio
|
|
from collections.abc import AsyncGenerator
|
|
from unittest.mock import AsyncMock, MagicMock, patch
|
|
|
|
from matter_server.client.models.node import MatterNode
|
|
from matter_server.common.const import SCHEMA_VERSION
|
|
from matter_server.common.models import ServerInfoMessage
|
|
import pytest
|
|
|
|
from homeassistant.core import HomeAssistant
|
|
|
|
from .common import setup_integration_with_node_fixture
|
|
|
|
from tests.common import MockConfigEntry
|
|
|
|
MOCK_FABRIC_ID = 12341234
|
|
MOCK_COMPR_FABRIC_ID = 1234
|
|
|
|
|
|
@pytest.fixture(name="matter_client")
|
|
async def matter_client_fixture() -> AsyncGenerator[MagicMock]:
|
|
"""Fixture for a Matter client."""
|
|
with patch(
|
|
"homeassistant.components.matter.MatterClient", autospec=True
|
|
) as client_class:
|
|
client = client_class.return_value
|
|
|
|
async def connect() -> None:
|
|
"""Mock connect."""
|
|
await asyncio.sleep(0)
|
|
|
|
async def listen(init_ready: asyncio.Event | None) -> None:
|
|
"""Mock listen."""
|
|
if init_ready is not None:
|
|
init_ready.set()
|
|
listen_block = asyncio.Event()
|
|
await listen_block.wait()
|
|
pytest.fail("Listen was not cancelled!")
|
|
|
|
client.connect = AsyncMock(side_effect=connect)
|
|
client.start_listening = AsyncMock(side_effect=listen)
|
|
client.server_info = ServerInfoMessage(
|
|
fabric_id=MOCK_FABRIC_ID,
|
|
compressed_fabric_id=MOCK_COMPR_FABRIC_ID,
|
|
schema_version=1,
|
|
sdk_version="2022.11.1",
|
|
wifi_credentials_set=True,
|
|
thread_credentials_set=True,
|
|
min_supported_schema_version=SCHEMA_VERSION,
|
|
bluetooth_enabled=False,
|
|
)
|
|
|
|
yield client
|
|
|
|
|
|
@pytest.fixture(name="integration")
|
|
async def integration_fixture(
|
|
hass: HomeAssistant, matter_client: MagicMock
|
|
) -> MockConfigEntry:
|
|
"""Set up the Matter integration."""
|
|
entry = MockConfigEntry(domain="matter", data={"url": "ws://localhost:5580/ws"})
|
|
entry.add_to_hass(hass)
|
|
await hass.config_entries.async_setup(entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
return entry
|
|
|
|
|
|
@pytest.fixture(name="door_lock")
|
|
async def door_lock_fixture(
|
|
hass: HomeAssistant, matter_client: MagicMock
|
|
) -> MatterNode:
|
|
"""Fixture for a door lock node."""
|
|
return await setup_integration_with_node_fixture(hass, "door-lock", matter_client)
|
|
|
|
|
|
@pytest.fixture(name="door_lock_with_unbolt")
|
|
async def door_lock_with_unbolt_fixture(
|
|
hass: HomeAssistant, matter_client: MagicMock
|
|
) -> MatterNode:
|
|
"""Fixture for a door lock node with unbolt feature."""
|
|
return await setup_integration_with_node_fixture(
|
|
hass, "door-lock-with-unbolt", matter_client
|
|
)
|
|
|
|
|
|
@pytest.fixture(name="eve_contact_sensor_node")
|
|
async def eve_contact_sensor_node_fixture(
|
|
hass: HomeAssistant, matter_client: MagicMock
|
|
) -> MatterNode:
|
|
"""Fixture for a contact sensor node."""
|
|
return await setup_integration_with_node_fixture(
|
|
hass, "eve-contact-sensor", matter_client
|
|
)
|