Fix tests for common.py to reflect fk FederalAgency

This commit is contained in:
Erin 2024-05-02 11:43:41 -07:00
parent 763e7b6351
commit 98183a961f
No known key found for this signature in database
GPG key ID: 1CAD275313C62460

View file

@ -540,8 +540,8 @@ class MockDb(TestCase):
self.end_date = current_date + timedelta(days=2)
self.start_date = current_date - timedelta(days=2)
self.federal_agency_1, _ = FederalAgency.get_or_create(name="World War I Centennial Commission")
self.federal_agency_2, _ = FederalAgency.get_or_create(name="Armed Forces Retirement Home")
self.federal_agency_1, _ = FederalAgency.objects.get_or_create(agency="World War I Centennial Commission")
self.federal_agency_2, _ = FederalAgency.objects.get_or_create(agency="Armed Forces Retirement Home")
self.domain_1, _ = Domain.objects.get_or_create(
name="cdomain1.gov", state=Domain.State.READY, first_ready=timezone.now()
@ -586,7 +586,7 @@ class MockDb(TestCase):
creator=self.user,
domain=self.domain_1,
generic_org_type="federal",
federal_agency=federal_agency_1,
federal_agency=self.federal_agency_1,
federal_type="executive",
is_election_board=False,
)
@ -597,63 +597,63 @@ class MockDb(TestCase):
creator=self.user,
domain=self.domain_3,
generic_org_type="federal",
federal_agency=federal_agency_2,
federal_agency=self.federal_agency_2,
is_election_board=False,
)
self.domain_information_4, _ = DomainInformation.objects.get_or_create(
creator=self.user,
domain=self.domain_4,
generic_org_type="federal",
federal_agency=federal_agency_2,
federal_agency=self.federal_agency_2,
is_election_board=False,
)
self.domain_information_5, _ = DomainInformation.objects.get_or_create(
creator=self.user,
domain=self.domain_5,
generic_org_type="federal",
federal_agency=federal_agency_2,
federal_agency=self.federal_agency_2,
is_election_board=False,
)
self.domain_information_6, _ = DomainInformation.objects.get_or_create(
creator=self.user,
domain=self.domain_6,
generic_org_type="federal",
federal_agency=federal_agency_2,
federal_agency=self.federal_agency_2,
is_election_board=False,
)
self.domain_information_7, _ = DomainInformation.objects.get_or_create(
creator=self.user,
domain=self.domain_7,
generic_org_type="federal",
federal_agency=federal_agency_2,
federal_agency=self.federal_agency_2,
is_election_board=False,
)
self.domain_information_8, _ = DomainInformation.objects.get_or_create(
creator=self.user,
domain=self.domain_8,
generic_org_type="federal",
federal_agency=federal_agency_2,
federal_agency=self.federal_agency_2,
is_election_board=False,
)
self.domain_information_9, _ = DomainInformation.objects.get_or_create(
creator=self.user,
domain=self.domain_9,
generic_org_type="federal",
federal_agency=federal_agency_2,
federal_agency=self.federal_agency_2,
is_election_board=False,
)
self.domain_information_10, _ = DomainInformation.objects.get_or_create(
creator=self.user,
domain=self.domain_10,
generic_org_type="federal",
federal_agency=federal_agency_2,
federal_agency=self.federal_agency_2,
is_election_board=False,
)
self.domain_information_11, _ = DomainInformation.objects.get_or_create(
creator=self.user,
domain=self.domain_11,
generic_org_type="federal",
federal_agency=federal_agency_1,
federal_agency=self.federal_agency_1,
federal_type="executive",
is_election_board=False,
)