858fb1fa37
* Redo Binary Sensors * Redo Button * Redo Climate * Stage unfixed platforms * Redo Cover * Redo device tracker * Redo lock * Redo Media Player * Redo Number * Redo Select * Redo Sensor * Redo Switch * Redo Update * Fix setup_platform * Add mixing snapshot * Fix config flow * Centralise entity testing * Update snapshot * Rename test_entities * Fix assert_entities
84 lines
2.8 KiB
Python
84 lines
2.8 KiB
Python
"""Tessie common helpers for tests."""
|
|
|
|
from http import HTTPStatus
|
|
from unittest.mock import patch
|
|
|
|
from aiohttp import ClientConnectionError, ClientResponseError
|
|
from aiohttp.client import RequestInfo
|
|
from syrupy import SnapshotAssertion
|
|
|
|
from homeassistant.components.tessie.const import DOMAIN, TessieStatus
|
|
from homeassistant.const import CONF_ACCESS_TOKEN, Platform
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers import entity_registry as er
|
|
|
|
from tests.common import MockConfigEntry, load_json_object_fixture
|
|
|
|
TEST_STATE_OF_ALL_VEHICLES = load_json_object_fixture("vehicles.json", DOMAIN)
|
|
TEST_VEHICLE_STATE_ONLINE = load_json_object_fixture("online.json", DOMAIN)
|
|
TEST_VEHICLE_STATUS_AWAKE = {"status": TessieStatus.AWAKE}
|
|
TEST_VEHICLE_STATUS_ASLEEP = {"status": TessieStatus.ASLEEP}
|
|
|
|
TEST_RESPONSE = {"result": True}
|
|
TEST_RESPONSE_ERROR = {"result": False, "reason": "reason why"}
|
|
|
|
TEST_CONFIG = {CONF_ACCESS_TOKEN: "1234567890"}
|
|
TESSIE_URL = "https://api.tessie.com/"
|
|
|
|
TEST_REQUEST_INFO = RequestInfo(
|
|
url=TESSIE_URL, method="GET", headers={}, real_url=TESSIE_URL
|
|
)
|
|
|
|
ERROR_AUTH = ClientResponseError(
|
|
request_info=TEST_REQUEST_INFO, history=None, status=HTTPStatus.UNAUTHORIZED
|
|
)
|
|
ERROR_TIMEOUT = ClientResponseError(
|
|
request_info=TEST_REQUEST_INFO, history=None, status=HTTPStatus.REQUEST_TIMEOUT
|
|
)
|
|
ERROR_UNKNOWN = ClientResponseError(
|
|
request_info=TEST_REQUEST_INFO, history=None, status=HTTPStatus.BAD_REQUEST
|
|
)
|
|
ERROR_VIRTUAL_KEY = ClientResponseError(
|
|
request_info=TEST_REQUEST_INFO,
|
|
history=None,
|
|
status=HTTPStatus.INTERNAL_SERVER_ERROR,
|
|
)
|
|
ERROR_CONNECTION = ClientConnectionError()
|
|
|
|
|
|
async def setup_platform(
|
|
hass: HomeAssistant, platforms: list[Platform] = [], side_effect=None
|
|
) -> MockConfigEntry:
|
|
"""Set up the Tessie platform."""
|
|
|
|
mock_entry = MockConfigEntry(
|
|
domain=DOMAIN,
|
|
data=TEST_CONFIG,
|
|
)
|
|
mock_entry.add_to_hass(hass)
|
|
|
|
with patch(
|
|
"homeassistant.components.tessie.get_state_of_all_vehicles",
|
|
return_value=TEST_STATE_OF_ALL_VEHICLES,
|
|
side_effect=side_effect,
|
|
), patch("homeassistant.components.tessie.PLATFORMS", platforms):
|
|
await hass.config_entries.async_setup(mock_entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
return mock_entry
|
|
|
|
|
|
def assert_entities(
|
|
hass: HomeAssistant,
|
|
entry_id: str,
|
|
entity_registry: er.EntityRegistry,
|
|
snapshot: SnapshotAssertion,
|
|
) -> None:
|
|
"""Test that all entities match their snapshot."""
|
|
entity_entries = er.async_entries_for_config_entry(entity_registry, entry_id)
|
|
|
|
assert entity_entries
|
|
for entity_entry in entity_entries:
|
|
assert entity_entry == snapshot(name=f"{entity_entry.entity_id}-entry")
|
|
assert (state := hass.states.get(entity_entry.entity_id))
|
|
assert state == snapshot(name=f"{entity_entry.entity_id}-state")
|