From 3732e45126d934547d0171aa5ccdfc3775047b7b Mon Sep 17 00:00:00 2001
From: Katie Smith <katie.smith@digital.cabinet-office.gov.uk>
Date: Mon, 21 Oct 2019 14:34:46 +0100
Subject: [PATCH] Stop allowing uploaded letters to be sent for trial mode
 services

We weren't checking if a service was in trial mode when they try to send
an uploaded file. If a service is in trial mode, we now show a banner at
the top of the preview page and no send button.
---
 app/templates/views/uploads/preview.html | 10 +++++-
 tests/app/main/views/test_uploads.py     | 42 ++++++++++++++++++++++--
 2 files changed, 48 insertions(+), 4 deletions(-)

diff --git a/app/templates/views/uploads/preview.html b/app/templates/views/uploads/preview.html
index 8ecc96a3e..c448b3b4d 100644
--- a/app/templates/views/uploads/preview.html
+++ b/app/templates/views/uploads/preview.html
@@ -14,6 +14,14 @@
         <p>{{ error.detail | safe }}</p>
       {% endif %}
     {% endcall %}
+  {% elif current_service.trial_mode %}
+    {% call banner_wrapper(type='dangerous') %}
+      {% with
+        count_of_recipients=1
+      %}
+        {% include "partials/check/trying-to-send-letters-in-trial-mode.html" %}
+      {% endwith %}
+    {% endcall %}
   {% else %}
       {{ page_header(
         original_filename,
@@ -25,7 +33,7 @@
       {{ template|string }}
     </div>
 
-    {% if status == 'valid' %}
+    {% if status == 'valid' and current_service.live %}
     <div class="js-stick-at-bottom-when-scrolling">
       <form method="post" enctype="multipart/form-data" action="{{url_for(
           'main.send_uploaded_letter',
diff --git a/tests/app/main/views/test_uploads.py b/tests/app/main/views/test_uploads.py
index 722c53806..891972037 100644
--- a/tests/app/main/views/test_uploads.py
+++ b/tests/app/main/views/test_uploads.py
@@ -26,7 +26,7 @@ def test_get_upload_letter(client_request):
     assert normalize_spaces(page.find('label', class_='file-upload-button').text) == 'Choose file'
 
 
-def test_post_upload_letter_redirects_for_valid_file(mocker, client_request):
+def test_post_upload_letter_redirects_for_valid_file(mocker, active_user_with_permissions, service_one, client_request):
     mocker.patch('uuid.uuid4', return_value='fake-uuid')
     antivirus_mock = mocker.patch('app.main.views.uploads.antivirus_client.scan', return_value=True)
     mocker.patch(
@@ -38,6 +38,9 @@ def test_post_upload_letter_redirects_for_valid_file(mocker, client_request):
         'filename': 'tests/test_pdf_files/one_page_pdf.pdf', 'page_count': '1', 'status': 'valid'})
     mocker.patch('app.main.views.uploads.service_api_client.get_precompiled_template')
 
+    service_one['restricted'] = False
+    client_request.login(active_user_with_permissions, service=service_one)
+
     with open('tests/test_pdf_files/one_page_pdf.pdf', 'rb') as file:
         page = client_request.post(
             'main.upload_letter',
@@ -62,7 +65,12 @@ def test_post_upload_letter_redirects_for_valid_file(mocker, client_request):
     assert page.find('button', {'type': 'submit'}).text == 'Send 1 letter'
 
 
-def test_post_upload_letter_shows_letter_preview_for_valid_file(mocker, client_request):
+def test_post_upload_letter_shows_letter_preview_for_valid_file(
+    mocker,
+    active_user_with_permissions,
+    service_one,
+    client_request,
+):
     letter_template = {'template_type': 'letter',
                        'reply_to_text': '',
                        'postage': 'second',
@@ -81,6 +89,9 @@ def test_post_upload_letter_shows_letter_preview_for_valid_file(mocker, client_r
         'filename': 'tests/test_pdf_files/one_page_pdf.pdf', 'page_count': '3', 'status': 'valid'})
     mocker.patch('app.main.views.uploads.service_api_client.get_precompiled_template', return_value=letter_template)
 
+    service_one['restricted'] = False
+    client_request.login(active_user_with_permissions, service=service_one)
+
     with open('tests/test_pdf_files/one_page_pdf.pdf', 'rb') as file:
         page = client_request.post(
             'main.upload_letter',
@@ -281,11 +292,14 @@ def test_post_upload_letter_does_not_upload_to_s3_if_template_preview_raises_unk
     assert not mock_s3.called
 
 
-def test_uploaded_letter_preview(mocker, client_request):
+def test_uploaded_letter_preview(mocker, active_user_with_permissions, service_one, client_request):
     mocker.patch('app.main.views.uploads.service_api_client')
     mocker.patch('app.main.views.uploads.get_letter_metadata', return_value={
         'filename': 'my_letter.pdf', 'page_count': '1', 'status': 'valid'})
 
+    service_one['restricted'] = False
+    client_request.login(active_user_with_permissions, service=service_one)
+
     page = client_request.get(
         'main.uploaded_letter_preview',
         service_id=SERVICE_ONE_ID,
@@ -300,6 +314,28 @@ def test_uploaded_letter_preview(mocker, client_request):
     assert page.find('div', class_='letter-sent')
 
 
+def test_uploaded_letter_preview_does_not_show_send_button_if_service_in_trial_mode(mocker, client_request):
+    mocker.patch('app.main.views.uploads.service_api_client')
+    mocker.patch('app.main.views.uploads.get_letter_metadata', return_value={
+        'filename': 'my_letter.pdf', 'page_count': '1', 'status': 'valid'})
+
+    # client_request uses service_one, which is in trial mode
+    page = client_request.get(
+        'main.uploaded_letter_preview',
+        service_id=SERVICE_ONE_ID,
+        file_id='fake-uuid',
+        original_filename='my_letter.pdf',
+        page_count=1,
+        status='valid',
+        error={},
+        _test_page_title=False,
+    )
+
+    assert normalize_spaces(page.find('h1').text) == 'You cannot send this letter'
+    assert page.find('div', class_='letter-sent')
+    assert not page.find('button', {'type': 'submit'})
+
+
 def test_uploaded_letter_preview_image_shows_overlay_when_content_outside_printable_area(
     mocker,
     logged_in_client,