def test_delete_doi_bad_response(self):
response = Mock()
response.status_code = 404
external_service = Mock(spec=['delete_doi'])
external_service.delete_doi.return_value = response
doi_service = services.DOIService(external_service, Mock(), Mock())
response = doi_service.delete_doi('')
expected = False
self.assertEqual(response, expected)
def test_publish_doi_normal_response(self):
response = Mock()
response.status_code = 201
external_service = Mock(spec=['update_doi'])
external_service.update_doi.return_value = response
doi_service = services.DOIService(external_service, Mock(), Mock())
response = doi_service.publish_doi('')
expected = True
self.assertEqual(response, expected)
def test_submit_doi_bad_response(self):
response = Mock()
response.status_code = 404
external_service = Mock(spec=['add_doi'])
external_service.add_doi.return_value = response
service_data_creator = Mock(spec=['create_payload'])
expected = 'ERROR'
doi_service = services.DOIService(external_service, service_data_creator, ['FOO'])
response = doi_service.submit_doi({}, True)
self.assertEqual(response, expected)
def test_submit_doi_normal_response(self):
response = Mock()
response.status_code = 201
external_service = Mock(spec=['add_doi'])
external_service.add_doi.return_value = response
service_data_creator = Mock(spec=['create_payload'])
service_data_creator.return_value = {'data': {'id': 'DOINAME'}}
expected = 'DOINAME'
doi_service = services.DOIService(external_service, service_data_creator, ['FOO'])
response = doi_service.submit_doi({}, True)
self.assertEqual(response, expected)