darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-08-19 14:08:47 -04:00
Code Issues Packages Projects Releases Wiki Activity

Update JSON schema postage validation for new values

Browse Source
This commit is contained in:
Katie Smith
2020-03-13 15:39:20 +00:00
parent 186670e10e
commit e2effb6ee1
3 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
app/schema_validation/__init__.py
View File

@@ -34,8 +34,8 @@ def validate_schema_email_address(instance):
@format_checker.checks('postage', raises=ValidationError) @format_checker.checks('postage', raises=ValidationError)
def validate_schema_postage(instance): def validate_schema_postage(instance):
if isinstance(instance, str): if isinstance(instance, str):
if instance not in ["first", "second"]: if instance not in ["first", "second", "europe", "rest-of-world"]:
raise ValidationError("invalid. It must be either first or second.") raise ValidationError("invalid. It must be first, second, europe or rest-of-world.")
return True return True

3
tests/app/service/test_rest.py
View File

@@ -2313,7 +2313,8 @@ def test_create_pdf_letter(mocker, sample_service_full_permissions, client, fake
{"postage": "third", "filename": "string", "created_by": "string", "file_id": "string", {"postage": "third", "filename": "string", "created_by": "string", "file_id": "string",
"recipient_address": "Some Address"}, "recipient_address": "Some Address"},
[ [
{'error': 'ValidationError', 'message': 'postage invalid. It must be either first or second.'} {'error': 'ValidationError',
'message': 'postage invalid. It must be first, second, europe or rest-of-world.'}
] ]
) )
]) ])

2
tests/app/v2/notifications/test_post_letter_notifications.py
View File

@@ -543,7 +543,7 @@ def test_post_letter_notification_throws_error_for_invalid_postage(client, notif
assert response.status_code == 400, response.get_data(as_text=True) assert response.status_code == 400, response.get_data(as_text=True)
resp_json = json.loads(response.get_data(as_text=True)) resp_json = json.loads(response.get_data(as_text=True))
assert resp_json['errors'][0]['message'] == "postage invalid. It must be either first or second." assert resp_json['errors'][0]['message'] == "postage invalid. It must be first, second, europe or rest-of-world."
assert not Notification.query.first() assert not Notification.query.first()