Merge pull request #1224 from alphagov/letters-from-a-service

Letters from template-preview service

tests/app/main/views/test_send.py

@@ -2,14 +2,13 @@ import uuid
 from io import BytesIO
 from os import path
 from glob import glob
-import re
 from itertools import repeat
 from functools import partial
-from unittest.mock import patch
 
 import pytest
 from bs4 import BeautifulSoup
 from flask import url_for
+from notifications_utils.template import LetterPreviewTemplate
 
 from app.main.views.send import get_check_messages_back_url
 
@@ -524,18 +523,9 @@ def test_can_start_letters_job(
     assert response.status_code == 302
 
 
-@pytest.mark.parametrize(
-    'view, expected_content_type',
-    [
-        ('.check_messages_as_pdf', 'application/pdf'),
-        ('.check_messages_as_png', 'image/png'),
-    ]
-)
-@patch('app.utils.LetterPreviewTemplate.jinja_template.render', return_value='')
+@pytest.mark.parametrize('filetype', ['pdf', 'png'])
 def test_should_show_preview_letter_message(
-    mock_letter_preview,
-    view,
-    expected_content_type,
+    filetype,
     logged_in_platform_admin_client,
     mock_get_service_letter_template,
     mock_get_users_by_service,
@@ -554,6 +544,10 @@ def test_should_show_preview_letter_message(
             ['123 street, abc123']
         )
     )
+    mocked_preview = mocker.patch(
+        'app.main.views.send.TemplatePreview.from_utils_template',
+        return_value='foo'
+    )
 
     service_id = service_one['id']
     template_id = fake_uuid
@@ -565,18 +559,41 @@ def test_should_show_preview_letter_message(
             'valid': True
         }
     response = logged_in_platform_admin_client.get(
-        url_for(view, service_id=service_id, template_type='letter', upload_id=fake_uuid)
+        url_for(
+            'main.check_messages_preview',
+            service_id=service_id,
+            template_type='letter',
+            upload_id=fake_uuid,
+            filetype=filetype
+        )
     )
 
-    assert response.status_code == 200
-    assert response.content_type == expected_content_type
     mock_get_service_letter_template.assert_called_with(service_id, template_id)
-    assert mock_letter_preview.call_args[0][0]['subject'] == (
-        'Subject'
-    )
-    assert mock_letter_preview.call_args[0][0]['message'] == (
-        '<p>Template &lt;em&gt;content&lt;/em&gt; with &amp; entity</p>'
+
+    assert response.status_code == 200
+    assert response.get_data(as_text=True) == 'foo'
+    assert mocked_preview.call_args[0][0].id == template_id
+    assert type(mocked_preview.call_args[0][0]) == LetterPreviewTemplate
+    assert mocked_preview.call_args[0][1] == filetype
+
+
+def test_dont_show_preview_letter_templates_for_bad_filetype(
+    logged_in_client,
+    mock_get_service_template,
+    service_one,
+    fake_uuid
+):
+    resp = logged_in_client.get(
+        url_for(
+            'main.check_messages_preview',
+            service_id=service_one['id'],
+            template_type='letter',
+            upload_id=fake_uuid,
+            filetype='blah'
+        )
     )
+    assert resp.status_code == 404
+    assert mock_get_service_template.called is False
 
 
 def test_check_messages_should_revalidate_file_when_uploading_file(

tests/app/main/views/test_templates.py

@@ -1,6 +1,5 @@
-from itertools import repeat
 from datetime import datetime
-from unittest.mock import Mock, patch, ANY
+from unittest.mock import Mock, ANY
 
 import pytest
 from bs4 import BeautifulSoup
@@ -148,44 +147,60 @@ def test_should_show_page_template_with_priority_select_if_platform_admin(
     mock_get_service_template.assert_called_with('1234', template_id)
 
 
-@pytest.mark.parametrize('view_suffix, expected_content_type', [
-    ('as_pdf', 'application/pdf'),
-    ('as_png', 'image/png'),
-])
+@pytest.mark.parametrize('filetype', ['pdf', 'png'])
 @pytest.mark.parametrize('view, extra_view_args', [
-    ('.view_letter_template', {}),
-    ('.view_template_version', {'version': 1}),
+    ('.view_letter_template_preview', {}),
+    ('.view_template_version_preview', {'version': 1}),
 ])
-@patch('app.main.views.templates.LetterPreviewTemplate.jinja_template.render', return_value='foo')
 def test_should_show_preview_letter_templates(
-    mock_letter_preview,
     view,
     extra_view_args,
-    view_suffix,
-    expected_content_type,
+    filetype,
     logged_in_client,
     mock_get_service_email_template,
-    mock_get_user_by_email,
     service_one,
-    fake_uuid
+    fake_uuid,
+    mocker
 ):
+    mocked_preview = mocker.patch(
+        'app.main.views.templates.TemplatePreview.from_database_object',
+        return_value='foo'
+    )
+
     service_id, template_id = service_one['id'], fake_uuid
+
     response = logged_in_client.get(url_for(
-        '{}_{}'.format(view, view_suffix),
+        view,
         service_id=service_id,
         template_id=template_id,
+        filetype=filetype,
         **extra_view_args
     ))
 
     assert response.status_code == 200
-    assert response.content_type == expected_content_type
+    assert response.get_data(as_text=True) == 'foo'
     mock_get_service_email_template.assert_called_with(service_id, template_id, **extra_view_args)
-    assert mock_letter_preview.call_args[0][0]['subject'] == (
-        "Your <span class='placeholder'>((thing))</span> is due soon"
-    )
-    assert mock_letter_preview.call_args[0][0]['message'] == (
-        "<p>Your vehicle tax expires on <span class='placeholder'>((date))</span></p>"
+    assert mocked_preview.call_args[0][0]['id'] == template_id
+    assert mocked_preview.call_args[0][0]['service'] == service_id
+    assert mocked_preview.call_args[0][1] == filetype
+
+
+def test_dont_show_preview_letter_templates_for_bad_filetype(
+    logged_in_client,
+    mock_get_service_template,
+    service_one,
+    fake_uuid
+):
+    resp = logged_in_client.get(
+        url_for(
+            '.view_letter_template_preview',
+            service_id=service_one['id'],
+            template_id=fake_uuid,
+            filetype='blah'
+        )
     )
+    assert resp.status_code == 404
+    assert mock_get_service_template.called is False
 
 
 def test_should_redirect_when_saving_a_template(

tests/app/test_cloudfoundry_config.py

@@ -52,12 +52,24 @@ def deskpro_config():
     }
 
 
+@pytest.fixture
+def template_preview_config():
+    return {
+        'name': 'notify-template-preview',
+        'credentials': {
+            'api_host': 'template-preview api host',
+            'api_key': 'template-preview api key'
+        }
+    }
+
+
 @pytest.fixture
 def cloudfoundry_config(
         notify_config,
         aws_config,
         hosted_graphite_config,
         deskpro_config,
+        template_preview_config,
 ):
     return {
         'user-provided': [
@@ -65,6 +77,7 @@ def cloudfoundry_config(
             aws_config,
             hosted_graphite_config,
             deskpro_config,
+            template_preview_config,
         ]
     }
 
@@ -128,3 +141,11 @@ def test_deskpro_config():
 
     assert os.environ['DESKPRO_API_HOST'] == 'deskpro api host'
     assert os.environ['DESKPRO_API_KEY'] == 'deskpro api key'
+
+
+@pytest.mark.usefixtures('os_environ', 'cloudfoundry_environ')
+def test_template_preview_config():
+    extract_cloudfoundry_config()
+
+    assert os.environ['TEMPLATE_PREVIEW_API_HOST'] == 'template-preview api host'
+    assert os.environ['TEMPLATE_PREVIEW_API_KEY'] == 'template-preview api key'

tests/app/test_template_previews.py

@@ -0,0 +1,35 @@
+from unittest.mock import Mock
+from notifications_utils.template import LetterPreviewTemplate
+
+from app.template_previews import TemplatePreview
+
+
+def test_from_utils_template_calls_through(mocker, mock_get_service_letter_template):
+    mock_from_db = mocker.patch('app.template_previews.TemplatePreview.from_database_object')
+    template = LetterPreviewTemplate(mock_get_service_letter_template(None, None)['data'])
+
+    ret = TemplatePreview.from_utils_template(template, 'foo')
+
+    assert ret == mock_from_db.return_value
+    mock_from_db.assert_called_once_with(template._template, 'foo', template.values)
+
+
+def test_from_database_object_makes_request(mocker, client):
+    resp = Mock(content='a', status_code='b', headers={'c': 'd'})
+    request_mock = mocker.patch('app.template_previews.requests.post', return_value=resp)
+    mocker.patch('app.template_previews.current_service', __getitem__=Mock(return_value='123'))
+
+    ret = TemplatePreview.from_database_object(template='foo', filetype='bar')
+
+    assert ret[0] == 'a'
+    assert ret[1] == 'b'
+    assert list(ret[2]) == [('c', 'd')]
+    url = 'http://localhost:6013/preview.bar'
+    data = {
+        'letter_contact_block': '123',
+        'template': 'foo',
+        'values': None
+    }
+    headers = {'Authorization': 'Token my-secret-key'}
+
+    request_mock.assert_called_once_with(url, json=data, headers=headers)