From b21c8de9a398d50bf062d6ce05422f8cb48df9fb Mon Sep 17 00:00:00 2001
From: Leo Hemsted <leo.hemsted@digital.cabinet-office.gov.uk>
Date: Tue, 30 Aug 2016 15:51:56 +0100
Subject: [PATCH] add new sample_email_notification fixture

dont include any of the parameters for now - only add em when we have an actual need for them
---
 tests/app/conftest.py | 46 +++++++++++++++++++++++++++++++++++++------
 1 file changed, 40 insertions(+), 6 deletions(-)

diff --git a/tests/app/conftest.py b/tests/app/conftest.py
index a14dbb5fc..c3d2389d3 100644
--- a/tests/app/conftest.py
+++ b/tests/app/conftest.py
@@ -1,8 +1,8 @@
-import requests_mock
-import pytest
 import uuid
 from datetime import (datetime, date)
 
+import requests_mock
+import pytest
 from flask import current_app
 
 from app import db
@@ -28,7 +28,6 @@ from app.dao.jobs_dao import dao_create_job
 from app.dao.notifications_dao import dao_create_notification
 from app.dao.invited_user_dao import save_invited_user
 from app.clients.sms.firetext import FiretextClient
-from app.clients.sms.mmg import MMGClient
 
 
 @pytest.yield_fixture
@@ -272,9 +271,9 @@ def sample_job(notify_db,
 
 @pytest.fixture(scope='function')
 def sample_job_with_placeholdered_template(
-    notify_db,
-    notify_db_session,
-    service=None
+        notify_db,
+        notify_db_session,
+        service=None
 ):
     return sample_job(
         notify_db,
@@ -370,6 +369,41 @@ def sample_notification(notify_db,
     return notification
 
 
+@pytest.fixture(scope='function')
+def sample_email_notification(notify_db, notify_db_session):
+    created_at = datetime.utcnow()
+    service = sample_service(notify_db, notify_db_session)
+    template = sample_email_template(notify_db, notify_db_session, service=service)
+    job = sample_job(notify_db, notify_db_session, service=service, template=template)
+
+    notification_id = uuid.uuid4()
+
+    to = 'foo@bar.com'
+
+    data = {
+        'id': notification_id,
+        'to': to,
+        'job_id': job.id,
+        'job': job,
+        'service_id': service.id,
+        'service': service,
+        'template': template,
+        'template_version': template.version,
+        'status': 'created',
+        'reference': None,
+        'created_at': created_at,
+        'billable_units': 0,
+        'personalisation': None,
+        'notification_type': template.template_type,
+        'api_key_id': None,
+        'key_type': KEY_TYPE_NORMAL,
+        'job_row_number': 1
+    }
+    notification = Notification(**data)
+    dao_create_notification(notification)
+    return notification
+
+
 @pytest.fixture(scope='function')
 def mock_statsd_inc(mocker):
     return mocker.patch('app.statsd_client.incr')