From e5a377edd82dc986346597bcf0c828797120c518 Mon Sep 17 00:00:00 2001 From: Ken Tsang Date: Tue, 11 Apr 2017 10:59:50 +0100 Subject: [PATCH] Add letter jobs page, client and test --- app/__init__.py | 3 + app/main/views/letter_jobs.py | 56 +++++------- app/notify_client/letter_jobs_client.py | 21 +++++ app/templates/views/letter-jobs.html | 20 +++-- tests/app/main/views/test_letter_jobs.py | 103 +++++++++++++++++++++++ 5 files changed, 161 insertions(+), 42 deletions(-) create mode 100644 app/notify_client/letter_jobs_client.py create mode 100644 tests/app/main/views/test_letter_jobs.py diff --git a/app/__init__.py b/app/__init__.py index a3e22afbd..23f4f8bd5 100644 --- a/app/__init__.py +++ b/app/__init__.py @@ -52,6 +52,7 @@ from app.notify_client.events_api_client import EventsApiClient from app.notify_client.provider_client import ProviderClient from app.notify_client.organisations_client import OrganisationsClient from app.notify_client.models import AnonymousUser +from app.notify_client.letter_jobs_client import LetterJobsClient login_manager = LoginManager() csrf = CsrfProtect() @@ -69,6 +70,7 @@ provider_client = ProviderClient() organisations_client = OrganisationsClient() asset_fingerprinter = AssetFingerprinter() statsd_client = StatsdClient() +letter_jobs_client = LetterJobsClient() # The current service attached to the request stack. current_service = LocalProxy(partial(_lookup_req_object, 'service')) @@ -100,6 +102,7 @@ def create_app(): events_api_client.init_app(application) provider_client.init_app(application) organisations_client.init_app(application) + letter_jobs_client.init_app(application) login_manager.init_app(application) login_manager.login_view = 'main.sign_in' diff --git a/app/main/views/letter_jobs.py b/app/main/views/letter_jobs.py index 00cb314db..e371f33f2 100644 --- a/app/main/views/letter_jobs.py +++ b/app/main/views/letter_jobs.py @@ -1,45 +1,31 @@ -from flask import (render_template, url_for, redirect, request, abort) +import datetime + +from flask import render_template, request +from flask_login import login_required + +from app import letter_jobs_client, format_datetime_24h from app.main import main -from app import convert_to_boolean -from flask_login import (login_required, current_user) +from app.utils import user_has_permissions @main.route("/letter-jobs", methods=['GET', 'POST']) @login_required +@user_has_permissions(admin_override=True) def letter_jobs(): - letter_jobs_list = get_letter_jobs() - msg = '' - if request.method == 'POST': - send_letters = request.form.getlist('send_letter') - for job_id in send_letters: - job = [j for j in letter_jobs_list if job_id == j['job_id']][0] - job['send'] = True + letter_jobs_list = letter_jobs_client.get_letter_jobs() - msg = 'sending:{}'.format(send_letters) + if request.method == 'POST': + if len(request.form.getlist('job_id')) > 0: + job_ids = request.form.getlist('job_id') + + response = letter_jobs_client.send_letter_jobs(job_ids) + msg = response['response'] + + for job_id in job_ids: + job = [j for j in letter_jobs_list if job_id == j['id']][0] + job['sending'] = 'sending' + else: + msg = 'No jobs selected' return render_template('views/letter-jobs.html', letter_jobs_list=letter_jobs_list, message=msg) - - -def get_letter_jobs(): - return [ - { - 'service_name': 'test_name', - 'job_id': 'test_id', - 'status': 'test_status', - 'created_at': '2017-04-01' - }, - { - 'service_name': 'test_name 2', - 'job_id': 'test_id 2', - 'status': 'test_status 2', - 'created_at': '2017-04-02' - - }, - { - 'service_name': 'test_name 3', - 'job_id': 'test_id 3', - 'status': 'test_status 3', - 'created_at': '2017-04-03' - } - ] diff --git a/app/notify_client/letter_jobs_client.py b/app/notify_client/letter_jobs_client.py new file mode 100644 index 000000000..a3337917b --- /dev/null +++ b/app/notify_client/letter_jobs_client.py @@ -0,0 +1,21 @@ +from app.notify_client import NotifyAdminAPIClient + + +class LetterJobsClient(NotifyAdminAPIClient): + + def __init__(self): + super().__init__("a", "b", "c") + + def init_app(self, app): + self.base_url = app.config['API_HOST_NAME'] + self.service_id = app.config['ADMIN_CLIENT_USER_NAME'] + self.api_key = app.config['ADMIN_CLIENT_SECRET'] + + def get_letter_jobs(self): + return self.get(url='/letter-jobs')['data'] + + def send_letter_jobs(self, job_ids): + return self.post( + url='/send-letter-jobs', + data={"job_ids": job_ids} + )['data'] diff --git a/app/templates/views/letter-jobs.html b/app/templates/views/letter-jobs.html index aaa50f549..4f039aef4 100644 --- a/app/templates/views/letter-jobs.html +++ b/app/templates/views/letter-jobs.html @@ -17,25 +17,31 @@ Service name Job ID Status - Created at + Created at + {% for job in letter_jobs_list %} - {{ job.service_name }} - {{ job.job_id }} - {{ job.status }} - {{ job.created_at }} - + {{ job.service_name.name }} + {{ job.id }} + {{ job.job_status }} + {{ job.created_at|format_datetime_short }} + + {{ job.sending }} {% endfor %} +

{{ page_footer('Send') }} {% if message %} -

+

{{ message }} + {% if message != 'No jobs selected' %} +

Refresh page to see status updates
+ {% endif %}

{% endif %} diff --git a/tests/app/main/views/test_letter_jobs.py b/tests/app/main/views/test_letter_jobs.py new file mode 100644 index 000000000..5e7467a23 --- /dev/null +++ b/tests/app/main/views/test_letter_jobs.py @@ -0,0 +1,103 @@ +from flask import url_for +from bs4 import BeautifulSoup + +from app import format_datetime_short + +valid_letter_jobs = [ + { + 'service_name': {'name': 'test_name'}, + 'template_version': 1, + 'id': 'test_id', + 'job_status': 'test_status', + 'created_at': '2017-04-01T12:00:00' + }, + { + 'service_name': {'name': 'test_name 2'}, + 'template_version': 2, + 'id': 'test_id 2', + 'job_status': 'test_status 2', + 'created_at': '2017-04-02T13:00:00' + }, + { + 'service_name': {'name': 'test_name 3'}, + 'template_version': 3, + 'id': 'test_id 3', + 'job_status': 'test_status 3', + 'created_at': '2017-04-03T14:00:00' + } +] + +send_letter_jobs_response = {"response": "Task created to send files to DVLA"} + + +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' + + table = page.find('table') + table_body = table.find('tbody') + rows = table_body.find_all('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[0].text + assert valid_letter_jobs[row_pos]['id'] == cols[1].text + assert valid_letter_jobs[row_pos]['job_status'] == cols[2].text + assert format_datetime_short(valid_letter_jobs[row_pos]['created_at']) == cols[3].text + + +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) + + assert mock_get_letters.called + assert mock_send_letters.called + assert response.status_code == 200 + + page = BeautifulSoup(response.data.decode('utf-8'), 'html.parser') + + table = page.find('table') + table_body = table.find('tbody') + rows = table_body.find_all('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[5].text == "sending" + assert colr1[5].text == "" + assert colr2[5].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={}) + + 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