mirror of
https://github.com/home-assistant/core.git
synced 2025-04-25 09:47:52 +00:00
106 lines
3.0 KiB
Python
106 lines
3.0 KiB
Python
"""Flick Electric tests configuration."""
|
|
|
|
from collections.abc import Generator
|
|
from unittest.mock import AsyncMock, patch
|
|
|
|
import json_api_doc
|
|
from pyflick import FlickPrice
|
|
import pytest
|
|
|
|
from homeassistant.components.flick_electric.const import CONF_ACCOUNT_ID, DOMAIN
|
|
from homeassistant.const import CONF_PASSWORD, CONF_USERNAME
|
|
|
|
from . import CONF
|
|
|
|
from tests.common import MockConfigEntry, load_json_value_fixture
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_config_entry() -> MockConfigEntry:
|
|
"""Mock a config entry."""
|
|
return MockConfigEntry(
|
|
domain=DOMAIN,
|
|
title="123 Fake Street, Newtown, Wellington 6021",
|
|
data={**CONF},
|
|
version=2,
|
|
entry_id="974e52a5c0724d17b7ed876dd6ff4bc8",
|
|
unique_id=CONF[CONF_ACCOUNT_ID],
|
|
)
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_old_config_entry() -> MockConfigEntry:
|
|
"""Mock an outdated config entry."""
|
|
return MockConfigEntry(
|
|
domain=DOMAIN,
|
|
data={
|
|
CONF_USERNAME: CONF[CONF_USERNAME],
|
|
CONF_PASSWORD: CONF[CONF_PASSWORD],
|
|
},
|
|
title=CONF[CONF_USERNAME],
|
|
unique_id=CONF[CONF_USERNAME],
|
|
version=1,
|
|
)
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_flick_client() -> Generator[AsyncMock]:
|
|
"""Mock a Flick Electric client."""
|
|
with (
|
|
patch(
|
|
"homeassistant.components.flick_electric.FlickAPI",
|
|
autospec=True,
|
|
) as mock_api,
|
|
patch(
|
|
"homeassistant.components.flick_electric.config_flow.FlickAPI",
|
|
new=mock_api,
|
|
),
|
|
patch(
|
|
"homeassistant.components.flick_electric.config_flow.SimpleFlickAuth.async_get_access_token",
|
|
return_value="123456789abcdef",
|
|
),
|
|
):
|
|
api = mock_api.return_value
|
|
|
|
api.getCustomerAccounts.return_value = json_api_doc.deserialize(
|
|
load_json_value_fixture("accounts.json", DOMAIN)
|
|
)
|
|
api.getPricing.return_value = FlickPrice(
|
|
json_api_doc.deserialize(
|
|
load_json_value_fixture("rated_period.json", DOMAIN)
|
|
)
|
|
)
|
|
|
|
yield api
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_flick_client_multiple() -> Generator[AsyncMock]:
|
|
"""Mock a Flick Electric with multiple accounts."""
|
|
with (
|
|
patch(
|
|
"homeassistant.components.flick_electric.FlickAPI",
|
|
autospec=True,
|
|
) as mock_api,
|
|
patch(
|
|
"homeassistant.components.flick_electric.config_flow.FlickAPI",
|
|
new=mock_api,
|
|
),
|
|
patch(
|
|
"homeassistant.components.flick_electric.config_flow.SimpleFlickAuth.async_get_access_token",
|
|
return_value="123456789abcdef",
|
|
),
|
|
):
|
|
api = mock_api.return_value
|
|
|
|
api.getCustomerAccounts.return_value = json_api_doc.deserialize(
|
|
load_json_value_fixture("accounts_multi.json", DOMAIN)
|
|
)
|
|
api.getPricing.return_value = FlickPrice(
|
|
json_api_doc.deserialize(
|
|
load_json_value_fixture("rated_period.json", DOMAIN)
|
|
)
|
|
)
|
|
|
|
yield api
|