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

Updated federal agency with new foreign key field

Browse source
This commit is contained in:
Erin 2024-05-02 10:24:11 -07:00
parent e34dfbccdb
commit f9eb4746a1
No known key found for this signature in database
GPG key ID: 1CAD275313C62460
6 changed files with 55 additions and 26 deletions
Download patch file Download diff file Expand all files Collapse all files

13
src/registrar/fixtures_domain_requests.py
View file

@ -9,6 +9,7 @@ from registrar.models import (
DraftDomain,
Contact,
Website,
FederalAgency
)
fake = Faker()
@ -101,12 +102,6 @@ class DomainRequestFixture:
# TODO for a future ticket: Allow for more than just "federal" here
da.generic_org_type = app["generic_org_type"] if "generic_org_type" in app else "federal"
da.federal_agency = (
app["federal_agency"]
if "federal_agency" in app
# Random choice of agency for selects, used as placeholders for testing.
else random.choice(DomainRequest.AGENCIES) # nosec
)
da.submission_date = fake.date()
da.federal_type = (
app["federal_type"]
@ -146,6 +141,12 @@ class DomainRequestFixture:
da.requested_domain, _ = DraftDomain.objects.get_or_create(name=app["requested_domain"])
else:
da.requested_domain = DraftDomain.objects.create(name=cls.fake_dot_gov())
if not da.federal_agency:
if "federal_agency" in app and app["federal_agency"] is not None:
da.federal_agency, _ = FederalAgency.objects.get_or_create(name=app["federal_agency"])
else:
federal_agencies = FederalAgency.objects.all()
da.federal_agency = random.choice(federal_agencies)
@classmethod
def _set_many_to_many_relations(cls, da: DomainRequest, app: dict):