mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-02-13 23:14:32 -05:00
26a985720c
werkzeug's internal workings keep files under 500kb in memory, and files greater than 500kb as a TemporaryFile (https://github.com/pallets/werkzeug/blob/0.11-maintenance/werkzeug/formparser.py#L38) when we encounter a CSV or TSV, we call normalise_newlines, which invokes `.read()`, however when we were passing straight into pyexcel, we called `file.getvalue()` - this exists on BytesIO (small files) but not on TemporaryFile objects (large files) - we were seeing 500 errors
79 lines
2.5 KiB
Python
79 lines
2.5 KiB
Python
from pathlib import Path
|
||
from io import StringIO
|
||
from csv import DictReader
|
||
|
||
import pytest
|
||
from freezegun import freeze_time
|
||
|
||
from app.utils import email_safe, generate_notifications_csv, generate_previous_dict, generate_next_dict, Spreadsheet
|
||
|
||
|
||
def test_email_safe_return_dot_separated_email_domain():
|
||
test_name = 'SOME service with+stuff+ b123'
|
||
expected = 'some.service.withstuff.b123'
|
||
actual = email_safe(test_name)
|
||
assert actual == expected
|
||
|
||
|
||
@pytest.mark.parametrize(
|
||
"status, template_type, expected_status",
|
||
[
|
||
('sending', None, 'Sending'),
|
||
('delivered', None, 'Delivered'),
|
||
('failed', None, 'Failed'),
|
||
('technical-failure', None, 'Technical failure'),
|
||
('temporary-failure', 'email', 'Inbox not accepting messages right now'),
|
||
('permanent-failure', 'email', 'Email address doesn’t exist'),
|
||
('temporary-failure', 'sms', 'Phone not accepting messages right now'),
|
||
('permanent-failure', 'sms', 'Phone number doesn’t exist')
|
||
]
|
||
)
|
||
@freeze_time("2016-01-01 15:09:00.061258")
|
||
def test_generate_csv_from_notifications(
|
||
app_,
|
||
service_one,
|
||
active_user_with_permissions,
|
||
mock_get_notifications,
|
||
status,
|
||
template_type,
|
||
expected_status
|
||
):
|
||
with app_.test_request_context():
|
||
csv_content = generate_notifications_csv(
|
||
mock_get_notifications(
|
||
service_one['id'],
|
||
rows=1,
|
||
set_template_type=template_type,
|
||
set_status=status
|
||
)['notifications']
|
||
)
|
||
|
||
for row in DictReader(StringIO(csv_content)):
|
||
assert row['Time'] == 'Friday 01 January 2016 at 15:09'
|
||
assert row['Status'] == expected_status
|
||
|
||
|
||
def test_generate_previous_dict(client):
|
||
ret = generate_previous_dict('main.view_jobs', 'foo', 2, {})
|
||
assert 'page=1' in ret['url']
|
||
assert ret['title'] == 'Previous page'
|
||
assert ret['label'] == 'page 1'
|
||
|
||
|
||
def test_generate_next_dict(client):
|
||
ret = generate_next_dict('main.view_jobs', 'foo', 2, {})
|
||
assert 'page=3' in ret['url']
|
||
assert ret['title'] == 'Next page'
|
||
assert ret['label'] == 'page 3'
|
||
|
||
|
||
def test_generate_previous_next_dict_adds_other_url_args(client):
|
||
ret = generate_next_dict('main.view_notifications', 'foo', 2, {'message_type': 'blah'})
|
||
assert 'notifications/blah' in ret['url']
|
||
|
||
|
||
def test_can_create_spreadsheet_from_large_excel_file():
|
||
with open(str(Path.cwd() / 'tests' / 'spreadsheet_files' / 'excel 2007.xlsx'), 'rb') as xl:
|
||
ret = Spreadsheet.from_file(xl, filename='xl.xlsx')
|
||
assert ret.as_csv_data
|