mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-07-24 01:49:15 -04:00
584533eb11
At the moment the file contents are not persisted by checked in memory. The first and last three records are show if all are valid. If there are invalid rows, they are reported and the user is prompted to go back and sort out upload file. The storing of upload result (i.e. validation of file) in session will be removed in next story which is about persisting of file for later processing.
109 lines
4.6 KiB
Python
109 lines
4.6 KiB
Python
from io import BytesIO
|
|
from tests.app.main import create_test_user
|
|
|
|
|
|
def test_upload_empty_csvfile_returns_to_upload_page(notifications_admin, notifications_admin_db, notify_db_session):
|
|
with notifications_admin.test_request_context():
|
|
with notifications_admin.test_client() as client:
|
|
user = create_test_user('active')
|
|
client.login(user)
|
|
|
|
upload_data = {'file': (BytesIO(''.encode('utf-8')), 'emtpy.csv')}
|
|
response = client.post('/sms/send', data=upload_data,
|
|
follow_redirects=True)
|
|
|
|
assert response.status_code == 200
|
|
content = response.get_data(as_text=True)
|
|
assert 'The file emtpy.csv contained no data' in content
|
|
|
|
|
|
def test_upload_csvfile_with_invalid_phone_shows_check_page_with_errors(
|
|
notifications_admin, notifications_admin_db, notify_db_session):
|
|
|
|
with notifications_admin.test_request_context():
|
|
with notifications_admin.test_client() as client:
|
|
user = create_test_user('active')
|
|
client.login(user)
|
|
|
|
file_contents = 'phone\n+44 123\n+44 456'.encode('utf-8')
|
|
upload_data = {'file': (BytesIO(file_contents), 'invalid.csv')}
|
|
response = client.post('/sms/send', data=upload_data,
|
|
follow_redirects=True)
|
|
assert response.status_code == 200
|
|
content = response.get_data(as_text=True)
|
|
assert 'There was a problem with some of the numbers' in content
|
|
assert 'The following numbers are invalid' in content
|
|
assert '+44 123' in content
|
|
assert '+44 456' in content
|
|
assert 'Go back and resolve errors' in content
|
|
|
|
|
|
def test_upload_csvfile_with_valid_phone_shows_first3_and_last3_numbers(
|
|
notifications_admin, notifications_admin_db, notify_db_session):
|
|
|
|
with notifications_admin.test_request_context():
|
|
with notifications_admin.test_client() as client:
|
|
user = create_test_user('active')
|
|
client.login(user)
|
|
file_contents = 'phone\n+44 7700 900981\n+44 7700 900982\n+44 7700 900983\n+44 7700 900984\n+44 7700 900985\n+44 7700 900986\n+44 7700 900987\n+44 7700 900988\n+44 7700 900989'.encode('utf-8') # noqa
|
|
|
|
upload_data = {'file': (BytesIO(file_contents), 'valid.csv')}
|
|
response = client.post('/sms/send', data=upload_data,
|
|
follow_redirects=True)
|
|
|
|
content = response.get_data(as_text=True)
|
|
|
|
assert response.status_code == 200
|
|
assert 'Check and confirm' in content
|
|
assert 'First three message in file' in content
|
|
assert 'Last three messages in file' in content
|
|
assert '+44 7700 900981' in content
|
|
assert '+44 7700 900982' in content
|
|
assert '+44 7700 900983' in content
|
|
assert '+44 7700 900984' not in content
|
|
assert '+44 7700 900985' not in content
|
|
assert '+44 7700 900986' not in content
|
|
assert '+44 7700 900987' in content
|
|
assert '+44 7700 900988' in content
|
|
assert '+44 7700 900989' in content
|
|
|
|
|
|
def test_upload_csvfile_with_valid_phone_shows_all_if_6_or_less_numbers(
|
|
notifications_admin, notifications_admin_db, notify_db_session):
|
|
|
|
with notifications_admin.test_request_context():
|
|
with notifications_admin.test_client() as client:
|
|
user = create_test_user('active')
|
|
client.login(user)
|
|
|
|
file_contents = 'phone\n+44 7700 900981\n+44 7700 900982\n+44 7700 900983\n+44 7700 900984\n+44 7700 900985\n+44 7700 900986'.encode('utf-8') # noqa
|
|
|
|
upload_data = {'file': (BytesIO(file_contents), 'valid.csv')}
|
|
response = client.post('/sms/send', data=upload_data,
|
|
follow_redirects=True)
|
|
|
|
content = response.get_data(as_text=True)
|
|
|
|
assert response.status_code == 200
|
|
assert 'Check and confirm' in content
|
|
assert 'All messages in file' in content
|
|
assert '+44 7700 900981' in content
|
|
assert '+44 7700 900982' in content
|
|
assert '+44 7700 900983' in content
|
|
assert '+44 7700 900984' in content
|
|
assert '+44 7700 900985' in content
|
|
assert '+44 7700 900986' in content
|
|
|
|
|
|
def test_should_redirect_to_job(notifications_admin, notifications_admin_db,
|
|
notify_db_session):
|
|
with notifications_admin.test_request_context():
|
|
with notifications_admin.test_client() as client:
|
|
user = create_test_user('active')
|
|
client.login(user)
|
|
|
|
response = client.post('/sms/check')
|
|
|
|
assert response.status_code == 302
|
|
assert response.location == 'http://localhost/jobs/job'
|