From cfc031cfbf00ce39bd688ec83a1e74ac12557379 Mon Sep 17 00:00:00 2001
From: Chris Hill-Scott <me@quis.cc>
Date: Thu, 20 Apr 2017 12:15:30 +0100
Subject: [PATCH] Pass dvla_org_id through to template preview app

---
 app/template_previews.py            | 3 ++-
 tests/__init__.py                   | 3 ++-
 tests/app/test_template_previews.py | 3 ++-
 3 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/app/template_previews.py b/app/template_previews.py
index e3b5adf6c..466b60a31 100644
--- a/app/template_previews.py
+++ b/app/template_previews.py
@@ -10,7 +10,8 @@ class TemplatePreview:
         data = {
             'letter_contact_block': current_service['letter_contact_block'],
             'template': template,
-            'values': values
+            'values': values,
+            'dvla_org_id': current_service['dvla_organisation'],
         }
         resp = requests.post(
             '{}/preview.{}'.format(current_app.config['TEMPLATE_PREVIEW_API_HOST'], filetype),
diff --git a/tests/__init__.py b/tests/__init__.py
index a333d5f94..d7f014cb9 100644
--- a/tests/__init__.py
+++ b/tests/__init__.py
@@ -72,7 +72,8 @@ def service_json(
         'organisation': organisation,
         'branding': branding,
         'created_at': created_at or str(datetime.utcnow()),
-        'letter_contact_block': letter_contact_block
+        'letter_contact_block': letter_contact_block,
+        'dvla_organisation': '001',
     }
 
 
diff --git a/tests/app/test_template_previews.py b/tests/app/test_template_previews.py
index be50e642b..24b9505b0 100644
--- a/tests/app/test_template_previews.py
+++ b/tests/app/test_template_previews.py
@@ -28,7 +28,8 @@ def test_from_database_object_makes_request(mocker, client):
     data = {
         'letter_contact_block': '123',
         'template': 'foo',
-        'values': None
+        'values': None,
+        'dvla_org_id': '123',
     }
     headers = {'Authorization': 'Token my-secret-key'}