jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-23 21:27:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

Rename calls fixture in calendar tests (#118353)

Browse Source
This commit is contained in:
epenet 2024-05-29 09:24:36 +02:00 committed by GitHub
parent 0888233f06
commit e488f9b87f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 43 additions and 43 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

86
tests/components/calendar/test_trigger.py
View File

@ -150,7 +150,7 @@ async def create_automation(
@pytest.fixture @pytest.fixture
def calls(hass: HomeAssistant) -> Callable[[], list[dict[str, Any]]]: def calls_data(hass: HomeAssistant) -> Callable[[], list[dict[str, Any]]]:
"""Fixture to return payload data for automation calls.""" """Fixture to return payload data for automation calls."""
service_calls = async_mock_service(hass, "test", "automation") service_calls = async_mock_service(hass, "test", "automation")
@ -172,7 +172,7 @@ def mock_update_interval() -> Generator[None, None, None]:
async def test_event_start_trigger( async def test_event_start_trigger(
hass: HomeAssistant, hass: HomeAssistant,
calls: Callable[[], list[dict[str, Any]]], calls_data: Callable[[], list[dict[str, Any]]],
fake_schedule: FakeSchedule, fake_schedule: FakeSchedule,
test_entity: MockCalendarEntity, test_entity: MockCalendarEntity,
) -> None: ) -> None:
@ -182,13 +182,13 @@ async def test_event_start_trigger(
end=datetime.datetime.fromisoformat("2022-04-19 11:30:00+00:00"), end=datetime.datetime.fromisoformat("2022-04-19 11:30:00+00:00"),
) )
async with create_automation(hass, EVENT_START): async with create_automation(hass, EVENT_START):
assert len(calls()) == 0 assert len(calls_data()) == 0
await fake_schedule.fire_until( await fake_schedule.fire_until(
datetime.datetime.fromisoformat("2022-04-19 11:15:00+00:00"), datetime.datetime.fromisoformat("2022-04-19 11:15:00+00:00"),
) )
assert calls() == [ assert calls_data() == [
{ {
"platform": "calendar", "platform": "calendar",
"event": EVENT_START, "event": EVENT_START,
@ -206,7 +206,7 @@ async def test_event_start_trigger(
) )
async def test_event_start_trigger_with_offset( async def test_event_start_trigger_with_offset(
hass: HomeAssistant, hass: HomeAssistant,
calls: Callable[[], list[dict[str, Any]]], calls_data: Callable[[], list[dict[str, Any]]],
fake_schedule: FakeSchedule, fake_schedule: FakeSchedule,
test_entity: MockCalendarEntity, test_entity: MockCalendarEntity,
offset_str, offset_str,
@ -222,13 +222,13 @@ async def test_event_start_trigger_with_offset(
await fake_schedule.fire_until( await fake_schedule.fire_until(
datetime.datetime.fromisoformat("2022-04-19 11:55:00+00:00") + offset_delta, datetime.datetime.fromisoformat("2022-04-19 11:55:00+00:00") + offset_delta,
) )
assert len(calls()) == 0 assert len(calls_data()) == 0
# Event has started w/ offset # Event has started w/ offset
await fake_schedule.fire_until( await fake_schedule.fire_until(
datetime.datetime.fromisoformat("2022-04-19 12:05:00+00:00") + offset_delta, datetime.datetime.fromisoformat("2022-04-19 12:05:00+00:00") + offset_delta,
) )
assert calls() == [ assert calls_data() == [
{ {
"platform": "calendar", "platform": "calendar",
"event": EVENT_START, "event": EVENT_START,
@ -239,7 +239,7 @@ async def test_event_start_trigger_with_offset(
async def test_event_end_trigger( async def test_event_end_trigger(
hass: HomeAssistant, hass: HomeAssistant,
calls: Callable[[], list[dict[str, Any]]], calls_data: Callable[[], list[dict[str, Any]]],
fake_schedule: FakeSchedule, fake_schedule: FakeSchedule,
test_entity: MockCalendarEntity, test_entity: MockCalendarEntity,
) -> None: ) -> None:
@ -253,13 +253,13 @@ async def test_event_end_trigger(
await fake_schedule.fire_until( await fake_schedule.fire_until(
datetime.datetime.fromisoformat("2022-04-19 11:10:00+00:00") datetime.datetime.fromisoformat("2022-04-19 11:10:00+00:00")
) )
assert len(calls()) == 0 assert len(calls_data()) == 0
# Event ends # Event ends
await fake_schedule.fire_until( await fake_schedule.fire_until(
datetime.datetime.fromisoformat("2022-04-19 12:10:00+00:00") datetime.datetime.fromisoformat("2022-04-19 12:10:00+00:00")
) )
assert calls() == [ assert calls_data() == [
{ {
"platform": "calendar", "platform": "calendar",
"event": EVENT_END, "event": EVENT_END,
@ -277,7 +277,7 @@ async def test_event_end_trigger(
) )
async def test_event_end_trigger_with_offset( async def test_event_end_trigger_with_offset(
hass: HomeAssistant, hass: HomeAssistant,
calls: Callable[[], list[dict[str, Any]]], calls_data: Callable[[], list[dict[str, Any]]],
fake_schedule: FakeSchedule, fake_schedule: FakeSchedule,
test_entity: MockCalendarEntity, test_entity: MockCalendarEntity,
offset_str, offset_str,
@ -293,13 +293,13 @@ async def test_event_end_trigger_with_offset(
await fake_schedule.fire_until( await fake_schedule.fire_until(
datetime.datetime.fromisoformat("2022-04-19 12:05:00+00:00") + offset_delta, datetime.datetime.fromisoformat("2022-04-19 12:05:00+00:00") + offset_delta,
) )
assert len(calls()) == 0 assert len(calls_data()) == 0
# Event has started w/ offset # Event has started w/ offset
await fake_schedule.fire_until( await fake_schedule.fire_until(
datetime.datetime.fromisoformat("2022-04-19 12:35:00+00:00") + offset_delta, datetime.datetime.fromisoformat("2022-04-19 12:35:00+00:00") + offset_delta,
) )
assert calls() == [ assert calls_data() == [
{ {
"platform": "calendar", "platform": "calendar",
"event": EVENT_END, "event": EVENT_END,
@ -310,7 +310,7 @@ async def test_event_end_trigger_with_offset(
async def test_calendar_trigger_with_no_events( async def test_calendar_trigger_with_no_events(
hass: HomeAssistant, hass: HomeAssistant,
calls: Callable[[], list[dict[str, Any]]], calls_data: Callable[[], list[dict[str, Any]]],
fake_schedule: FakeSchedule, fake_schedule: FakeSchedule,
) -> None: ) -> None:
"""Test a calendar trigger setup with no events.""" """Test a calendar trigger setup with no events."""
@ -320,12 +320,12 @@ async def test_calendar_trigger_with_no_events(
await fake_schedule.fire_until( await fake_schedule.fire_until(
datetime.datetime.fromisoformat("2022-04-19 11:00:00+00:00") datetime.datetime.fromisoformat("2022-04-19 11:00:00+00:00")
) )
assert len(calls()) == 0 assert len(calls_data()) == 0
async def test_multiple_start_events( async def test_multiple_start_events(
hass: HomeAssistant, hass: HomeAssistant,
calls: Callable[[], list[dict[str, Any]]], calls_data: Callable[[], list[dict[str, Any]]],
fake_schedule: FakeSchedule, fake_schedule: FakeSchedule,
test_entity: MockCalendarEntity, test_entity: MockCalendarEntity,
) -> None: ) -> None:
@ -343,7 +343,7 @@ async def test_multiple_start_events(
await fake_schedule.fire_until( await fake_schedule.fire_until(
datetime.datetime.fromisoformat("2022-04-19 11:30:00+00:00") datetime.datetime.fromisoformat("2022-04-19 11:30:00+00:00")
) )
assert calls() == [ assert calls_data() == [
{ {
"platform": "calendar", "platform": "calendar",
"event": EVENT_START, "event": EVENT_START,
@ -359,7 +359,7 @@ async def test_multiple_start_events(
async def test_multiple_end_events( async def test_multiple_end_events(
hass: HomeAssistant, hass: HomeAssistant,
calls: Callable[[], list[dict[str, Any]]], calls_data: Callable[[], list[dict[str, Any]]],
fake_schedule: FakeSchedule, fake_schedule: FakeSchedule,
test_entity: MockCalendarEntity, test_entity: MockCalendarEntity,
) -> None: ) -> None:
@ -378,7 +378,7 @@ async def test_multiple_end_events(
datetime.datetime.fromisoformat("2022-04-19 11:30:00+00:00") datetime.datetime.fromisoformat("2022-04-19 11:30:00+00:00")
) )
assert calls() == [ assert calls_data() == [
{ {
"platform": "calendar", "platform": "calendar",
"event": EVENT_END, "event": EVENT_END,
@ -394,7 +394,7 @@ async def test_multiple_end_events(
async def test_multiple_events_sharing_start_time( async def test_multiple_events_sharing_start_time(
hass: HomeAssistant, hass: HomeAssistant,
calls: Callable[[], list[dict[str, Any]]], calls_data: Callable[[], list[dict[str, Any]]],
fake_schedule: FakeSchedule, fake_schedule: FakeSchedule,
test_entity: MockCalendarEntity, test_entity: MockCalendarEntity,
) -> None: ) -> None:
@ -413,7 +413,7 @@ async def test_multiple_events_sharing_start_time(
datetime.datetime.fromisoformat("2022-04-19 11:35:00+00:00") datetime.datetime.fromisoformat("2022-04-19 11:35:00+00:00")
) )
assert calls() == [ assert calls_data() == [
{ {
"platform": "calendar", "platform": "calendar",
"event": EVENT_START, "event": EVENT_START,
@ -429,7 +429,7 @@ async def test_multiple_events_sharing_start_time(
async def test_overlap_events( async def test_overlap_events(
hass: HomeAssistant, hass: HomeAssistant,
calls: Callable[[], list[dict[str, Any]]], calls_data: Callable[[], list[dict[str, Any]]],
fake_schedule: FakeSchedule, fake_schedule: FakeSchedule,
test_entity: MockCalendarEntity, test_entity: MockCalendarEntity,
) -> None: ) -> None:
@ -448,7 +448,7 @@ async def test_overlap_events(
datetime.datetime.fromisoformat("2022-04-19 11:20:00+00:00") datetime.datetime.fromisoformat("2022-04-19 11:20:00+00:00")
) )
assert calls() == [ assert calls_data() == [
{ {
"platform": "calendar", "platform": "calendar",
"event": EVENT_START, "event": EVENT_START,
@ -506,7 +506,7 @@ async def test_legacy_entity_type(
async def test_update_next_event( async def test_update_next_event(
hass: HomeAssistant, hass: HomeAssistant,
calls: Callable[[], list[dict[str, Any]]], calls_data: Callable[[], list[dict[str, Any]]],
fake_schedule: FakeSchedule, fake_schedule: FakeSchedule,
test_entity: MockCalendarEntity, test_entity: MockCalendarEntity,
) -> None: ) -> None:
@ -521,7 +521,7 @@ async def test_update_next_event(
await fake_schedule.fire_until( await fake_schedule.fire_until(
datetime.datetime.fromisoformat("2022-04-19 10:45:00+00:00") datetime.datetime.fromisoformat("2022-04-19 10:45:00+00:00")
) )
assert len(calls()) == 0 assert len(calls_data()) == 0
# Create a new event between now and when the event fires # Create a new event between now and when the event fires
event_data2 = test_entity.create_event( event_data2 = test_entity.create_event(
@ -533,7 +533,7 @@ async def test_update_next_event(
await fake_schedule.fire_until( await fake_schedule.fire_until(
datetime.datetime.fromisoformat("2022-04-19 11:30:00+00:00") datetime.datetime.fromisoformat("2022-04-19 11:30:00+00:00")
) )
assert calls() == [ assert calls_data() == [
{ {
"platform": "calendar", "platform": "calendar",
"event": EVENT_START, "event": EVENT_START,
@ -549,7 +549,7 @@ async def test_update_next_event(
async def test_update_missed( async def test_update_missed(
hass: HomeAssistant, hass: HomeAssistant,
calls: Callable[[], list[dict[str, Any]]], calls_data: Callable[[], list[dict[str, Any]]],
fake_schedule: FakeSchedule, fake_schedule: FakeSchedule,
test_entity: MockCalendarEntity, test_entity: MockCalendarEntity,
) -> None: ) -> None:
@ -565,7 +565,7 @@ async def test_update_missed(
await fake_schedule.fire_until( await fake_schedule.fire_until(
datetime.datetime.fromisoformat("2022-04-19 10:38:00+00:00") datetime.datetime.fromisoformat("2022-04-19 10:38:00+00:00")
) )
assert len(calls()) == 0 assert len(calls_data()) == 0
test_entity.create_event( test_entity.create_event(
start=datetime.datetime.fromisoformat("2022-04-19 10:40:00+00:00"), start=datetime.datetime.fromisoformat("2022-04-19 10:40:00+00:00"),
@ -576,7 +576,7 @@ async def test_update_missed(
await fake_schedule.fire_until( await fake_schedule.fire_until(
datetime.datetime.fromisoformat("2022-04-19 11:05:00+00:00") datetime.datetime.fromisoformat("2022-04-19 11:05:00+00:00")
) )
assert calls() == [ assert calls_data() == [
{ {
"platform": "calendar", "platform": "calendar",
"event": EVENT_START, "event": EVENT_START,
@ -639,7 +639,7 @@ async def test_update_missed(
) )
async def test_event_payload( async def test_event_payload(
hass: HomeAssistant, hass: HomeAssistant,
calls: Callable[[], list[dict[str, Any]]], calls_data: Callable[[], list[dict[str, Any]]],
fake_schedule: FakeSchedule, fake_schedule: FakeSchedule,
test_entity: MockCalendarEntity, test_entity: MockCalendarEntity,
set_time_zone: None, set_time_zone: None,
@ -650,10 +650,10 @@ async def test_event_payload(
"""Test the fields in the calendar event payload are set.""" """Test the fields in the calendar event payload are set."""
test_entity.create_event(**create_data) test_entity.create_event(**create_data)
async with create_automation(hass, EVENT_START): async with create_automation(hass, EVENT_START):
assert len(calls()) == 0 assert len(calls_data()) == 0
await fake_schedule.fire_until(fire_time) await fake_schedule.fire_until(fire_time)
assert calls() == [ assert calls_data() == [
{ {
"platform": "calendar", "platform": "calendar",
"event": EVENT_START, "event": EVENT_START,
@ -664,7 +664,7 @@ async def test_event_payload(
async def test_trigger_timestamp_window_edge( async def test_trigger_timestamp_window_edge(
hass: HomeAssistant, hass: HomeAssistant,
calls: Callable[[], list[dict[str, Any]]], calls_data: Callable[[], list[dict[str, Any]]],
fake_schedule: FakeSchedule, fake_schedule: FakeSchedule,
test_entity: MockCalendarEntity, test_entity: MockCalendarEntity,
freezer: FrozenDateTimeFactory, freezer: FrozenDateTimeFactory,
@ -678,12 +678,12 @@ async def test_trigger_timestamp_window_edge(
end=datetime.datetime.fromisoformat("2022-04-19 11:30:00+00:00"), end=datetime.datetime.fromisoformat("2022-04-19 11:30:00+00:00"),
) )
async with create_automation(hass, EVENT_START): async with create_automation(hass, EVENT_START):
assert len(calls()) == 0 assert len(calls_data()) == 0
await fake_schedule.fire_until( await fake_schedule.fire_until(
datetime.datetime.fromisoformat("2022-04-19 11:20:00+00:00") datetime.datetime.fromisoformat("2022-04-19 11:20:00+00:00")
) )
assert calls() == [ assert calls_data() == [
{ {
"platform": "calendar", "platform": "calendar",
"event": EVENT_START, "event": EVENT_START,
@ -694,7 +694,7 @@ async def test_trigger_timestamp_window_edge(
async def test_event_start_trigger_dst( async def test_event_start_trigger_dst(
hass: HomeAssistant, hass: HomeAssistant,
calls: Callable[[], list[dict[str, Any]]], calls_data: Callable[[], list[dict[str, Any]]],
fake_schedule: FakeSchedule, fake_schedule: FakeSchedule,
test_entity: MockCalendarEntity, test_entity: MockCalendarEntity,
freezer: FrozenDateTimeFactory, freezer: FrozenDateTimeFactory,
@ -723,13 +723,13 @@ async def test_event_start_trigger_dst(
end=datetime.datetime(2023, 3, 12, 3, 45, tzinfo=tzinfo), end=datetime.datetime(2023, 3, 12, 3, 45, tzinfo=tzinfo),
) )
async with create_automation(hass, EVENT_START): async with create_automation(hass, EVENT_START):
assert len(calls()) == 0 assert len(calls_data()) == 0
await fake_schedule.fire_until( await fake_schedule.fire_until(
datetime.datetime.fromisoformat("2023-03-12 05:00:00-08:00"), datetime.datetime.fromisoformat("2023-03-12 05:00:00-08:00"),
) )
assert calls() == [ assert calls_data() == [
{ {
"platform": "calendar", "platform": "calendar",
"event": EVENT_START, "event": EVENT_START,
@ -750,7 +750,7 @@ async def test_event_start_trigger_dst(
async def test_config_entry_reload( async def test_config_entry_reload(
hass: HomeAssistant, hass: HomeAssistant,
calls: Callable[[], list[dict[str, Any]]], calls_data: Callable[[], list[dict[str, Any]]],
fake_schedule: FakeSchedule, fake_schedule: FakeSchedule,
test_entities: list[MockCalendarEntity], test_entities: list[MockCalendarEntity],
setup_platform: None, setup_platform: None,
@ -764,7 +764,7 @@ async def test_config_entry_reload(
invalid after a config entry was reloaded. invalid after a config entry was reloaded.
""" """
async with create_automation(hass, EVENT_START): async with create_automation(hass, EVENT_START):
assert len(calls()) == 0 assert len(calls_data()) == 0
assert await hass.config_entries.async_reload(config_entry.entry_id) assert await hass.config_entries.async_reload(config_entry.entry_id)
@ -779,7 +779,7 @@ async def test_config_entry_reload(
datetime.datetime.fromisoformat("2022-04-19 11:15:00+00:00"), datetime.datetime.fromisoformat("2022-04-19 11:15:00+00:00"),
) )
assert calls() == [ assert calls_data() == [
{ {
"platform": "calendar", "platform": "calendar",
"event": EVENT_START, "event": EVENT_START,
@ -790,7 +790,7 @@ async def test_config_entry_reload(
async def test_config_entry_unload( async def test_config_entry_unload(
hass: HomeAssistant, hass: HomeAssistant,
calls: Callable[[], list[dict[str, Any]]], calls_data: Callable[[], list[dict[str, Any]]],
fake_schedule: FakeSchedule, fake_schedule: FakeSchedule,
test_entities: list[MockCalendarEntity], test_entities: list[MockCalendarEntity],
setup_platform: None, setup_platform: None,
@ -799,7 +799,7 @@ async def test_config_entry_unload(
) -> None: ) -> None:
"""Test an automation that references a calendar entity that is unloaded.""" """Test an automation that references a calendar entity that is unloaded."""
async with create_automation(hass, EVENT_START): async with create_automation(hass, EVENT_START):
assert len(calls()) == 0 assert len(calls_data()) == 0
assert await hass.config_entries.async_unload(config_entry.entry_id) assert await hass.config_entries.async_unload(config_entry.entry_id)