mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-18 04:41:45 -05:00
107 lines
3.5 KiB
Python
107 lines
3.5 KiB
Python
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 = {
|
|
'template': 'template',
|
|
'template_version': '1',
|
|
'to': 'someone',
|
|
'personalisation': {}
|
|
}
|
|
notification = Notification.from_api_request(
|
|
created_at=now.strftime(DATETIME_FORMAT),
|
|
notification=notification,
|
|
notification_id="notification_id",
|
|
service_id="service_id",
|
|
notification_type='SMS',
|
|
api_key_id='api_key_id',
|
|
key_type='key_type'
|
|
)
|
|
assert notification.created_at == now
|
|
assert notification.id == "notification_id"
|
|
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 = {
|
|
'template': 'template',
|
|
'template_version': '1',
|
|
'to': 'someone',
|
|
'personalisation': {'key': 'value'},
|
|
'job': 'job_id',
|
|
'row_number': 100
|
|
}
|
|
notification = Notification.from_api_request(
|
|
created_at=now.strftime(DATETIME_FORMAT),
|
|
notification=notification,
|
|
notification_id="notification_id",
|
|
service_id="service_id",
|
|
notification_type='SMS',
|
|
api_key_id='api_key_id',
|
|
key_type='key_type'
|
|
)
|
|
assert notification.created_at == now
|
|
assert notification.id == "notification_id"
|
|
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)
|