mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-23 08:51:30 -05:00
104 lines
4.6 KiB
Python
104 lines
4.6 KiB
Python
import uuid
|
|
|
|
from flask import json
|
|
|
|
from app.models import Notification
|
|
from tests import create_authorization_header
|
|
|
|
|
|
def test_post_sms_notification_returns_201(notify_api, sample_template, mocker):
|
|
with notify_api.test_request_context():
|
|
with notify_api.test_client() as client:
|
|
mocked = mocker.patch('app.celery.provider_tasks.deliver_sms.apply_async')
|
|
data = {
|
|
'phone_number': '+447700900855',
|
|
'template_id': str(sample_template.id)
|
|
}
|
|
auth_header = create_authorization_header(service_id=sample_template.service_id)
|
|
|
|
response = client.post(
|
|
path='/v2/notifications/sms',
|
|
data=json.dumps(data),
|
|
headers=[('Content-Type', 'application/json'), auth_header])
|
|
|
|
assert response.status_code == 201
|
|
resp_json = json.loads(response.get_data(as_text=True))
|
|
notifications = Notification.query.all()
|
|
assert len(notifications) == 1
|
|
notification_id = notifications[0].id
|
|
assert resp_json['id'] is not None
|
|
assert resp_json['reference'] is None
|
|
assert resp_json['content']['body'] == sample_template.content
|
|
assert resp_json['content']['from_number'] == sample_template.service.sms_sender
|
|
assert 'v2/notifications/{}'.format(notification_id) in resp_json['uri']
|
|
assert resp_json['template']['id'] == str(sample_template.id)
|
|
assert resp_json['template']['version'] == sample_template.version
|
|
assert 'v2/templates/{}'.format(sample_template.id) in resp_json['template']['uri']
|
|
assert mocked.called
|
|
|
|
|
|
def test_post_sms_notification_returns_404_and_missing_template(notify_api, sample_service):
|
|
with notify_api.test_request_context():
|
|
with notify_api.test_client() as client:
|
|
data = {
|
|
'phone_number': '+447700900855',
|
|
'template_id': str(uuid.uuid4())
|
|
}
|
|
auth_header = create_authorization_header(service_id=sample_service.id)
|
|
|
|
response = client.post(
|
|
path='/v2/notifications/sms',
|
|
data=json.dumps(data),
|
|
headers=[('Content-Type', 'application/json'), auth_header])
|
|
|
|
assert response.status_code == 400
|
|
assert response.headers['Content-type'] == 'application/json'
|
|
|
|
error_json = json.loads(response.get_data(as_text=True))
|
|
assert error_json['code'] == 10400
|
|
assert error_json['message'] == 'Template not found'
|
|
assert error_json['fields'] == [{'template': 'Template not found'}]
|
|
|
|
|
|
def test_post_sms_notification_returns_403_and_well_formed_auth_error(notify_api, sample_template):
|
|
with notify_api.test_request_context():
|
|
with notify_api.test_client() as client:
|
|
data = {
|
|
'phone_number': '+447700900855',
|
|
'template_id': str(sample_template.id)
|
|
}
|
|
|
|
response = client.post(
|
|
path='/v2/notifications/sms',
|
|
data=json.dumps(data),
|
|
headers=[('Content-Type', 'application/json')])
|
|
|
|
assert response.status_code == 401
|
|
assert response.headers['Content-type'] == 'application/json'
|
|
error_resp = json.loads(response.get_data(as_text=True))
|
|
assert error_resp['code'] == 401
|
|
assert error_resp['message'] == 'Unauthorized, authentication token must be provided'
|
|
assert error_resp['fields'] == {'token': ['Unauthorized, authentication token must be provided']}
|
|
|
|
|
|
def test_post_sms_notification_returns_400_and_for_schema_problems(notify_api, sample_template):
|
|
with notify_api.test_request_context():
|
|
with notify_api.test_client() as client:
|
|
data = {
|
|
'phone_number': '+447700900855',
|
|
'template': str(sample_template.id)
|
|
}
|
|
auth_header = create_authorization_header(service_id=sample_template.service_id)
|
|
|
|
response = client.post(
|
|
path='/v2/notifications/sms',
|
|
data=json.dumps(data),
|
|
headers=[('Content-Type', 'application/json'), auth_header])
|
|
|
|
assert response.status_code == 400
|
|
assert response.headers['Content-type'] == 'application/json'
|
|
error_resp = json.loads(response.get_data(as_text=True))
|
|
assert error_resp['code'] == '1001'
|
|
assert error_resp['message'] == 'Validation error occurred - POST v2/notifications/sms'
|
|
assert error_resp['fields'] == [{"template_id": "is a required property"}]
|