add factory for ContactType

yverdon · Aug 15, 2023 · 8026e20 · 8026e20
1 parent ed43ec4
commit 8026e20
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/geocity/tests/factories.py b/geocity/tests/factories.py
@@ -412,11 +412,19 @@ def administrative_entities(self, create, extracted, **kwargs):
         self.administrative_entities.add(*extracted)
 
 
+class ContactTypeFactory(factory.django.DjangoModelFactory):
+    class Meta:
+        model = submissions_models.ContactForm
+
+    name = factory.Faker("word")
+
+
 class ContactFormFactory(factory.django.DjangoModelFactory):
     class Meta:
         model = submissions_models.ContactForm
 
     form_category = factory.SubFactory(FormCategoryFactory)
+    type = factory.SubFactory(ContactTypeFactory)
 
     @factory.post_generation
     def integrator(self, create, extracted, **kwargs):