zydronium/manage.get.gov
1
0
Fork 0
mirror of https://github.com/cisagov/manage.get.gov.git synced 2025-08-16 14:34:10 +02:00
Code Issues Projects Releases Packages Wiki Activity

Fix tests for common.py to reflect fk FederalAgency

Browse source
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
1 changed files with 12 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

24
src/registrar/tests/common.py
View file

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