mirror of
https://github.com/home-assistant/core.git
synced 2025-04-25 17:57:55 +00:00
4050c216ed
* Add remote_calendar with storage * Use coordinator and remove storage * cleanup * cleanup * remove init from config_flow * add some tests * some fixes * test-before-setup * fix error handling * remove unneeded code * fix updates * load calendar in the event loop * allow redirects * test_update_failed * tests * address review * use error from local_calendar * adress more comments * remove unique_id * add unique entity_id * add excemption * abort_entries_match * unique_id * add , * cleanup * deduplicate call * don't raise for status end de-nest * multiline * test * tests * use raise_for_status again * use respx * just use config_entry argument that already is defined * Also assert on the config entry result title and data * improve config_flow * update quality scale * address review --------- Co-authored-by: Allen Porter <allen@thebends.org>
90 lines
2.7 KiB
Python
90 lines
2.7 KiB
Python
"""Fixtures for Remote Calendar."""
|
|
|
|
from collections.abc import Awaitable, Callable
|
|
from http import HTTPStatus
|
|
import textwrap
|
|
from typing import Any
|
|
import urllib
|
|
|
|
import pytest
|
|
|
|
from homeassistant.components.remote_calendar.const import CONF_CALENDAR_NAME, DOMAIN
|
|
from homeassistant.const import CONF_URL
|
|
from homeassistant.core import HomeAssistant
|
|
|
|
from tests.common import MockConfigEntry
|
|
from tests.typing import ClientSessionGenerator
|
|
|
|
CALENDAR_NAME = "Home Assistant Events"
|
|
TEST_ENTITY = "calendar.home_assistant_events"
|
|
CALENDER_URL = "https://some.calendar.com/calendar.ics"
|
|
FRIENDLY_NAME = "Home Assistant Events"
|
|
|
|
|
|
@pytest.fixture(name="time_zone")
|
|
def mock_time_zone() -> str:
|
|
"""Fixture for time zone to use in tests."""
|
|
# Set our timezone to CST/Regina so we can check calculations
|
|
# This keeps UTC-6 all year round
|
|
return "America/Regina"
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
async def set_time_zone(hass: HomeAssistant, time_zone: str):
|
|
"""Set the time zone for the tests."""
|
|
# Set our timezone to CST/Regina so we can check calculations
|
|
# This keeps UTC-6 all year round
|
|
await hass.config.async_set_time_zone(time_zone)
|
|
|
|
|
|
@pytest.fixture(name="config_entry")
|
|
def mock_config_entry() -> MockConfigEntry:
|
|
"""Fixture for mock configuration entry."""
|
|
return MockConfigEntry(
|
|
domain=DOMAIN, data={CONF_CALENDAR_NAME: CALENDAR_NAME, CONF_URL: CALENDER_URL}
|
|
)
|
|
|
|
|
|
type GetEventsFn = Callable[[str, str], Awaitable[list[dict[str, Any]]]]
|
|
|
|
|
|
@pytest.fixture(name="get_events")
|
|
def get_events_fixture(hass_client: ClientSessionGenerator) -> GetEventsFn:
|
|
"""Fetch calendar events from the HTTP API."""
|
|
|
|
async def _fetch(start: str, end: str) -> list[dict[str, Any]]:
|
|
client = await hass_client()
|
|
response = await client.get(
|
|
f"/api/calendars/{TEST_ENTITY}?start={urllib.parse.quote(start)}&end={urllib.parse.quote(end)}"
|
|
)
|
|
assert response.status == HTTPStatus.OK
|
|
return await response.json()
|
|
|
|
return _fetch
|
|
|
|
|
|
def event_fields(data: dict[str, str]) -> dict[str, str]:
|
|
"""Filter event API response to minimum fields."""
|
|
return {
|
|
k: data[k]
|
|
for k in ("summary", "start", "end", "recurrence_id", "location")
|
|
if data.get(k)
|
|
}
|
|
|
|
|
|
@pytest.fixture(name="ics_content")
|
|
def mock_ics_content(request: pytest.FixtureRequest) -> str:
|
|
"""Fixture to allow tests to set initial ics content for the calendar store."""
|
|
default_content = textwrap.dedent(
|
|
"""\
|
|
BEGIN:VCALENDAR
|
|
BEGIN:VEVENT
|
|
SUMMARY:Bastille Day Party
|
|
DTSTART:19970714T170000Z
|
|
DTEND:19970715T040000Z
|
|
END:VEVENT
|
|
END:VCALENDAR
|
|
"""
|
|
)
|
|
return request.param if hasattr(request, "param") else default_content
|