88 lines
2.7 KiB
Python
88 lines
2.7 KiB
Python
"""Tests for the Mealie calendar."""
|
|
|
|
from datetime import date
|
|
from http import HTTPStatus
|
|
from unittest.mock import AsyncMock, patch
|
|
|
|
from aiomealie import MealplanResponse
|
|
from syrupy.assertion import SnapshotAssertion
|
|
|
|
from homeassistant.const import STATE_OFF, Platform
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers import entity_registry as er
|
|
|
|
from . import setup_integration
|
|
|
|
from tests.common import MockConfigEntry, snapshot_platform
|
|
from tests.typing import ClientSessionGenerator
|
|
|
|
|
|
async def test_api_calendar(
|
|
hass: HomeAssistant,
|
|
snapshot: SnapshotAssertion,
|
|
mock_mealie_client: AsyncMock,
|
|
mock_config_entry: MockConfigEntry,
|
|
hass_client: ClientSessionGenerator,
|
|
) -> None:
|
|
"""Test the API returns the calendar."""
|
|
await setup_integration(hass, mock_config_entry)
|
|
|
|
client = await hass_client()
|
|
response = await client.get("/api/calendars")
|
|
assert response.status == HTTPStatus.OK
|
|
data = await response.json()
|
|
assert data == snapshot
|
|
|
|
|
|
async def test_entities(
|
|
hass: HomeAssistant,
|
|
snapshot: SnapshotAssertion,
|
|
entity_registry: er.EntityRegistry,
|
|
mock_mealie_client: AsyncMock,
|
|
mock_config_entry: MockConfigEntry,
|
|
) -> None:
|
|
"""Test the calendar entities."""
|
|
with patch("homeassistant.components.mealie.PLATFORMS", [Platform.CALENDAR]):
|
|
await setup_integration(hass, mock_config_entry)
|
|
|
|
await snapshot_platform(hass, entity_registry, snapshot, mock_config_entry.entry_id)
|
|
|
|
|
|
async def test_no_meal_planned(
|
|
hass: HomeAssistant,
|
|
snapshot: SnapshotAssertion,
|
|
entity_registry: er.EntityRegistry,
|
|
mock_mealie_client: AsyncMock,
|
|
mock_config_entry: MockConfigEntry,
|
|
) -> None:
|
|
"""Test the calendar handles no meal planned."""
|
|
mock_mealie_client.get_mealplans.return_value = MealplanResponse([])
|
|
|
|
await setup_integration(hass, mock_config_entry)
|
|
|
|
assert hass.states.get("calendar.mealie_dinner").state == STATE_OFF
|
|
|
|
|
|
async def test_api_events(
|
|
hass: HomeAssistant,
|
|
snapshot: SnapshotAssertion,
|
|
mock_mealie_client: AsyncMock,
|
|
mock_config_entry: MockConfigEntry,
|
|
hass_client: ClientSessionGenerator,
|
|
) -> None:
|
|
"""Test the Mealie calendar view."""
|
|
await setup_integration(hass, mock_config_entry)
|
|
|
|
client = await hass_client()
|
|
response = await client.get(
|
|
"/api/calendars/calendar.mealie_dinner?start=2023-08-01&end=2023-11-01"
|
|
)
|
|
assert mock_mealie_client.get_mealplans.called == 1
|
|
assert mock_mealie_client.get_mealplans.call_args_list[1].args == (
|
|
date(2023, 8, 1),
|
|
date(2023, 11, 1),
|
|
)
|
|
assert response.status == HTTPStatus.OK
|
|
events = await response.json()
|
|
assert events == snapshot
|