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

Add bulk update

Browse source
This commit is contained in:
zandercymatics 2025-01-07 08:59:44 -07:00
parent 7e274920fe
commit d2804a64f5
No known key found for this signature in database
GPG key ID: FF4636ABEC9682B7
2 changed files with 18 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

11
src/registrar/management/commands/create_federal_portfolio.py
View file

@ -246,7 +246,16 @@ class Command(BaseCommand):
domain_request.suborganization_state_territory = domain_request.state_territory domain_request.suborganization_state_territory = domain_request.state_territory
self.updated_portfolios.add(portfolio) self.updated_portfolios.add(portfolio)
DomainRequest.objects.bulk_update(domain_requests, ["portfolio", "sub_organization"]) DomainRequest.objects.bulk_update(
domain_requests,
[
"portfolio",
"sub_organization",
"requested_suborganization",
"suborganization_city",
"suborganization_state_territory",
],
)
message = f"Added portfolio '{portfolio}' to {len(domain_requests)} domain requests." message = f"Added portfolio '{portfolio}' to {len(domain_requests)} domain requests."
TerminalHelper.colorful_logger(logger.info, TerminalColors.OKGREEN, message) TerminalHelper.colorful_logger(logger.info, TerminalColors.OKGREEN, message)

8
src/registrar/tests/test_management_scripts.py
View file

@ -1590,6 +1590,7 @@ class TestCreateFederalPortfolio(TestCase):
def test_script_adds_requested_suborganization_information(self): def test_script_adds_requested_suborganization_information(self):
"""Tests that the script adds the requested suborg fields for domain requests""" """Tests that the script adds the requested suborg fields for domain requests"""
# Create a new domain request with some errant spacing
custom_suborg_request = completed_domain_request( custom_suborg_request = completed_domain_request(
name="custom_org.gov", name="custom_org.gov",
status=DomainRequest.DomainRequestStatus.IN_REVIEW, status=DomainRequest.DomainRequestStatus.IN_REVIEW,
@ -1598,7 +1599,7 @@ class TestCreateFederalPortfolio(TestCase):
user=self.user, user=self.user,
organization_name=" requested org name ", organization_name=" requested org name ",
city="Austin ", city="Austin ",
state_territory=DomainRequest.StateTerritoryChoices.TEXAS state_territory=DomainRequest.StateTerritoryChoices.TEXAS,
) )
self.assertIsNone(custom_suborg_request.requested_suborganization) self.assertIsNone(custom_suborg_request.requested_suborganization)
@ -1607,10 +1608,13 @@ class TestCreateFederalPortfolio(TestCase):
# Run the script and test it # Run the script and test it
self.run_create_federal_portfolio(branch="executive", parse_requests=True) self.run_create_federal_portfolio(branch="executive", parse_requests=True)
custom_suborg_request.refresh_from_db()
self.assertEqual(custom_suborg_request.requested_suborganization, "requested org name") self.assertEqual(custom_suborg_request.requested_suborganization, "requested org name")
self.assertEqual(custom_suborg_request.suborganization_city, "Austin") self.assertEqual(custom_suborg_request.suborganization_city, "Austin")
self.assertEqual(custom_suborg_request.suborganization_state_territory, DomainRequest.StateTerritoryChoices.TEXAS) self.assertEqual(
custom_suborg_request.suborganization_state_territory, DomainRequest.StateTerritoryChoices.TEXAS
)
def test_create_multiple_portfolios_for_branch_executive(self): def test_create_multiple_portfolios_for_branch_executive(self):
"""Tests creating all portfolios under a given branch""" """Tests creating all portfolios under a given branch"""