core/tests/components/powerfox/conftest.py

"""Common fixtures for the Powerfox tests."""

from collections.abc import Generator
from datetime import UTC, datetime
from unittest.mock import AsyncMock, patch

from powerfox import Device, DeviceType, HeatMeter, PowerMeter, WaterMeter
import pytest

from homeassistant.components.powerfox.const import DOMAIN
from homeassistant.const import CONF_EMAIL, CONF_PASSWORD

from tests.common import MockConfigEntry


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


@pytest.fixture
def mock_powerfox_client() -> Generator[AsyncMock]:
    """Mock a Powerfox client."""
    with (
        patch(
            "homeassistant.components.powerfox.Powerfox",
            autospec=True,
        ) as mock_client,
        patch(
            "homeassistant.components.powerfox.config_flow.Powerfox",
            new=mock_client,
        ),
    ):
        client = mock_client.return_value
        client.all_devices.return_value = [
            Device(
                id="9x9x1f12xx3x",
                date_added=datetime(2024, 11, 26, 9, 22, 35, tzinfo=UTC),
                main_device=True,
                bidirectional=True,
                type=DeviceType.POWER_METER,
                name="Poweropti",
            ),
            Device(
                id="9x9x1f12xx4x",
                date_added=datetime(2024, 11, 26, 9, 22, 35, tzinfo=UTC),
                main_device=False,
                bidirectional=False,
                type=DeviceType.COLD_WATER_METER,
                name="Wateropti",
            ),
            Device(
                id="9x9x1f12xx5x",
                date_added=datetime(2024, 11, 26, 9, 22, 35, tzinfo=UTC),
                main_device=False,
                bidirectional=False,
                type=DeviceType.HEAT_METER,
                name="Heatopti",
            ),
        ]
        client.device.side_effect = [
            PowerMeter(
                outdated=False,
                timestamp=datetime(2024, 11, 26, 10, 48, 51, tzinfo=UTC),
                power=111,
                energy_usage=1111.111,
                energy_return=111.111,
                energy_usage_high_tariff=111.111,
                energy_usage_low_tariff=111.111,
            ),
            WaterMeter(
                outdated=False,
                timestamp=datetime(2024, 11, 26, 10, 48, 51, tzinfo=UTC),
                cold_water=1111.111,
                warm_water=0.0,
            ),
            HeatMeter(
                outdated=False,
                timestamp=datetime(2024, 11, 26, 10, 48, 51, tzinfo=UTC),
                total_energy=1111.111,
                delta_energy=111,
                total_volume=1111.111,
                delta_volume=0.111,
            ),
        ]
        yield client


@pytest.fixture
def mock_config_entry() -> MockConfigEntry:
    """Mock a Powerfox config entry."""
    return MockConfigEntry(
        domain=DOMAIN,
        title="Powerfox",
        data={
            CONF_EMAIL: "test@powerfox.test",
            CONF_PASSWORD: "test-password",
        },
    )