mirror of https://github.com/home-assistant/core
57 lines
1.6 KiB
Python
57 lines
1.6 KiB
Python
"""Common fixtures for the Arve tests."""
|
|
|
|
from collections.abc import Generator
|
|
from unittest.mock import AsyncMock, MagicMock, patch
|
|
|
|
from asyncarve import ArveCustomer, ArveDevices, ArveSensPro, ArveSensProData
|
|
import pytest
|
|
|
|
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
|