mirror of
https://github.com/GSA/notifications-api.git
synced 2026-07-16 03:10:48 -04:00
Serialise notifications for csv separately:
* Introduces separate method on Notification to serialise the notification * ready for csv output * Fixes issue where job_row_number = 0 not being accounted for correctly
This commit is contained in:
@@ -405,14 +405,13 @@ def sample_notification_with_job(
|
||||
):
|
||||
if job is None:
|
||||
job = sample_job(notify_db, notify_db_session, service=service, template=template)
|
||||
|
||||
return sample_notification(
|
||||
notify_db,
|
||||
notify_db_session,
|
||||
service,
|
||||
template,
|
||||
job=job,
|
||||
job_row_number=job_row_number if job_row_number else None,
|
||||
job_row_number=job_row_number if job_row_number is not None else None,
|
||||
to_field=to_field,
|
||||
status=status,
|
||||
reference=reference,
|
||||
@@ -482,7 +481,7 @@ def sample_notification(
|
||||
'updated_at': created_at if status in NOTIFICATION_STATUS_TYPES_COMPLETED else None,
|
||||
'client_reference': client_reference
|
||||
}
|
||||
if job_row_number:
|
||||
if job_row_number is not None:
|
||||
data['job_row_number'] = job_row_number
|
||||
notification = Notification(**data)
|
||||
dao_create_notification(notification)
|
||||
|
||||
@@ -1,5 +1,7 @@
|
||||
import pytest
|
||||
|
||||
from freezegun import freeze_time
|
||||
|
||||
from app.models import (
|
||||
ServiceWhitelist,
|
||||
Notification,
|
||||
@@ -11,6 +13,10 @@ from app.models import (
|
||||
NOTIFICATION_TECHNICAL_FAILURE,
|
||||
NOTIFICATION_STATUS_TYPES_FAILED
|
||||
)
|
||||
from tests.app.conftest import (
|
||||
sample_template as create_sample_template,
|
||||
sample_notification_with_job as create_sample_notification_with_job
|
||||
)
|
||||
|
||||
|
||||
@pytest.mark.parametrize('mobile_number', [
|
||||
@@ -65,3 +71,96 @@ def test_status_conversion_handles_failed_statuses(initial_statuses, expected_st
|
||||
converted_statuses = Notification.substitute_status(initial_statuses)
|
||||
assert len(converted_statuses) == len(expected_statuses)
|
||||
assert set(converted_statuses) == set(expected_statuses)
|
||||
|
||||
|
||||
@freeze_time("2016-01-01 11:09:00.000000")
|
||||
@pytest.mark.parametrize('template_type, recipient', [
|
||||
('sms', '+447700900855'),
|
||||
('email', 'foo@bar.com'),
|
||||
])
|
||||
def test_notification_for_csv_returns_correct_type(notify_db, notify_db_session, template_type, recipient):
|
||||
template = create_sample_template(notify_db, notify_db_session, template_type=template_type)
|
||||
notification = create_sample_notification_with_job(
|
||||
notify_db,
|
||||
notify_db_session,
|
||||
template=template,
|
||||
to_field=recipient
|
||||
)
|
||||
|
||||
expected_resp = {
|
||||
"recipient": recipient,
|
||||
"row_number": "",
|
||||
"template_name": "Template Name",
|
||||
"template_type": template_type,
|
||||
"created_at": "Friday 01 January 2016 at 11:09",
|
||||
"job_name": "some.csv",
|
||||
"status": "Sending"
|
||||
}
|
||||
|
||||
serialized = notification.serialize_for_csv()
|
||||
assert serialized == expected_resp
|
||||
|
||||
|
||||
@freeze_time("2016-01-01 11:09:00.000000")
|
||||
def test_notification_for_csv_returns_correct_job_row_number(notify_db, notify_db_session):
|
||||
notification = create_sample_notification_with_job(
|
||||
notify_db,
|
||||
notify_db_session,
|
||||
job_row_number=0
|
||||
)
|
||||
expected_resp = {
|
||||
"recipient": "+447700900855",
|
||||
"row_number": 1,
|
||||
"template_name": "Template Name",
|
||||
"template_type": "sms",
|
||||
"created_at": "Friday 01 January 2016 at 11:09",
|
||||
"job_name": "some.csv",
|
||||
"status": "Sending"
|
||||
}
|
||||
|
||||
serialized = notification.serialize_for_csv()
|
||||
assert serialized == expected_resp
|
||||
|
||||
|
||||
@freeze_time("2016-01-30 12:39:58.321312")
|
||||
def test_notifiation_for_csv_returns_formatted_status(notify_db, notify_db_session):
|
||||
notification = create_sample_notification_with_job(
|
||||
notify_db,
|
||||
notify_db_session,
|
||||
job_row_number=51,
|
||||
status='temporary-failure'
|
||||
)
|
||||
expected_resp = {
|
||||
"recipient": "+447700900855",
|
||||
"row_number": 52,
|
||||
"template_name": "Template Name",
|
||||
"template_type": "sms",
|
||||
"created_at": "Saturday 30 January 2016 at 12:39",
|
||||
"job_name": "some.csv",
|
||||
"status": "Phone not accepting messages right now"
|
||||
}
|
||||
|
||||
serialized = notification.serialize_for_csv()
|
||||
assert serialized == expected_resp
|
||||
|
||||
|
||||
@freeze_time("2017-03-26 23:01:53.321312")
|
||||
def test_notifiation_for_csv_returns_bst_correctly(notify_db, notify_db_session):
|
||||
notification = create_sample_notification_with_job(
|
||||
notify_db,
|
||||
notify_db_session,
|
||||
job_row_number=100,
|
||||
status='permanent-failure'
|
||||
)
|
||||
expected_resp = {
|
||||
"recipient": "+447700900855",
|
||||
"row_number": 101,
|
||||
"template_name": "Template Name",
|
||||
"template_type": "sms",
|
||||
"created_at": "Monday 27 March 2017 at 00:01",
|
||||
"job_name": "some.csv",
|
||||
"status": "Phone number doesn’t exist"
|
||||
}
|
||||
|
||||
serialized = notification.serialize_for_csv()
|
||||
assert serialized == expected_resp
|
||||
|
||||
Reference in New Issue
Block a user