def test_create_staging_job_failure(self, mock_post, mock_session):
# given
ingest_api = IngestApi(token_manager=self.token_manager)
ingest_api.get_staging_jobs_url = MagicMock(return_value='url')
mock_post.return_value.json.return_value = {
'staging-area-uuid': 'uuid'
}
mock_post.return_value.status_code = requests.codes.ok
mock_post.return_value.raise_for_status = MagicMock(
side_effect=HTTPError())
# when
with self.assertRaises(HTTPError):
ingest_api.create_staging_job('uuid', 'filename', 'metadata_uuid')
def test_create_staging_job_success(self, mock_post, mock_session):
# given
ingest_api = IngestApi(token_manager=self.token_manager)
ingest_api.get_staging_jobs_url = MagicMock(return_value='url')
mock_post.return_value.json.return_value = {
'staging-area-uuid': 'uuid'
}
mock_post.return_value.status_code = requests.codes.ok
# when
staging_job = ingest_api.create_staging_job('uuid', 'filename',
'metadata-uuid')
self.assertEqual(staging_job, {'staging-area-uuid': 'uuid'})
