diff --git a/api/teams/tests/factories.py b/api/teams/tests/factories.py index 6a09ce1c1..182ae4bb2 100644 --- 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