From cb87e668014aa57b74b44cdc8679d43da75941db Mon Sep 17 00:00:00 2001
From: Rebecca Law <rebecca.law@digital.cabinet-office.gov.uk>
Date: Tue, 7 Jan 2020 12:19:41 +0000
Subject: [PATCH] Add client_reference to the serialised notification returned
 for the csv reports.

---
 app/models.py              | 1 +
 tests/app/job/test_rest.py | 3 ++-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/app/models.py b/app/models.py
index 3a4b83f53..7ab5c5e89 100644
--- a/app/models.py
+++ b/app/models.py
@@ -1569,6 +1569,7 @@ class Notification(db.Model):
         serialized = {
             "row_number": '' if self.job_row_number is None else self.job_row_number + 1,
             "recipient": self.to,
+            "client_reference": self.client_reference or '',
             "template_name": self.template.name,
             "template_type": self.template.template_type,
             "job_name": self.job.original_file_name if self.job else '',
diff --git a/tests/app/job/test_rest.py b/tests/app/job/test_rest.py
index 79645049d..e94cb2076 100644
--- a/tests/app/job/test_rest.py
+++ b/tests/app/job/test_rest.py
@@ -794,7 +794,8 @@ def test_get_all_notifications_for_job_returns_csv_format(admin_request, sample_
         'job_name',
         'status',
         'row_number',
-        'recipient'
+        'recipient',
+        'client_reference'
     }