CONCD-816 Create tests for cards in views

LibraryOfCongress · May 20, 2024 · 3f11661 · 3f11661
1 parent cadba0d
commit 3f11661
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/concordia/tests/test_views.py b/concordia/tests/test_views.py
@@ -39,6 +39,7 @@
     JSONAssertMixin,
     create_asset,
     create_campaign,
+    create_card_family,
     create_item,
     create_project,
     create_topic,
@@ -375,6 +376,9 @@ def test_asset_detail_view(self):
         )
         self.transcription.save()
 
+        asset.item.project.campaign.card_family = create_card_family()
+        asset.item.project.campaign.save()
+
         response = self.client.get(
             reverse(
                 "transcriptions:asset-detail",
@@ -388,6 +392,7 @@ def test_asset_detail_view(self):
         )
 
         self.assertEqual(response.status_code, 200)
+        self.assertIn("cards", response.context)
 
     @patch.object(Asset, "get_ocr_transcript")
     def test_generate_ocr_transcription(self, mock):

diff --git a/concordia/tests/utils.py b/concordia/tests/utils.py
@@ -8,6 +8,7 @@
     Asset,
     Campaign,
     Card,
+    CardFamily,
     Item,
     MediaType,
     Project,
@@ -224,6 +225,13 @@ def create_card(*, title="Test Card", do_save=True, **kwargs):
     return card
 
 
+def create_card_family(*, do_save=True, **kwargs):
+    card_family = CardFamily(**kwargs)
+    if do_save:
+        card_family.save()
+    return card_family
+
+
 class JSONAssertMixin(object):
     def assertValidJSON(self, response, expected_status=200):
         """