darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-02-01 07:35:34 -05:00
Code Issues Packages Projects Releases Wiki Activity

Added base object for response statuses, and tests around it's behaviour

Browse Source
This commit is contained in:
Martyn Inglis
2016-03-21 09:20:38 +00:00
parent 86579fba71
commit e2cfbce8c4
6 changed files with 133 additions and 61 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
app/clients/__init__.py
View File

@@ -11,3 +11,24 @@ class Client(object):
Base client for sending notifications.
'''
pass
class ClientResponse:
def __init__(self):
self.__response_model__ = None
def response_code_to_object(self, response_code):
return self.__response_model__[response_code]
def response_code_to_message(self, response_code):
return self.response_code_to_object(response_code)['message']
def response_code_to_notification_status(self, response_code):
print(response_code)
return self.response_code_to_object(response_code)['notification_status']
def response_code_to_notification_statistics_status(self, response_code):
return self.response_code_to_object(response_code)['notification_statistics_status']
def response_code_to_notification_success(self, response_code):
return self.response_code_to_object(response_code)['success']

39
app/clients/email/aws_ses.py
View File

@@ -1,22 +1,33 @@
import boto3
from flask import current_app
from monotonic import monotonic
from app.clients import ClientResponse
from app.clients.email import (EmailClientException, EmailClient)
ses_response_status = {
'Bounce': {
"success": False,
"notify_status": 'bounce'
},
'Delivery': {
"success": True,
"notify_status": 'delivered'
},
'Complaint': {
"success": False,
"notify_status": 'complaint'
}
}
class AwsSesResponses(ClientResponse):
def __init__(self):
ClientResponse.__init__(self)
self.__response_model__ = {
'Bounce': {
"message": 'Bounced',
"success": False,
"notification_status": 'bounce',
"notification_statistics_status": 'failed'
},
'Delivery': {
"message": 'Delivered',
"success": True,
"notification_status": 'delivered',
"notification_statistics_status": 'delivered'
},
'Complaint': {
"message": 'Complaint',
"success": False,
"notification_status": 'complaint',
"notification_statistics_status": 'failed'
}
}
class AwsSesClientException(EmailClientException):

70
app/clients/sms/firetext.py
View File

@@ -6,58 +6,34 @@ from app.clients.sms import (
)
from flask import current_app
from requests import request, RequestException, HTTPError
from app.clients import ClientResponse
logger = logging.getLogger(__name__)
firetext_response_status = {
'0': {
"firetext_message": 'delivered',
"success": True,
"notify_status": 'delivered'
},
'1': {
"firetext_message": 'declined',
"success": False,
"notify_status": 'failed'
},
'2': {
"firetext_message": 'Undelivered (Pending with Network)',
"success": False,
"notify_status": 'sent'
}
}
class FiretextResponses:
response_model = {
'0': {
"firetext_message": 'delivered',
"stats_mapping": 'delivered',
"success": True,
"notify_status": 'delivered'
},
'1': {
"firetext_message": 'declined',
"success": False,
"stats_mapping": 'failure',
"notify_status": 'failed'
},
'2': {
"firetext_message": 'Undelivered (Pending with Network)',
"success": False,
"stats_mapping": None,
"notify_status": 'sent'
class FiretextResponses(ClientResponse):
def __init__(self):
ClientResponse.__init__(self)
self.__response_model__ = {
'0': {
"message": 'Delivered',
"notification_statistics_status": 'delivered',
"success": True,
"notification_status": 'delivered'
},
'1': {
"message": 'Declined',
"success": False,
"notification_statistics_status": 'failed',
"notification_status": 'failed'
},
'2': {
"message": 'Undelivered (Pending with Network)',
"success": False,
"notification_statistics_status": None,
"notification_status": 'sent'
}
}
}
def response_code_to_message(self, response_code):
return self.response_model.get(response_code)['firetext_message']
def response_code_to_notify_status(self, response_code):
return self.response_model.get(response_code)['notify_status']
def response_code_to_notify_stats(self, response_code):
return self.response_model.get(response_code)['stats_mapping']
class FiretextClientException(SmsClientException):