This commit is contained in:
asaki222 2024-11-06 17:18:19 -05:00
parent 9a5ed14e88
commit 45f8a85348
No known key found for this signature in database
GPG key ID: 2C4F802060E06EA4
2 changed files with 2 additions and 6 deletions

View file

@ -682,7 +682,6 @@ class ExportDataTest(MockDbForIndividualTests, MockEppLib):
def test_domain_request_growth(self):
"""Shows submitted requests within a date range, sorted"""
# Remove "Submitted at" because we can't guess this immutable, dynamically generated test data
print("we are in here")
columns = [
"Domain request",
"Domain type",
@ -693,7 +692,6 @@ class ExportDataTest(MockDbForIndividualTests, MockEppLib):
# Create a CSV file in memory
csv_file = StringIO()
# Call the export functions
print("something")
DomainRequestGrowth.export_data_to_csv(
csv_file,
start_date=self.start_date.strftime("%Y-%m-%d"),
@ -701,11 +699,9 @@ class ExportDataTest(MockDbForIndividualTests, MockEppLib):
)
# Reset the CSV file's position to the beginning
csv_file.seek(0)
print("uuuuu")
print(csv_file)
# Read the content into a variable
csv_content = csv_file.read()
print(csv_content)
expected_content = (
"Domain request,Domain type,Federal type\n"
"city3.gov,Federal,Executive\n"

View file

@ -1269,7 +1269,7 @@ class DomainRequestExport(BaseExport):
human_readable_federal_type = BranchChoices.get_branch_label(federal_type) if federal_type else None
# Handle the org_type field
org_type = model.get("generic_org_type")
org_type = model.get("generic_org_type") or model.get("organization_type")
human_readable_org_type = DomainRequest.OrganizationChoices.get_org_label(org_type) if org_type else None
# Handle the status field. Defaults to the wrong format.