diff --git a/webapp/home/test/data/notices.py b/webapp/home/test/data/notices.py index aa833ae1..6b90610d 100644 --- 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 index 77235ae4..d787bdac 100644 --- 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: