mirror of
https://github.com/home-assistant/core.git
synced 2026-07-03 12:46:09 +01:00
seanpasino
142 lines
4.5 KiB
Python
142 lines
4.5 KiB
Python
"""The tests for Sonarr calendar platform."""
|
|
|
|
from datetime import datetime
|
|
import json
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
from aiopyarr import SonarrCalendar
|
|
from freezegun.api import FrozenDateTimeFactory
|
|
import pytest
|
|
from syrupy.assertion import SnapshotAssertion
|
|
|
|
from homeassistant.components.calendar import (
|
|
DOMAIN as CALENDAR_DOMAIN,
|
|
SERVICE_GET_EVENTS,
|
|
)
|
|
from homeassistant.const import STATE_OFF, STATE_ON, Platform
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers import entity_registry as er
|
|
|
|
from tests.common import MockConfigEntry, async_load_fixture, snapshot_platform
|
|
|
|
ENTITY_ID = "calendar.sonarr"
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("now", "expected_state"),
|
|
[
|
|
("2014-01-27 01:45:00+00:00", STATE_ON), # During the episode air window
|
|
("2014-01-27 03:00:00+00:00", STATE_OFF), # After the episode aired
|
|
],
|
|
)
|
|
async def test_calendar_state(
|
|
hass: HomeAssistant,
|
|
freezer: FrozenDateTimeFactory,
|
|
mock_config_entry: MockConfigEntry,
|
|
mock_sonarr: MagicMock,
|
|
now: str,
|
|
expected_state: str,
|
|
) -> None:
|
|
"""Test the Sonarr calendar entity state."""
|
|
await hass.config.async_set_time_zone("UTC")
|
|
freezer.move_to(now)
|
|
|
|
mock_config_entry.add_to_hass(hass)
|
|
await hass.config_entries.async_setup(mock_config_entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
state = hass.states.get(ENTITY_ID)
|
|
assert state
|
|
assert state.state == expected_state
|
|
|
|
|
|
async def test_calendar_attributes(
|
|
hass: HomeAssistant,
|
|
freezer: FrozenDateTimeFactory,
|
|
mock_config_entry: MockConfigEntry,
|
|
mock_sonarr: MagicMock,
|
|
entity_registry: er.EntityRegistry,
|
|
snapshot: SnapshotAssertion,
|
|
) -> None:
|
|
"""Test the Sonarr calendar entity attributes for the active event."""
|
|
await hass.config.async_set_time_zone("UTC")
|
|
freezer.move_to("2014-01-27 01:45:00+00:00")
|
|
|
|
mock_config_entry.add_to_hass(hass)
|
|
with patch("homeassistant.components.sonarr.PLATFORMS", [Platform.CALENDAR]):
|
|
await hass.config_entries.async_setup(mock_config_entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
await snapshot_platform(hass, entity_registry, snapshot, mock_config_entry.entry_id)
|
|
|
|
|
|
async def test_calendar_get_events(
|
|
hass: HomeAssistant,
|
|
mock_config_entry: MockConfigEntry,
|
|
mock_sonarr: MagicMock,
|
|
) -> None:
|
|
"""Test the Sonarr calendar get_events service."""
|
|
await hass.config.async_set_time_zone("UTC")
|
|
|
|
mock_config_entry.add_to_hass(hass)
|
|
await hass.config_entries.async_setup(mock_config_entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
response = await hass.services.async_call(
|
|
CALENDAR_DOMAIN,
|
|
SERVICE_GET_EVENTS,
|
|
{
|
|
"entity_id": ENTITY_ID,
|
|
"start_date_time": datetime(2014, 1, 1).isoformat(),
|
|
"end_date_time": datetime(2014, 2, 1).isoformat(),
|
|
},
|
|
blocking=True,
|
|
return_response=True,
|
|
)
|
|
|
|
events = response[ENTITY_ID]["events"]
|
|
assert len(events) == 1
|
|
assert (
|
|
events[0]["summary"]
|
|
== "Bob's Burgers - S04E11 - Easy Com-mercial, Easy Go-mercial"
|
|
)
|
|
assert events[0]["start"] == "2014-01-27T01:30:00+00:00"
|
|
assert events[0]["end"] == "2014-01-27T02:00:00+00:00"
|
|
assert mock_sonarr.async_get_calendar.call_args.kwargs["include_series"] is True
|
|
|
|
|
|
async def test_calendar_get_events_without_overview(
|
|
hass: HomeAssistant,
|
|
mock_config_entry: MockConfigEntry,
|
|
mock_sonarr: MagicMock,
|
|
) -> None:
|
|
"""Test that episodes without an overview are handled (real Sonarr omits it)."""
|
|
await hass.config.async_set_time_zone("UTC")
|
|
raw = json.loads(await async_load_fixture(hass, "calendar.json", "sonarr"))[0]
|
|
raw.pop("overview")
|
|
mock_sonarr.async_get_calendar.return_value = [SonarrCalendar(raw)]
|
|
|
|
mock_config_entry.add_to_hass(hass)
|
|
await hass.config_entries.async_setup(mock_config_entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
response = await hass.services.async_call(
|
|
CALENDAR_DOMAIN,
|
|
SERVICE_GET_EVENTS,
|
|
{
|
|
"entity_id": ENTITY_ID,
|
|
"start_date_time": datetime(2014, 1, 1).isoformat(),
|
|
"end_date_time": datetime(2014, 2, 1).isoformat(),
|
|
},
|
|
blocking=True,
|
|
return_response=True,
|
|
)
|
|
|
|
events = response[ENTITY_ID]["events"]
|
|
assert len(events) == 1
|
|
assert (
|
|
events[0]["summary"]
|
|
== "Bob's Burgers - S04E11 - Easy Com-mercial, Easy Go-mercial"
|
|
)
|
|
assert events[0].get("description") is None
|