core/tests/components/loqed/conftest.py

"""Contains fixtures for Loqed tests."""

from collections.abc import AsyncGenerator, Callable
from contextlib import contextmanager
import json
from typing import Any
from unittest.mock import AsyncMock, Mock, patch

from loqedAPI import loqed
import pytest

from homeassistant.components.loqed.const import CONF_CLOUDHOOK_URL, DOMAIN
from homeassistant.const import CONF_API_TOKEN, CONF_NAME, CONF_WEBHOOK_ID
from homeassistant.core import HomeAssistant
from homeassistant.setup import async_setup_component

from tests.common import MockConfigEntry, async_load_fixture

type PatchLockCreationFlow = Callable[[dict[str, Any], loqed.Lock, str], Any]


@pytest.fixture(name="config_entry")
async def config_entry_fixture(hass: HomeAssistant) -> MockConfigEntry:
    """Mock config entry."""

    config = await async_load_fixture(hass, "integration_config.json", DOMAIN)
    json_config = json.loads(config)
    return MockConfigEntry(
        version=1,
        domain=DOMAIN,
        data={
            "id": "Foo",
            "bridge_ip": json_config["bridge_ip"],
            "bridge_mdns_hostname": json_config["bridge_mdns_hostname"],
            "bridge_key": json_config["bridge_key"],
            "lock_key_local_id": int(json_config["lock_key_local_id"]),
            "lock_key_key": json_config["lock_key_key"],
            CONF_WEBHOOK_ID: "Webhook_id",
            CONF_API_TOKEN: "Token",
            CONF_NAME: "Home",
        },
    )


@pytest.fixture(name="cloud_config_entry")
async def cloud_config_entry_fixture(hass: HomeAssistant) -> MockConfigEntry:
    """Mock config entry."""

    config = await async_load_fixture(hass, "integration_config.json", DOMAIN)
    webhooks_fixture = json.loads(
        await async_load_fixture(hass, "get_all_webhooks.json", DOMAIN)
    )
    json_config = json.loads(config)
    return MockConfigEntry(
        version=1,
        domain=DOMAIN,
        data={
            "id": "Foo",
            "bridge_ip": json_config["bridge_ip"],
            "bridge_mdns_hostname": json_config["bridge_mdns_hostname"],
            "bridge_key": json_config["bridge_key"],
            "lock_key_local_id": int(json_config["lock_key_local_id"]),
            "lock_key_key": json_config["lock_key_key"],
            CONF_WEBHOOK_ID: "Webhook_id",
            CONF_API_TOKEN: "Token",
            CONF_NAME: "Home",
            CONF_CLOUDHOOK_URL: webhooks_fixture[0]["url"],
        },
    )


@pytest.fixture(name="lock")
async def lock_fixture(hass: HomeAssistant) -> loqed.Lock:
    """Set up a mock implementation of a Lock."""
    webhooks_fixture = json.loads(
        await async_load_fixture(hass, "get_all_webhooks.json", DOMAIN)
    )

    mock_lock = Mock(spec=loqed.Lock, id="Foo", last_key_id=2)
    mock_lock.name = "LOQED smart lock"
    mock_lock.getWebhooks = AsyncMock(return_value=webhooks_fixture)
    mock_lock.bolt_state = "locked"
    mock_lock.battery_percentage = 90
    return mock_lock


@pytest.fixture(name="integration")
async def integration_fixture(
    hass: HomeAssistant, config_entry: MockConfigEntry, lock: loqed.Lock
) -> AsyncGenerator[MockConfigEntry]:
    """Set up the loqed integration with a config entry."""
    config: dict[str, Any] = {DOMAIN: {CONF_API_TOKEN: ""}}
    config_entry.add_to_hass(hass)

    lock_status = json.loads(await async_load_fixture(hass, "status_ok.json", DOMAIN))

    with (
        patch("loqedAPI.loqed.LoqedAPI.async_get_lock", return_value=lock),
        patch(
            "loqedAPI.loqed.LoqedAPI.async_get_lock_details", return_value=lock_status
        ),
    ):
        await async_setup_component(hass, DOMAIN, config)
        await hass.async_block_till_done()
        yield config_entry


@pytest.fixture(name="patch_lock_creation_flow")
def patch_lock_creation_flow_fixture() -> PatchLockCreationFlow:
    """Patch config-flow calls used when creating a lock entry."""

    @contextmanager
    def _patch_lock_creation_flow(
        all_locks_response: dict[str, Any],
        lock: loqed.Lock,
        webhook_id: str,
    ) -> Any:
        with (
            patch(
                "loqedAPI.cloud_loqed.LoqedCloudAPI.async_get_locks",
                return_value=all_locks_response,
            ),
            patch(
                "loqedAPI.loqed.LoqedAPI.async_get_lock",
                return_value=lock,
            ),
            patch(
                "homeassistant.components.loqed.async_setup_entry",
                return_value=True,
            ),
            patch(
                "homeassistant.components.webhook.async_generate_id",
                return_value=webhook_id,
            ),
        ):
            yield

    return _patch_lock_creation_flow