Add DepartmentFactory

uktrade · Sep 28, 2021 · c59f81a · c59f81a
1 parent bc41b7e
commit c59f81a
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/api/teams/tests/factories.py b/api/teams/tests/factories.py
@@ -1,9 +1,16 @@
 import factory
-from api.teams.models import Team
+from api.teams.models import Team, Department
 
 
 class TeamFactory(factory.django.DjangoModelFactory):
     name = factory.Faker("word")
 
     class Meta:
         model = Team
+
+
+class DepartmentFactory(factory.django.DjangoModelFactory):
+    name = factory.Faker("word")
+
+    class Meta:
+        model = Department