core/tests/components/arve/conftest.py

"""Common fixtures for the Arve tests."""

from unittest.mock import AsyncMock, MagicMock, patch

from asyncarve import ArveCustomer, ArveDevices, ArveSensPro, ArveSensProData
import pytest
from typing_extensions import Generator

from homeassistant.components.arve.const import DOMAIN
from homeassistant.core import HomeAssistant

from . import USER_INPUT

from tests.common import MockConfigEntry


@pytest.fixture
def mock_setup_entry() -> Generator[AsyncMock]:
    """Override async_setup_entry."""
    with patch(
        "homeassistant.components.arve.async_setup_entry", return_value=True
    ) as mock_setup_entry:
        yield mock_setup_entry


@pytest.fixture
def mock_config_entry(hass: HomeAssistant, mock_arve: MagicMock) -> MockConfigEntry:
    """Return the default mocked config entry."""
    return MockConfigEntry(
        title="Arve", domain=DOMAIN, data=USER_INPUT, unique_id=mock_arve.customer_id
    )


@pytest.fixture
def mock_arve():
    """Return a mocked Arve client."""

    with (
        patch(
            "homeassistant.components.arve.coordinator.Arve", autospec=True
        ) as arve_mock,
        patch("homeassistant.components.arve.config_flow.Arve", new=arve_mock),
    ):
        arve = arve_mock.return_value
        arve.customer_id = 12345

        arve.get_customer_id.return_value = ArveCustomer(12345)

        arve.get_devices.return_value = ArveDevices(["test-serial-number"])
        arve.get_sensor_info.return_value = ArveSensPro("Test Sensor", "1.0", "prov1")

        arve.device_sensor_data.return_value = ArveSensProData(
            14, 595.75, 28.71, 0.16, 0.19, 26.02, 7
        )

        yield arve