def test_process_ses_results_in_complaint(sample_email_template):
notification = create_notification(template=sample_email_template,
reference='ref1')
handle_ses_complaint(json.loads(ses_complaint_callback()['Message']))
complaints = Complaint.query.all()
assert len(complaints) == 1
assert complaints[0].notification_id == notification.id
def test_process_ses_results_in_complaint_save_complaint_with_null_complaint_type(
notify_api, sample_email_template):
notification = create_notification(template=sample_email_template,
reference='ref1')
msg = json.loads(
ses_complaint_callback_with_missing_complaint_type()['Message'])
handle_ses_complaint(msg)
complaints = Complaint.query.all()
assert len(complaints) == 1
assert complaints[0].notification_id == notification.id
assert complaints[0].complaint_type == UNKNOWN_COMPLAINT_TYPE
def process_ses_results(self, response):
try:
ses_message = json.loads(response['Message'])
notification_type = ses_message.get('eventType')
if notification_type == 'Bounce':
notification_type = determine_notification_bounce_type(
notification_type, ses_message)
elif notification_type == 'Complaint':
complaint, notification, recipient_email = handle_ses_complaint(
ses_message)
return publish_complaint(complaint, notification, recipient_email)
aws_response_dict = get_aws_responses(notification_type)
notification_status = aws_response_dict['notification_status']
reference = ses_message['mail']['messageId']
try:
notification = notifications_dao.dao_get_notification_by_reference(
reference)
except NoResultFound:
message_time = iso8601.parse_date(
ses_message['mail']['timestamp']).replace(tzinfo=None)
if datetime.utcnow() - message_time < timedelta(minutes=5):
self.retry(queue=QueueNames.RETRY)
else:
current_app.logger.warning(
"notification not found for reference: {} (update to {})".
format(reference, notification_status))
return
if notification.status not in {
NOTIFICATION_SENDING, NOTIFICATION_PENDING
}:
notifications_dao.duplicate_update_warning(notification,
notification_status)
return
notifications_dao._update_notification_status(
notification=notification, status=notification_status)
if not aws_response_dict['success']:
current_app.logger.info(
"SES delivery failed: notification id {} and reference {} has error found. Status {}"
.format(notification.id, reference,
aws_response_dict['message']))
else:
current_app.logger.info(
'SES callback return status of {} for notification: {}'.format(
notification_status, notification.id))
statsd_client.incr('callback.ses.{}'.format(notification_status))
if notification.sent_at:
statsd_client.timing_with_dates('callback.ses.elapsed-time',
datetime.utcnow(),
notification.sent_at)
check_and_queue_callback_task(notification)
return True
except Retry:
raise
except Exception as e:
current_app.logger.exception('Error processing SES results: {}'.format(
type(e)))
self.retry(queue=QueueNames.RETRY)
def test_handle_complaint_does_raise_exception_if_notification_not_found(
notify_api):
response = json.loads(ses_complaint_callback()['Message'])
with pytest.raises(expected_exception=SQLAlchemyError):
handle_ses_complaint(response)
def test_handle_complaint_does_not_raise_exception_if_reference_is_missing(
notify_api):
response = json.loads(
ses_complaint_callback_malformed_message_id()['Message'])
handle_ses_complaint(response)
assert len(Complaint.query.all()) == 0