This commit is contained in:
Rebecca Law
2018-01-03 13:21:00 +00:00
parent 1fccb3da5e
commit e71bbc1bb5

View File

@@ -1,4 +1,3 @@
import uuid
from pathlib import Path
from io import StringIO
from collections import OrderedDict
@@ -16,6 +15,7 @@ from app.utils import (
get_letter_timings,
get_cdn_domain
)
from tests.conftest import fake_uuid
def _get_notifications_csv(
@@ -29,7 +29,8 @@ def _get_notifications_csv(
status='Delivered',
created_at='Thursday 19 April at 12:00',
rows=1,
with_links=False
with_links=False,
job_id=fake_uuid
):
links = {}
if with_links:
@@ -60,7 +61,7 @@ def _get_notifications_csv(
def _get_notifications_csv_mock(
mocker,
api_user_active,
job_id=None
job_id=fake_uuid
):
return mocker.patch(
'app.notification_api_client.get_notifications_for_service',
@@ -124,13 +125,13 @@ def test_can_create_spreadsheet_from_dict_with_filename():
def test_generate_notifications_csv_returns_correct_csv_file(_get_notifications_csv_mock):
csv_content = generate_notifications_csv(service_id='1234', job_id=uuid.uuid4())
csv_content = generate_notifications_csv(service_id='1234', job_id=fake_uuid)
csv_file = DictReader(StringIO('\n'.join(csv_content)))
assert csv_file.fieldnames == ['Row number', 'Recipient', 'Template', 'Type', 'Job', 'Status', 'Time']
def test_generate_notifications_csv_only_calls_once_if_no_next_link(_get_notifications_csv_mock):
list(generate_notifications_csv(service_id='1234', job_id=uuid.uuid4()))
list(generate_notifications_csv(service_id='1234', job_id=fake_uuid))
assert _get_notifications_csv_mock.call_count == 1
@@ -148,7 +149,7 @@ def test_generate_notifications_csv_calls_twice_if_next_link(mocker):
]
)
csv_content = generate_notifications_csv(service_id=service_id, job_id=uuid.uuid4())
csv_content = generate_notifications_csv(service_id=service_id, job_id=fake_uuid)
csv = DictReader(StringIO('\n'.join(csv_content)))
assert len(list(csv)) == 10