Fix factory to add department to team automatically

uktrade · Sep 28, 2021 · 0eaa154 · 0eaa154
1 parent ff00cb2
commit 0eaa154
Showing 1 changed file with 11 additions and 5 deletions.
diff --git a/api/teams/tests/factories.py b/api/teams/tests/factories.py
@@ -2,15 +2,21 @@
 from api.teams.models import Team, Department
 
 
-class TeamFactory(factory.django.DjangoModelFactory):
-    name = factory.Faker("word")
+class DepartmentFactory(factory.django.DjangoModelFactory):
+    name = factory.Iterator(["HMRC", "MOD", "BEIS", "DIT", "NCSC"])
+    # Commented out since probably we want to manually add this.
+    # team = factory.RelatedFactory(
+    #     TeamFactory,
+    #     factory_related_name='department',
+    # )
 
     class Meta:
-        model = Team
+        model = Department
 
 
-class DepartmentFactory(factory.django.DjangoModelFactory):
+class TeamFactory(factory.django.DjangoModelFactory):
     name = factory.Faker("word")
+    department = factory.SubFactory(DepartmentFactory)
 
     class Meta:
-        model = Department
+        model = Team