def _raise_publish_error(self, status_code, e, destination=None):
if status_code >= 400 and status_code < 500:
raise PublishHTTPPushClientError.httpPushError(e, destination)
elif status_code >= 500 and status_code < 600:
raise PublishHTTPPushServerError.httpPushError(e, destination)
else:
raise PublishHTTPPushError.httpPushError(e, destination)
def test_retry_on_http_push_server_error(self, *mocks):
self.app.config['MAX_TRANSMIT_RETRY_ATTEMPT'] = 4
item_1 = {
'_id': 'item_1',
'destination': {},
'subscriber_id': '1',
'item_id': 'test',
'headline': 'test headline',
'item_version': 4,
'state': 'pending'
}
orig_item = item_1.copy() # item's original state in DB
fake_get_service = mocks[1]
fake_get_service().find_one.return_value = orig_item
fake_transmitter = MagicMock()
fake_transmitter.transmit.side_effect = PublishHTTPPushServerError.httpPushError(
Exception('server 5xx'))
fake_transmitters_list = mocks[0]
fake_transmitters_list.__getitem__.return_value = fake_transmitter
self.func_under_test(item_1['_id'])
fake_get_service().system_update.assert_called_with(
'item_1', {
'_updated': ANY,
'retry_attempt': 1,
'state': 'retrying',
'next_retry_attempt_at': ANY
}, orig_item)
def _raise_publish_error(self, status_code, e, destination=None):
if status_code >= 400 and status_code < 500:
raise PublishHTTPPushClientError.httpPushError(e, destination)
elif status_code >= 500 and status_code < 600:
raise PublishHTTPPushServerError.httpPushError(e, destination)
else:
raise PublishHTTPPushError.httpPushError(e, destination)
def test_retry_on_http_push_server_error(self, *mocks):
self.app.config['MAX_TRANSMIT_RETRY_ATTEMPT'] = 4
item_1 = {
'_id': 'item_1',
'destination': {},
'subscriber_id': '1',
'item_id': 'test',
'headline': 'test headline',
'item_version': 4,
'state': 'pending'
}
orig_item = item_1.copy() # item's original state in DB
fake_get_service = mocks[1]
fake_get_service().find_one.return_value = orig_item
queue_items = [item_1]
fake_transmitter = MagicMock()
fake_transmitter.transmit.side_effect = PublishHTTPPushServerError.httpPushError(Exception('server 5xx'))
fake_transmitters_list = mocks[0]
fake_transmitters_list.__getitem__.return_value = fake_transmitter
self.func_under_test(queue_items)
fake_get_service().system_update.assert_called_with('item_1', {'_updated': ANY, 'retry_attempt': 1,
'state': 'retrying',
'next_retry_attempt_at': ANY}, orig_item)
def test_server_publish_error_thrown(self, fake_post, fake_notifiers):
raise_http_exception = Mock(side_effect=PublishHTTPPushServerError.httpPushError(Exception('server 5xx')))
fake_post.return_value = Mock(status_code=503, text='server 5xx', raise_for_status=raise_http_exception)
# needed for bad exception handling classes
fake_notifiers.return_value = []
service = HTTPPushService()
with self.assertRaises(PublishHTTPPushServerError):
service._push_item(self.destination, json.dumps(self.item))
def test_server_publish_error_thrown(self, fake_post, fake_notifiers):
raise_http_exception = Mock(side_effect=PublishHTTPPushServerError.httpPushError(Exception('server 5xx')))
fake_post.return_value = Mock(status_code=503, text='server 5xx', raise_for_status=raise_http_exception)
# needed for bad exception handling classes
fake_notifiers.return_value = []
service = HTTPPushService()
with self.assertRaises(PublishHTTPPushServerError):
service._push_item(self.destination, json.dumps(self.item))
def test_retry_on_http_push_server_error(self, *mocks):
self.app.config["MAX_TRANSMIT_RETRY_ATTEMPT"] = 4
item_1 = {
"_id": "item_1",
"destination": {},
"subscriber_id": "1",
"item_id": "test",
"headline": "test headline",
"item_version": 4,
"state": "pending",
}
orig_item = item_1.copy() # item's original state in DB
fake_get_service = mocks[1]
fake_get_service().find_one.return_value = orig_item
fake_transmitter = MagicMock()
fake_transmitter.transmit.side_effect = PublishHTTPPushServerError.httpPushError(
Exception("server 5xx"))
fake_transmitters_list = mocks[0]
fake_transmitters_list.__getitem__.return_value = fake_transmitter
self.assertIsNone(self.func_under_test(item_1["_id"]))
fake_get_service().system_update.assert_called_with(
"item_1",
{
"_updated": ANY,
"retry_attempt": 1,
"state": "retrying",
"next_retry_attempt_at": ANY
},
orig_item,
)
