83 lines
2.6 KiB
Python
83 lines
2.6 KiB
Python
|
"""Tests for the Twente Milieu calendar."""
|
||
|
|
from http import HTTPStatus
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
from homeassistant.components.twentemilieu.const import DOMAIN
|
||
|
|
from homeassistant.const import ATTR_ICON, STATE_OFF
|
||
|
|
from homeassistant.core import HomeAssistant
|
||
|
|
from homeassistant.helpers import device_registry as dr, entity_registry as er
|
||
|
|
|
||
|
|
from tests.common import MockConfigEntry
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.freeze_time("2022-01-05 00:00:00+00:00")
|
||
|
|
async def test_waste_pickup_calendar(
|
||
|
|
hass: HomeAssistant,
|
||
|
|
init_integration: MockConfigEntry,
|
||
|
|
) -> None:
|
||
|
|
"""Test the Twente Milieu waste pickup calendar."""
|
||
|
|
entity_registry = er.async_get(hass)
|
||
|
|
device_registry = dr.async_get(hass)
|
||
|
|
|
||
|
|
state = hass.states.get("calendar.twente_milieu")
|
||
|
|
entry = entity_registry.async_get("calendar.twente_milieu")
|
||
|
|
assert entry
|
||
|
|
assert state
|
||
|
|
assert entry.unique_id == "12345"
|
||
|
|
assert state.attributes[ATTR_ICON] == "mdi:delete-empty"
|
||
|
|
assert state.attributes["all_day"] is True
|
||
|
|
assert state.attributes["message"] == "Christmas Tree Pickup"
|
||
|
|
assert not state.attributes["location"]
|
||
|
|
assert not state.attributes["description"]
|
||
|
|
assert state.state == STATE_OFF
|
||
|
|
|
||
|
|
assert entry.device_id
|
||
|
|
device_entry = device_registry.async_get(entry.device_id)
|
||
|
|
assert device_entry
|
||
|
|
assert device_entry.identifiers == {(DOMAIN, "12345")}
|
||
|
|
assert device_entry.manufacturer == "Twente Milieu"
|
||
|
|
assert device_entry.name == "Twente Milieu"
|
||
|
|
assert device_entry.entry_type is dr.DeviceEntryType.SERVICE
|
||
|
|
assert device_entry.configuration_url == "https://www.twentemilieu.nl"
|
||
|
|
assert not device_entry.model
|
||
|
|
assert not device_entry.sw_version
|
||
|
|
|
||
|
|
|
||
|
|
async def test_api_calendar(
|
||
|
|
hass: HomeAssistant,
|
||
|
|
init_integration: MockConfigEntry,
|
||
|
|
hass_client,
|
||
|
|
) -> None:
|
||
|
|
"""Test the API returns the calendar."""
|
||
|
|
client = await hass_client()
|
||
|
|
response = await client.get("/api/calendars")
|
||
|
|
assert response.status == HTTPStatus.OK
|
||
|
|
data = await response.json()
|
||
|
|
assert data == [
|
||
|
|
{
|
||
|
|
"entity_id": "calendar.twente_milieu",
|
||
|
|
"name": "Twente Milieu",
|
||
|
|
}
|
||
|
|
]
|
||
|
|
|
||
|
|
|
||
|
|
async def test_api_events(
|
||
|
|
hass: HomeAssistant,
|
||
|
|
init_integration: MockConfigEntry,
|
||
|
|
hass_client,
|
||
|
|
) -> None:
|
||
|
|
"""Test the Twente Milieu calendar view."""
|
||
|
|
client = await hass_client()
|
||
|
|
response = await client.get(
|
||
|
|
"/api/calendars/calendar.twente_milieu?start=2022-01-05&end=2022-01-06"
|
||
|
|
)
|
||
|
|
assert response.status == HTTPStatus.OK
|
||
|
|
events = await response.json()
|
||
|
|
assert len(events) == 1
|
||
|
|
assert events[0] == {
|
||
|
|
"start": {"date": "2022-01-06"},
|
||
|
|
"end": {"date": "2022-01-06"},
|
||
|
|
"summary": "Christmas Tree Pickup",
|
||
|
|
}
|