diff --git a/src/registrar/management/commands/populate_domain_updated_federal_agency.py b/src/registrar/management/commands/populate_domain_updated_federal_agency.py index 176849413..5443d3411 100644 --- a/src/registrar/management/commands/populate_domain_updated_federal_agency.py +++ b/src/registrar/management/commands/populate_domain_updated_federal_agency.py @@ -41,7 +41,7 @@ class Command(BaseCommand): Renames the Federal Agency to the correct new naming for both Domain Information and Domain Requests objects. - NOTE: If it's NULL for a domain request, we skip it as + NOTE: If it's None for a domain request, we skip it as a user most likely hasn't gotten to it yet. """ logger.info("Transferring federal agencies to FederalAgency object") @@ -65,7 +65,7 @@ class Command(BaseCommand): domain_info.updated_federal_agency = federal_agency_row domain_infos_to_update.append(domain_info) logger.info( - f"DomainInformation {domain_info} updated_federal_agency set to:" + f"DomainInformation {domain_info} => updated_federal_agency set to:" f"{domain_info.updated_federal_agency}" ) except Exception as err: @@ -83,7 +83,7 @@ class Command(BaseCommand): domain_request.updated_federal_agency = federal_agency_row domain_requests_to_update.append(domain_request) logger.info( - f"DomainRequest {domain_request} updated_federal_agency set to:" + f"DomainRequest {domain_request} => updated_federal_agency set to:" f"{domain_request.updated_federal_agency}" ) except Exception as err: diff --git a/src/registrar/tests/test_management_scripts.py b/src/registrar/tests/test_management_scripts.py index 746c4f8ab..1bb6c59b7 100644 --- a/src/registrar/tests/test_management_scripts.py +++ b/src/registrar/tests/test_management_scripts.py @@ -746,7 +746,7 @@ class TestDiscloseEmails(MockEppLib): # TODO in #1793: Remove this whole test class -class TestRenamingFederalAgency(TestCase): +class TestPopulateDomainUpdatedFederalAgency(TestCase): def setUp(self): super().setUp() @@ -802,11 +802,7 @@ class TestRenamingFederalAgency(TestCase): execute the populate_domain_updated_federal_agency command. """ with less_console_noise(): - with patch( - "registrar.management.commands.utility.terminal_helper.TerminalHelper.query_yes_no_exit", # noqa - return_value=True, - ): - call_command("populate_domain_updated_federal_agency") + call_command("populate_domain_updated_federal_agency") def test_domain_information_renaming_federal_agency_success(self): """