FIND-81:update tests

test/ciim/factories.py

@@ -1,15 +1,14 @@
 def create_record(
     template_details=None,
     iaid="C0000000",
-    source="CAT",
 ):
     """
     Return a sample response for a record.
     Useful for tidying up tests where response needs to be mocked.
     otherwise creates a record with iaid
     """
     if not template_details:
-        template_details = {"iaid": iaid, "source": source}
+        template_details = {"iaid": iaid}
 
     # a typical record
     record = {"@template": {"details": template_details}}

test/records/test_views.py

@@ -1,6 +1,7 @@
 from test.ciim.factories import create_record, create_response
 
 import responses
+from app.records.models import Record
 from django.conf import settings
 from django.test import TestCase
 
@@ -36,7 +37,9 @@ def test_record_rendered_for_single_result(self):
             f"{settings.CLIENT_BASE_URL}/get",
             json=create_response(
                 records=[
-                    create_record(iaid="C123456"),
+                    create_record(
+                        template_details={"iaid": "C123456", "source": "CAT"}
+                    ),
                 ]
             ),
         )
@@ -56,3 +59,7 @@ def test_record_rendered_for_single_result(self):
         self.assertEqual(
             response.context_data.get("page_title"), "Catalogue ID: C123456"
         )
+        self.assertIsInstance(response.context_data.get("record"), Record)
+
+        # record attribute
+        self.assertEqual(response.context_data.get("record").source, "CAT")