mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-02-05 10:53:28 -05:00
b4a8c2732d
So that we can get a better sense of what the URLs look like. Leaving unsigned for now because we don’t know if the ID will be a string or int or…
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('/123/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('/123/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('/123/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('/123/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('/123/sms/check')
|
|
|
|
assert response.status_code == 302
|
|
assert response.location == 'http://localhost/123/jobs/456'
|