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'})