Fix mistakes

openprocurement · May 11, 2017 · 03e3031 · 03e3031
1 parent ca39269
commit 03e3031
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/openprocurement/integrations/edr/tests/verify.py b/openprocurement/integrations/edr/tests/verify.py
@@ -184,6 +184,7 @@ def test_too_many_requests(self):
         self.assertEqual(response.content_type, 'application/json')
         self.assertEqual(response.status, '429 Too Many Requests')
         self.assertEqual(response.json['errors'][0]['description'], [{u'message': u'Retry request after 26 seconds.'}])
+        self.assertEqual(response.headers['Retry-After'], '26')
 
     def test_server_error(self):
         """Check 500 status EDR response"""
@@ -331,6 +332,7 @@ def test_too_many_requests_details(self):
         self.assertEqual(response.content_type, 'application/json')
         self.assertEqual(response.status, '429 Too Many Requests')
         self.assertEqual(response.json['errors'][0]['description'], [{u'message': u'Retry request after 26 seconds.'}])
+        self.assertEqual(response.headers['Retry-After'], '26')
 
     def test_bad_gateway_details(self):
         """Check 502 status EDR response"""

diff --git a/openprocurement/integrations/edr/views/verify.py b/openprocurement/integrations/edr/views/verify.py
@@ -49,6 +49,7 @@ def verify_user(request):
         LOGGER.info('Return data from EDR service for {}'.format(details.code))
         return {'data': [prepare_data(d) for d in data], 'meta': meta_data(response.headers['Date'])}
     elif response.status_code == 429:
+        request.response.headers['Retry-After'] = response.headers.get('Retry-After')
         return handle_error(request, [{u'message': u'Retry request after {} seconds.'.format(response.headers.get('Retry-After'))}], status=429)
     elif response.status_code == 502:
         return handle_error(request, [{u'message': u'Service is disabled or upgrade.'}])
@@ -76,6 +77,7 @@ def user_details(request):
         LOGGER.info('Return detailed data from EDR service for {}'.format(id))
         return {'data': prepare_data_details(data), 'meta': meta_data(response.headers['Date'])}
     elif response.status_code == 429:
+        request.response.headers['Retry-After'] = response.headers.get('Retry-After')
         return handle_error(request, [{u'message': u'Retry request after {} seconds.'.format(response.headers.get('Retry-After'))}], status=429)
     elif response.status_code == 502:
         return handle_error(request, [{u'message': u'Service is disabled or upgrade.'}])