mirror of
https://github.com/GSA/notifications-api.git
synced 2026-08-15 07:18:09 -04:00
raise 404 when unknown url
consistent with other endpoints. also refactor of notification_schema to separate some fns to a different file
This commit is contained in:
45
app/v2/notifications/create_response.py
Normal file
45
app/v2/notifications/create_response.py
Normal file
@@ -0,0 +1,45 @@
|
|||||||
|
|
||||||
|
def create_post_sms_response_from_notification(notification, content, from_number, url_root, scheduled_for):
|
||||||
|
noti = __create_notification_response(notification, url_root, scheduled_for)
|
||||||
|
noti['content'] = {
|
||||||
|
'from_number': from_number,
|
||||||
|
'body': content
|
||||||
|
}
|
||||||
|
return noti
|
||||||
|
|
||||||
|
|
||||||
|
def create_post_email_response_from_notification(notification, content, subject, email_from, url_root, scheduled_for):
|
||||||
|
noti = __create_notification_response(notification, url_root, scheduled_for)
|
||||||
|
noti['content'] = {
|
||||||
|
"from_email": email_from,
|
||||||
|
"body": content,
|
||||||
|
"subject": subject
|
||||||
|
}
|
||||||
|
return noti
|
||||||
|
|
||||||
|
|
||||||
|
def create_post_letter_response_from_notification(notification, content, subject, url_root, scheduled_for):
|
||||||
|
noti = __create_notification_response(notification, url_root, scheduled_for)
|
||||||
|
noti['content'] = {
|
||||||
|
"body": content,
|
||||||
|
"subject": subject
|
||||||
|
}
|
||||||
|
return noti
|
||||||
|
|
||||||
|
|
||||||
|
def __create_notification_response(notification, url_root, scheduled_for):
|
||||||
|
return {
|
||||||
|
"id": notification.id,
|
||||||
|
"reference": notification.client_reference,
|
||||||
|
"uri": "{}v2/notifications/{}".format(url_root, str(notification.id)),
|
||||||
|
'template': {
|
||||||
|
"id": notification.template_id,
|
||||||
|
"version": notification.template_version,
|
||||||
|
"uri": "{}services/{}/templates/{}".format(
|
||||||
|
url_root,
|
||||||
|
str(notification.service_id),
|
||||||
|
str(notification.template_id)
|
||||||
|
)
|
||||||
|
},
|
||||||
|
"scheduled_for": scheduled_for if scheduled_for else None
|
||||||
|
}
|
||||||
@@ -232,49 +232,3 @@ post_letter_response = {
|
|||||||
},
|
},
|
||||||
"required": ["id", "content", "uri", "template"]
|
"required": ["id", "content", "uri", "template"]
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
def create_post_sms_response_from_notification(notification, content, from_number, url_root, scheduled_for):
|
|
||||||
noti = __create_notification_response(notification, url_root, scheduled_for)
|
|
||||||
noti['content'] = {
|
|
||||||
'from_number': from_number,
|
|
||||||
'body': content
|
|
||||||
}
|
|
||||||
return noti
|
|
||||||
|
|
||||||
|
|
||||||
def create_post_email_response_from_notification(notification, content, subject, email_from, url_root, scheduled_for):
|
|
||||||
noti = __create_notification_response(notification, url_root, scheduled_for)
|
|
||||||
noti['content'] = {
|
|
||||||
"from_email": email_from,
|
|
||||||
"body": content,
|
|
||||||
"subject": subject
|
|
||||||
}
|
|
||||||
return noti
|
|
||||||
|
|
||||||
|
|
||||||
def create_post_letter_response_from_notification(notification, content, subject, url_root, scheduled_for):
|
|
||||||
noti = __create_notification_response(notification, url_root, scheduled_for)
|
|
||||||
noti['content'] = {
|
|
||||||
"body": content,
|
|
||||||
"subject": subject
|
|
||||||
}
|
|
||||||
return noti
|
|
||||||
|
|
||||||
|
|
||||||
def __create_notification_response(notification, url_root, scheduled_for):
|
|
||||||
return {
|
|
||||||
"id": notification.id,
|
|
||||||
"reference": notification.client_reference,
|
|
||||||
"uri": "{}v2/notifications/{}".format(url_root, str(notification.id)),
|
|
||||||
'template': {
|
|
||||||
"id": notification.template_id,
|
|
||||||
"version": notification.template_version,
|
|
||||||
"uri": "{}services/{}/templates/{}".format(
|
|
||||||
url_root,
|
|
||||||
str(notification.service_id),
|
|
||||||
str(notification.template_id)
|
|
||||||
)
|
|
||||||
},
|
|
||||||
"scheduled_for": scheduled_for if scheduled_for else None
|
|
||||||
}
|
|
||||||
|
|||||||
@@ -1,6 +1,6 @@
|
|||||||
import functools
|
import functools
|
||||||
|
|
||||||
from flask import request, jsonify, current_app
|
from flask import request, jsonify, current_app, abort
|
||||||
|
|
||||||
from app import api_user, authenticated_service
|
from app import api_user, authenticated_service
|
||||||
from app.config import QueueNames
|
from app.config import QueueNames
|
||||||
@@ -18,12 +18,13 @@ from app.notifications.validators import (
|
|||||||
validate_template
|
validate_template
|
||||||
)
|
)
|
||||||
from app.schema_validation import validate
|
from app.schema_validation import validate
|
||||||
from app.v2.errors import BadRequestError
|
|
||||||
from app.v2.notifications import v2_notification_blueprint
|
from app.v2.notifications import v2_notification_blueprint
|
||||||
from app.v2.notifications.notification_schemas import (
|
from app.v2.notifications.notification_schemas import (
|
||||||
post_sms_request,
|
post_sms_request,
|
||||||
post_email_request,
|
post_email_request,
|
||||||
post_letter_request,
|
post_letter_request
|
||||||
|
)
|
||||||
|
from app.v2.notifications.create_response import (
|
||||||
create_post_sms_response_from_notification,
|
create_post_sms_response_from_notification,
|
||||||
create_post_email_response_from_notification,
|
create_post_email_response_from_notification,
|
||||||
create_post_letter_response_from_notification
|
create_post_letter_response_from_notification
|
||||||
@@ -39,7 +40,7 @@ def post_notification(notification_type):
|
|||||||
elif notification_type == LETTER_TYPE:
|
elif notification_type == LETTER_TYPE:
|
||||||
form = validate(request.get_json(), post_letter_request)
|
form = validate(request.get_json(), post_letter_request)
|
||||||
else:
|
else:
|
||||||
raise BadRequestError(message='Unknown notification type {}'.format(notification_type))
|
abort(404)
|
||||||
|
|
||||||
check_service_has_permission(notification_type, authenticated_service.permissions)
|
check_service_has_permission(notification_type, authenticated_service.permissions)
|
||||||
|
|
||||||
|
|||||||
@@ -1,3 +1,4 @@
|
|||||||
|
|
||||||
import uuid
|
import uuid
|
||||||
|
|
||||||
from flask import url_for, json
|
from flask import url_for, json
|
||||||
@@ -145,7 +146,7 @@ def test_returns_a_429_limit_exceeded_if_rate_limit_exceeded(
|
|||||||
{'service_permissions': [EMAIL_TYPE, SMS_TYPE]},
|
{'service_permissions': [EMAIL_TYPE, SMS_TYPE]},
|
||||||
{'restricted': True}
|
{'restricted': True}
|
||||||
])
|
])
|
||||||
def test_post_letter_notification_returns_400_if_not_allowed_to_send_notification(
|
def test_post_letter_notification_returns_403_if_not_allowed_to_send_notification(
|
||||||
client,
|
client,
|
||||||
notify_db_session,
|
notify_db_session,
|
||||||
service_args
|
service_args
|
||||||
@@ -159,7 +160,7 @@ def test_post_letter_notification_returns_400_if_not_allowed_to_send_notificatio
|
|||||||
}
|
}
|
||||||
|
|
||||||
error_json = letter_request(client, data, service_id=service.id, _expected_status=400)
|
error_json = letter_request(client, data, service_id=service.id, _expected_status=400)
|
||||||
assert error_json['status_code'] == 400
|
assert error_json['status_code'] == 403
|
||||||
assert error_json['errors'] == [
|
assert error_json['errors'] == [
|
||||||
{'error': 'BadRequestError', 'message': 'Cannot send letters'}
|
{'error': 'BadRequestError', 'message': 'Cannot send letters'}
|
||||||
]
|
]
|
||||||
|
|||||||
@@ -442,8 +442,6 @@ def test_post_notification_raises_bad_request_if_not_valid_notification_type(cli
|
|||||||
data='{}',
|
data='{}',
|
||||||
headers=[('Content-Type', 'application/json'), auth_header]
|
headers=[('Content-Type', 'application/json'), auth_header]
|
||||||
)
|
)
|
||||||
assert response.status_code == 400
|
assert response.status_code == 404
|
||||||
error_json = json.loads(response.get_data(as_text=True))
|
error_json = json.loads(response.get_data(as_text=True))
|
||||||
assert error_json['errors'] == [
|
assert 'The requested URL was not found on the server.' in error_json['message']
|
||||||
{'error': 'BadRequestError', 'message': 'Unknown notification type foo'}
|
|
||||||
]
|
|
||||||
|
|||||||
Reference in New Issue
Block a user