Merge pull request #26 from maykinmedia/feature/25-traits

[#25] Add traits to factory
maykinmedia · May 16, 2024 · 6f25182 · 6f25182
2 parents f937b93 + 7abe5c9
commit 6f25182
Showing 1 changed file with 25 additions and 0 deletions.
diff --git a/backend/src/openarchiefbeheer/zaken/tests/factories.py b/backend/src/openarchiefbeheer/zaken/tests/factories.py
@@ -20,3 +20,28 @@ class ZaakFactory(factory.django.DjangoModelFactory):
 
     class Meta:
         model = Zaak
+
+    class Params:
+        with_expand = factory.Trait(
+            _expand={
+                "zaaktype": {
+                    "url": "http://catalogue-api.nl/zaaktypen/111-111-111",
+                    "selectielijst_procestype": {"nummer": 1},
+                },
+                "resultaat": {
+                    "resultaattype": "http://catalogue-api.nl/catalogi/api/v1/resultaattypen/111-111-111",
+                    "_expand": {
+                        "resultaattype": {
+                            "url": "http://catalogue-api.nl/catalogi/api/v1/resultaattypen/111-111-111",
+                            "archiefactietermijn": "P1D",
+                        }
+                    },
+                },
+            }
+        )
+        with_related_zaken = factory.Trait(
+            relevante_andere_zaken=[
+                "http://zaken-api.nl/zaken/api/v1/zaken/111-111-111",
+                "http://zaken-api.nl/zaken/api/v1/zaken/222-222-222",
+            ]
+        )