mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-05-03 23:51:22 -04:00
117 lines
4.3 KiB
Python
117 lines
4.3 KiB
Python
from enum import IntEnum
|
|
from flask import url_for
|
|
from bs4 import BeautifulSoup
|
|
|
|
from app import format_datetime_short
|
|
|
|
valid_letter_jobs = [
|
|
{
|
|
'service_name': {'name': 'test_name'},
|
|
'id': 'test_id',
|
|
'notification_count': 2,
|
|
'job_status': 'ready to send',
|
|
'created_at': '2017-04-01T12:00:00'
|
|
},
|
|
{
|
|
'service_name': {'name': 'test_name 2'},
|
|
'id': 'test_id 2',
|
|
'notification_count': 1,
|
|
'job_status': 'sent to dvla',
|
|
'created_at': '2017-04-02T13:00:00'
|
|
},
|
|
{
|
|
'service_name': {'name': 'test_name 3'},
|
|
'id': 'test_id 3',
|
|
'notification_count': 1,
|
|
'job_status': 'in progress',
|
|
'created_at': '2017-04-03T14:00:00'
|
|
}
|
|
]
|
|
|
|
send_letter_jobs_response = {"response": "Task created to send files to DVLA"}
|
|
|
|
|
|
class letter_jobs_header(IntEnum):
|
|
SERVICE_NAME = 0
|
|
JOB_ID = 1
|
|
NOTIFICATION_COUNT = 2
|
|
JOB_STATUS = 3
|
|
CREATED_AT = 4
|
|
CHECKBOX = 5
|
|
TEMP_STATUS = 6
|
|
|
|
|
|
def test_get_letter_jobs_returns_list_of_all_letter_jobs(logged_in_platform_admin_client, mocker):
|
|
mock_get_letters = mocker.patch('app.letter_jobs_client.get_letter_jobs', return_value=valid_letter_jobs)
|
|
|
|
response = logged_in_platform_admin_client.get(url_for('main.letter_jobs'))
|
|
|
|
assert mock_get_letters.called
|
|
assert response.status_code == 200
|
|
|
|
page = BeautifulSoup(response.data.decode('utf-8'), 'html.parser')
|
|
assert page.h1.string == 'Letter jobs'
|
|
|
|
rows = page.select('table tbody tr')
|
|
|
|
assert len(rows) == len(valid_letter_jobs)
|
|
|
|
for row_pos in range(len(rows)):
|
|
cols = rows[row_pos].find_all('td')
|
|
assert valid_letter_jobs[row_pos]['service_name']['name'] == cols[letter_jobs_header.SERVICE_NAME].text
|
|
assert valid_letter_jobs[row_pos]['id'] == cols[letter_jobs_header.JOB_ID].text
|
|
assert valid_letter_jobs[row_pos]['notification_count'] == int(cols[letter_jobs_header.NOTIFICATION_COUNT].text)
|
|
assert valid_letter_jobs[row_pos]['job_status'] == cols[letter_jobs_header.JOB_STATUS].text
|
|
assert format_datetime_short(
|
|
valid_letter_jobs[row_pos]['created_at']) == cols[letter_jobs_header.CREATED_AT].text
|
|
if not (valid_letter_jobs[row_pos]['job_status'] == 'ready to send' or
|
|
valid_letter_jobs[row_pos]['job_status'] == 'sent to dvla'):
|
|
assert 'disabled' in str(cols[letter_jobs_header.CHECKBOX])
|
|
|
|
|
|
def test_post_letter_jobs_select_1_letter_job_submits_1_job(logged_in_platform_admin_client, mocker):
|
|
letter_jobs_first_selected = {'job_id': ['test_id']}
|
|
|
|
mock_get_letters = mocker.patch('app.letter_jobs_client.get_letter_jobs', return_value=valid_letter_jobs)
|
|
mock_send_letters = mocker.patch('app.letter_jobs_client.send_letter_jobs', return_value=send_letter_jobs_response)
|
|
|
|
response = logged_in_platform_admin_client.post(url_for('main.letter_jobs'), data=letter_jobs_first_selected,
|
|
follow_redirects=True)
|
|
|
|
assert mock_get_letters.called
|
|
assert mock_send_letters.called
|
|
assert response.status_code == 200
|
|
|
|
page = BeautifulSoup(response.data.decode('utf-8'), 'html.parser')
|
|
|
|
rows = page.select('table tbody tr')
|
|
|
|
assert len(rows) == len(valid_letter_jobs)
|
|
|
|
colr0 = rows[0].find_all('td')
|
|
colr1 = rows[1].find_all('td')
|
|
colr2 = rows[2].find_all('td')
|
|
|
|
assert colr0[letter_jobs_header.TEMP_STATUS].text == "sending"
|
|
assert colr1[letter_jobs_header.TEMP_STATUS].text == ""
|
|
assert colr2[letter_jobs_header.TEMP_STATUS].text == ""
|
|
|
|
message = page.find('p', attrs={'id': 'message'}).text
|
|
assert "Task created to send files to DVLA" in message
|
|
|
|
|
|
def test_post_letter_jobs_none_selected_shows_message(logged_in_platform_admin_client, mocker):
|
|
mock_get_letters = mocker.patch('app.letter_jobs_client.get_letter_jobs', return_value=valid_letter_jobs)
|
|
mock_send_letters = mocker.patch('app.letter_jobs_client.send_letter_jobs', return_value=send_letter_jobs_response)
|
|
|
|
response = logged_in_platform_admin_client.post(url_for('main.letter_jobs'), data={}, follow_redirects=True)
|
|
|
|
assert mock_get_letters.called
|
|
assert not mock_send_letters.called
|
|
assert response.status_code == 200
|
|
|
|
page = BeautifulSoup(response.data.decode('utf-8'), 'html.parser')
|
|
message = page.find('p', attrs={'id': 'message'}).text
|
|
|
|
assert "No jobs selected" in message
|