mirror of https://github.com/home-assistant/core
112 lines
4.0 KiB
Python
112 lines
4.0 KiB
Python
"""Tests for diagnostics platform of local calendar."""
|
|
|
|
from aiohttp.test_utils import TestClient
|
|
from freezegun import freeze_time
|
|
import pytest
|
|
from syrupy.assertion import SnapshotAssertion
|
|
|
|
from homeassistant.auth.models import Credentials
|
|
from homeassistant.core import HomeAssistant
|
|
|
|
from .conftest import TEST_ENTITY, Client
|
|
|
|
from tests.common import CLIENT_ID, MockConfigEntry, MockUser
|
|
from tests.components.diagnostics import get_diagnostics_for_config_entry
|
|
from tests.typing import ClientSessionGenerator, WebSocketGenerator
|
|
|
|
|
|
async def generate_new_hass_access_token(
|
|
hass: HomeAssistant, hass_admin_user: MockUser, hass_admin_credential: Credentials
|
|
) -> str:
|
|
"""Return an access token to access Home Assistant."""
|
|
await hass.auth.async_link_user(hass_admin_user, hass_admin_credential)
|
|
|
|
refresh_token = await hass.auth.async_create_refresh_token(
|
|
hass_admin_user, CLIENT_ID, credential=hass_admin_credential
|
|
)
|
|
return hass.auth.async_create_access_token(refresh_token)
|
|
|
|
|
|
def _get_test_client_generator(
|
|
hass: HomeAssistant, aiohttp_client: ClientSessionGenerator, new_token: str
|
|
):
|
|
"""Return a test client generator.""."""
|
|
|
|
async def auth_client() -> TestClient:
|
|
return await aiohttp_client(
|
|
hass.http.app, headers={"Authorization": f"Bearer {new_token}"}
|
|
)
|
|
|
|
return auth_client
|
|
|
|
|
|
@freeze_time("2023-03-13 12:05:00-07:00")
|
|
@pytest.mark.usefixtures("socket_enabled")
|
|
async def test_empty_calendar(
|
|
hass: HomeAssistant,
|
|
setup_integration: None,
|
|
hass_admin_user: MockUser,
|
|
hass_admin_credential: Credentials,
|
|
config_entry: MockConfigEntry,
|
|
aiohttp_client: ClientSessionGenerator,
|
|
snapshot: SnapshotAssertion,
|
|
) -> None:
|
|
"""Test diagnostics against an empty calendar."""
|
|
# Since we are freezing time only when we enter this test, we need to
|
|
# manually create a new token and clients since the token created by
|
|
# the fixtures would not be valid.
|
|
#
|
|
# Ideally we would use pytest.mark.freeze_time before the fixtures, but that does not
|
|
# work with the ical library and freezegun because
|
|
# `TypeError: '<' not supported between instances of 'FakeDatetimeMeta' and 'FakeDateMeta'`
|
|
new_token = await generate_new_hass_access_token(
|
|
hass, hass_admin_user, hass_admin_credential
|
|
)
|
|
data = await get_diagnostics_for_config_entry(
|
|
hass, _get_test_client_generator(hass, aiohttp_client, new_token), config_entry
|
|
)
|
|
assert data == snapshot
|
|
|
|
|
|
@freeze_time("2023-03-13 12:05:00-07:00")
|
|
@pytest.mark.usefixtures("socket_enabled")
|
|
async def test_api_date_time_event(
|
|
hass: HomeAssistant,
|
|
setup_integration: None,
|
|
hass_admin_user: MockUser,
|
|
hass_admin_credential: Credentials,
|
|
config_entry: MockConfigEntry,
|
|
hass_ws_client: WebSocketGenerator,
|
|
aiohttp_client: ClientSessionGenerator,
|
|
snapshot: SnapshotAssertion,
|
|
) -> None:
|
|
"""Test an event with a start/end date time."""
|
|
# Since we are freezing time only when we enter this test, we need to
|
|
# manually create a new token and clients since the token created by
|
|
# the fixtures would not be valid.
|
|
#
|
|
# Ideally we would use pytest.mark.freeze_time before the fixtures, but that does not
|
|
# work with the ical library and freezegun because
|
|
# `TypeError: '<' not supported between instances of 'FakeDatetimeMeta' and 'FakeDateMeta'`
|
|
new_token = await generate_new_hass_access_token(
|
|
hass, hass_admin_user, hass_admin_credential
|
|
)
|
|
client = Client(await hass_ws_client(hass, access_token=new_token))
|
|
await client.cmd_result(
|
|
"create",
|
|
{
|
|
"entity_id": TEST_ENTITY,
|
|
"event": {
|
|
"summary": "Bastille Day Party",
|
|
"dtstart": "1997-07-14T17:00:00+00:00",
|
|
"dtend": "1997-07-15T04:00:00+00:00",
|
|
"rrule": "FREQ=DAILY",
|
|
},
|
|
},
|
|
)
|
|
|
|
data = await get_diagnostics_for_config_entry(
|
|
hass, _get_test_client_generator(hass, aiohttp_client, new_token), config_entry
|
|
)
|
|
assert data == snapshot
|