Fix tests

usegalaxy-au · Jul 15, 2024 · ca78b8f · ca78b8f
1 parent cfbf91d
commit ca78b8f
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/webapp/home/test/data/notices.py b/webapp/home/test/data/notices.py
@@ -30,10 +30,7 @@
             "is_published": True,
         },
         "relations": {
-            "subsites": [
-                TEST_LABS[0],
-                TEST_LABS[1]
-            ],
+            "subsites": [TEST_LABS[0]],
         },
     },
     {

diff --git a/webapp/home/tests.py b/webapp/home/tests.py
@@ -6,7 +6,7 @@
 from pathlib import Path
 
 from .lab_export import ExportSubsiteContext
-from .models import Notice
+from .models import Notice, Subsite
 from .test.data import (
     TEST_NOTICES,
     MOCK_REQUESTS,
@@ -46,7 +46,11 @@ def setUp(self) -> None:
         super().setUp()
         self.client = Client()
         for notice in TEST_NOTICES:
+            subsites = notice['relations']['subsites']
             notice = Notice.objects.create(**notice['data'])
+            for subsite in subsites:
+                subsite = Subsite.objects.get(name=subsite['name'])
+                notice.subsites.add(subsite)
         for tag in TEST_TAGS:
             Tag.objects.create(**tag)
         for supporter in TEST_SUPPORTERS: