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)