def test_fetch_schedule_for_pipeline(self):
"""Test that you can fetch schedules for a pipeline."""
self.install_fixture('pipeline_schedule_interval')
self.assertEqual(
PipelineScheduleService.fetch_schedule_for_pipeline(
self.pipeline.id,
),
PipelineScheduleMapper.to_entity(self.pipeline_schedule_interval),
)
def test_fetch_schedules_to_run(self):
"""Test that schedules that need to run are returned."""
self.install_fixture('pipeline_schedule_interval')
self.assertEqual(
PipelineScheduleService.fetch_schedules_to_run(),
[
PipelineScheduleMapper.to_entity(
self.pipeline_schedule_interval,
),
],
)
def test_write_pipeline_schedule(self):
"""Test that we can write an entity to the repository."""
self.install_fixture('pipeline_schedule_interval')
entity = PipelineScheduleMapper.to_entity(self.pipeline_schedule_interval)
entity.schedule = '150'
PipelineScheduleService.write_pipeline_schedule(entity)
self._assert_pipeline_attribute_equals(
self.pipeline.id,
'schedule',
'150',
)
