From 593d753ed5738bf920df3960712d82ca8d42846b Mon Sep 17 00:00:00 2001
From: Martyn Inglis <martyn.inglis@digital.cabinet-office.gov.uk>
Date: Wed, 9 Mar 2016 14:52:07 +0000
Subject: [PATCH] Updated tests

---
 tests/app/conftest.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/tests/app/conftest.py b/tests/app/conftest.py
index 7e7034f3c..58f9f1de8 100644
--- a/tests/app/conftest.py
+++ b/tests/app/conftest.py
@@ -198,7 +198,8 @@ def sample_api_key(notify_db,
 def sample_job(notify_db,
                notify_db_session,
                service=None,
-               template=None):
+               template=None,
+               notification_count=1):
     if service is None:
         service = sample_service(notify_db, notify_db_session)
     if template is None:
@@ -214,7 +215,7 @@ def sample_job(notify_db,
         'bucket_name': bucket_name,
         'file_name': file_name,
         'original_file_name': 'some.csv',
-        'notification_count': 1
+        'notification_count': notification_count
     }
     job = Job(**data)
     dao_create_job(job)