bb31fc1ec7
* test storage save and load * fix bug exposed by test * refactor test * add JSON for test account/location * create helpers to load JSON * refactor test * baseline refactor * tweak * update requiremenst * rationalise code * remove conditional in test * refactor test * mypy fix * tweak tests * working test * working test 4 * working test 5 * add typed dicts * working dtms * lint * fix dtm asserts * doc strings * list * tweak conditional * tweak test data sets to extend coverage * leverage conftest.py for subsequent tests * revert test storage * revert part two * rename symbols * remove anachronism * stop unwanted DNS lookup * Clean up type ignores * Format --------- Co-authored-by: Martin Hjelmare <marhje52@gmail.com>
111 lines
4 KiB
Python
111 lines
4 KiB
Python
"""Fixtures and helpers for the evohome tests."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from datetime import datetime, timedelta
|
|
from typing import Any, Final
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
from aiohttp import ClientSession
|
|
from evohomeasync2 import EvohomeClient
|
|
from evohomeasync2.broker import Broker
|
|
import pytest
|
|
|
|
from homeassistant.components.evohome import CONF_PASSWORD, CONF_USERNAME, DOMAIN
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.setup import async_setup_component
|
|
from homeassistant.util.json import JsonArrayType, JsonObjectType
|
|
|
|
from .const import ACCESS_TOKEN, REFRESH_TOKEN
|
|
|
|
from tests.common import load_json_array_fixture, load_json_object_fixture
|
|
|
|
TEST_CONFIG: Final = {
|
|
CONF_USERNAME: "username",
|
|
CONF_PASSWORD: "password",
|
|
}
|
|
|
|
|
|
def user_account_config_fixture() -> JsonObjectType:
|
|
"""Load JSON for the config of a user's account."""
|
|
return load_json_object_fixture("user_account.json", DOMAIN)
|
|
|
|
|
|
def user_locations_config_fixture() -> JsonArrayType:
|
|
"""Load JSON for the config of a user's installation (a list of locations)."""
|
|
return load_json_array_fixture("user_locations.json", DOMAIN)
|
|
|
|
|
|
def location_status_fixture(loc_id: str) -> JsonObjectType:
|
|
"""Load JSON for the status of a specific location."""
|
|
return load_json_object_fixture(f"status_{loc_id}.json", DOMAIN)
|
|
|
|
|
|
def dhw_schedule_fixture() -> JsonObjectType:
|
|
"""Load JSON for the schedule of a domesticHotWater zone."""
|
|
return load_json_object_fixture("schedule_dhw.json", DOMAIN)
|
|
|
|
|
|
def zone_schedule_fixture() -> JsonObjectType:
|
|
"""Load JSON for the schedule of a temperatureZone zone."""
|
|
return load_json_object_fixture("schedule_zone.json", DOMAIN)
|
|
|
|
|
|
async def mock_get(
|
|
self: Broker, url: str, **kwargs: Any
|
|
) -> JsonArrayType | JsonObjectType:
|
|
"""Return the JSON for a HTTP get of a given URL."""
|
|
|
|
# a proxy for the behaviour of the real web API
|
|
if self.refresh_token is None:
|
|
self.refresh_token = f"new_{REFRESH_TOKEN}"
|
|
|
|
if self.access_token_expires is None or self.access_token_expires < datetime.now():
|
|
self.access_token = f"new_{ACCESS_TOKEN}"
|
|
self.access_token_expires = datetime.now() + timedelta(minutes=30)
|
|
|
|
# assume a valid GET, and return the JSON for that web API
|
|
if url == "userAccount": # userAccount
|
|
return user_account_config_fixture()
|
|
|
|
if url.startswith("location"):
|
|
if "installationInfo" in url: # location/installationInfo?userId={id}
|
|
return user_locations_config_fixture()
|
|
if "location" in url: # location/{id}/status
|
|
return location_status_fixture("2738909")
|
|
|
|
elif "schedule" in url:
|
|
if url.startswith("domesticHotWater"): # domesticHotWater/{id}/schedule
|
|
return dhw_schedule_fixture()
|
|
if url.startswith("temperatureZone"): # temperatureZone/{id}/schedule
|
|
return zone_schedule_fixture()
|
|
|
|
pytest.xfail(f"Unexpected URL: {url}")
|
|
|
|
|
|
@patch("evohomeasync2.broker.Broker.get", mock_get)
|
|
async def setup_evohome(hass: HomeAssistant, test_config: dict[str, str]) -> MagicMock:
|
|
"""Set up the evohome integration and return its client.
|
|
|
|
The class is mocked here to check the client was instantiated with the correct args.
|
|
"""
|
|
|
|
with (
|
|
patch("homeassistant.components.evohome.evo.EvohomeClient") as mock_client,
|
|
patch("homeassistant.components.evohome.ev1.EvohomeClient", return_value=None),
|
|
):
|
|
mock_client.side_effect = EvohomeClient
|
|
|
|
assert await async_setup_component(hass, DOMAIN, {DOMAIN: test_config})
|
|
await hass.async_block_till_done()
|
|
|
|
mock_client.assert_called_once()
|
|
|
|
assert mock_client.call_args.args[0] == test_config[CONF_USERNAME]
|
|
assert mock_client.call_args.args[1] == test_config[CONF_PASSWORD]
|
|
|
|
assert isinstance(mock_client.call_args.kwargs["session"], ClientSession)
|
|
|
|
assert mock_client.account_info is not None
|
|
|
|
return mock_client
|