def test_scheduler_change_set_adding_schedule():
schedule_1 = ScheduleDefinition('schedule_1', "*****", "pipeline_name", {})
schedule_2 = ScheduleDefinition('schedule_2', "*****", "pipeline_name", {})
schedule_3 = ScheduleDefinition('schedule_3', "*****", "pipeline_name", {})
schedule_4 = ScheduleDefinition('schedule_4', "*****", "pipeline_name", {})
modified_schedule_2 = ScheduleDefinition('schedule_2', "0****",
"pipeline_name",
{'new_key': "new_value"})
renamed_schedule_3 = ScheduleDefinition('renamed_schedule_3', "*****",
"pipeline_name", {})
running_1 = Schedule("1", schedule_1.schedule_definition_data,
ScheduleStatus.RUNNING, "", "")
running_2 = Schedule("2", schedule_2.schedule_definition_data,
ScheduleStatus.RUNNING, "", "")
running_3 = Schedule("3", schedule_3.schedule_definition_data,
ScheduleStatus.RUNNING, "", "")
running_4 = Schedule("4", schedule_4.schedule_definition_data,
ScheduleStatus.RUNNING, "", "")
# Add initial schedules
change_set_1 = get_schedule_change_set([], [schedule_1, schedule_2])
assert sorted(change_set_1) == sorted([('add', 'schedule_2', []),
('add', 'schedule_1', [])])
# Add more schedules
change_set_2 = get_schedule_change_set(
[running_1, running_2],
[schedule_1, schedule_2, schedule_3, schedule_4])
assert sorted(change_set_2) == sorted([('add', 'schedule_3', []),
('add', 'schedule_4', [])])
# Modify schedule_2
change_set_3 = get_schedule_change_set(
[running_1, running_2, running_3, running_4],
[schedule_1, modified_schedule_2, schedule_3, schedule_4],
)
assert change_set_3 == [('change', 'schedule_2', [('cron_schedule',
('*****', '0****'))])]
# Delete schedules
change_set_3 = get_schedule_change_set(
[running_1, running_2, running_3, running_4], [schedule_3, schedule_4])
assert sorted(change_set_3) == sorted([('remove', 'schedule_1', []),
('remove', 'schedule_2', [])])
# Rename schedules
change_set_4 = get_schedule_change_set(
[running_1, running_2, running_3, running_4],
[schedule_1, schedule_2, renamed_schedule_3, schedule_4],
)
assert sorted(change_set_4) == sorted([('add', 'renamed_schedule_3', []),
('remove', 'schedule_3', [])])
def build_schedule(
schedule_name,
cron_schedule,
status=ScheduleStatus.STOPPED,
):
return Schedule(
ScheduleDefinitionData(schedule_name, cron_schedule),
status,
"",
"",
)
