from flask import json from tests import create_authorization_header def test_post_sms_notification_returns_201(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) } 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 def test_post_sms_notification_returns_404_when_template_is_wrong_type(notify_api, sample_email_template): with notify_api.test_request_context(): with notify_api.test_client() as client: data = { 'phone_number': '+447700900855', 'template_id': str(sample_email_template.id) } auth_header = create_authorization_header(service_id=sample_email_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 resp_text = json.loads(response.get_data(as_text=True)) assert resp_text['code'] == '10400' assert resp_text['message'] == '{0} template is not suitable for {1} notification'.format('email', 'sms') assert resp_text['link'] == 'link to documentation' field = "{0} template is not suitable for {1} notification".format("email", "sms") assert resp_text['fields'][0]['template'] == field