def test_write_task_connection(self):
"""Test that the provided entity gets written to the repository."""
entity = TaskConnectionEntity.get_mock_object()
self.assertEquals(
TaskConnectionService.fetch_task_connections_for_pipeline(entity.pipeline_id),
[],
)
TaskConnectionService.write_task_connection(entity)
self.assertEquals(
TaskConnectionService.fetch_task_connections_for_pipeline(entity.pipeline_id),
[
entity,
],
)
def test_fetch_task_connections_for_pipeline(self, mock_fetch):
"""Test that a list of TaskConnectionEntities is returned."""
mock_fetch.return_value = [
self.url_to_log_connection,
self.raw_input_to_log_connection,
]
entities = TaskConnectionService.fetch_task_connections_for_pipeline(self.pipeline.id)
self.assertEqual(entities[0].id, self.url_to_log_connection.id)
