mirror of
https://github.com/GSA/notifications-api.git
synced 2026-04-12 13:19:30 -04:00
99 lines
3.6 KiB
Python
99 lines
3.6 KiB
Python
import uuid
|
|
from datetime import datetime
|
|
|
|
import pytest
|
|
|
|
from app import DATETIME_FORMAT
|
|
from app.models import (
|
|
Notification,
|
|
ServiceWhitelist,
|
|
MOBILE_TYPE, EMAIL_TYPE)
|
|
|
|
|
|
def test_should_build_notification_from_minimal_set_of_api_derived_params(notify_api):
|
|
now = datetime.utcnow()
|
|
|
|
notification = Notification.from_request(
|
|
template_id='template',
|
|
template_version='1',
|
|
recipient='someone',
|
|
service_id='service_id',
|
|
notification_type='SMS',
|
|
created_at=now,
|
|
api_key_id='api_key_id',
|
|
key_type='key_type',
|
|
personalisation={},
|
|
job_id=None,
|
|
job_row_number=None
|
|
)
|
|
assert notification.created_at == now
|
|
assert notification.template_id == 'template'
|
|
assert notification.template_version == '1'
|
|
assert not notification.job_row_number
|
|
assert not notification.job_id
|
|
assert notification.to == 'someone'
|
|
assert notification.service_id == 'service_id'
|
|
assert notification.status == 'created'
|
|
assert not notification.personalisation
|
|
assert notification.notification_type == 'SMS'
|
|
assert notification.api_key_id == 'api_key_id'
|
|
assert notification.key_type == 'key_type'
|
|
|
|
|
|
def test_should_build_notification_from_full_set_of_api_derived_params(notify_api):
|
|
now = datetime.utcnow()
|
|
notification = Notification.from_request(template_id='template',
|
|
template_version=1,
|
|
recipient='someone',
|
|
service_id='service_id',
|
|
personalisation={'key': 'value'},
|
|
notification_type='SMS',
|
|
api_key_id='api_key_id',
|
|
key_type='key_type',
|
|
job_id='job_id',
|
|
job_row_number=100,
|
|
created_at=now
|
|
)
|
|
assert notification.created_at == now
|
|
assert notification.id is None
|
|
assert notification.template_id == 'template'
|
|
assert notification.template_version == 1
|
|
assert notification.job_row_number == 100
|
|
assert notification.job_id == 'job_id'
|
|
assert notification.to == 'someone'
|
|
assert notification.service_id == 'service_id'
|
|
assert notification.status == 'created'
|
|
assert notification.personalisation == {'key': 'value'}
|
|
assert notification.notification_type == 'SMS'
|
|
assert notification.api_key_id == 'api_key_id'
|
|
assert notification.key_type == 'key_type'
|
|
|
|
|
|
@pytest.mark.parametrize('mobile_number', [
|
|
'07700 900678',
|
|
'+44 7700 900678'
|
|
])
|
|
def test_should_build_service_whitelist_from_mobile_number(mobile_number):
|
|
service_whitelist = ServiceWhitelist.from_string('service_id', MOBILE_TYPE, mobile_number)
|
|
|
|
assert service_whitelist.recipient == mobile_number
|
|
|
|
|
|
@pytest.mark.parametrize('email_address', [
|
|
'test@example.com'
|
|
])
|
|
def test_should_build_service_whitelist_from_email_address(email_address):
|
|
service_whitelist = ServiceWhitelist.from_string('service_id', EMAIL_TYPE, email_address)
|
|
|
|
assert service_whitelist.recipient == email_address
|
|
|
|
|
|
@pytest.mark.parametrize('contact, recipient_type', [
|
|
('', None),
|
|
('07700dsadsad', MOBILE_TYPE),
|
|
('gmail.com', EMAIL_TYPE)
|
|
])
|
|
def test_should_not_build_service_whitelist_from_invalid_contact(recipient_type, contact):
|
|
with pytest.raises(ValueError):
|
|
ServiceWhitelist.from_string('service_id', recipient_type, contact)
|