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

lint

Browse source
This commit is contained in:
Rachid Mrad 2023-10-27 15:03:34 -04:00
parent 98a9604ef3
commit 4803aaf971
No known key found for this signature in database
GPG key ID: EF38E4CEC4A8F3CF
3 changed files with 59 additions and 51 deletions
Download patch file Download diff file Expand all files Collapse all files

8
src/registrar/admin.py
View file

@ -762,18 +762,14 @@ class DomainAdmin(ListHeaderAdmin):
def export_data_full(self, request):
# Smaller export based on 1
response = HttpResponse(content_type="text/csv")
response[
"Content-Disposition"
] = 'attachment; filename="current-full.csv"'
response["Content-Disposition"] = 'attachment; filename="current-full.csv"'
csv_export.export_data_full_to_csv(response)
return response
def export_data_federal(self, request):
# Federal only
response = HttpResponse(content_type="text/csv")
response[
"Content-Disposition"
] = 'attachment; filename="current-federal.csv"'
response["Content-Disposition"] = 'attachment; filename="current-federal.csv"'
csv_export.export_data_federal_to_csv(response)
return response

67
src/registrar/tests/test_reports.py
View file

@ -1,21 +1,15 @@
from django.test import TestCase
from django.core.files import File
from io import StringIO
import csv
from registrar.models.domain_information import DomainInformation
from registrar.models.domain import Domain
from registrar.models.user import User
from django.contrib.auth import get_user_model
from registrar.utility.csv_export import (
export_domains_to_writer,
export_data_type_to_csv,
export_data_full_to_csv,
export_data_federal_to_csv,
)
from registrar.utility.csv_export import export_domains_to_writer
class ExportDataTest(TestCase):
def setUp(self):
username = "test_user"
first_name = "First"
last_name = "Last"
@ -24,11 +18,21 @@ class ExportDataTest(TestCase):
username=username, first_name=first_name, last_name=last_name, email=email
)
self.domain_1, _ = Domain.objects.get_or_create(name="cdomain1.gov", state=Domain.State.READY)
self.domain_2, _ = Domain.objects.get_or_create(name="adomain2.gov", state=Domain.State.READY)
self.domain_3, _ = Domain.objects.get_or_create(name="ddomain3.gov", state=Domain.State.READY)
self.domain_4, _ = Domain.objects.get_or_create(name="bdomain4.gov", state=Domain.State.UNKNOWN)
self.domain_4, _ = Domain.objects.get_or_create(name="bdomain4.gov", state=Domain.State.UNKNOWN)
self.domain_1, _ = Domain.objects.get_or_create(
name="cdomain1.gov", state=Domain.State.READY
)
self.domain_2, _ = Domain.objects.get_or_create(
name="adomain2.gov", state=Domain.State.READY
)
self.domain_3, _ = Domain.objects.get_or_create(
name="ddomain3.gov", state=Domain.State.READY
)
self.domain_4, _ = Domain.objects.get_or_create(
name="bdomain4.gov", state=Domain.State.UNKNOWN
)
self.domain_4, _ = Domain.objects.get_or_create(
name="bdomain4.gov", state=Domain.State.UNKNOWN
)
self.domain_information_1, _ = DomainInformation.objects.get_or_create(
creator=self.user,
@ -60,7 +64,6 @@ class ExportDataTest(TestCase):
User.objects.all().delete()
super().tearDown()
def test_export_domains_to_writer(self):
"""Test that export_domains_to_writer returns the
existing domain, test that sort by domain name works,
@ -100,17 +103,20 @@ class ExportDataTest(TestCase):
# We expect READY domains,
# sorted alphabetially by domain name
expected_content = """\
Domain name,Domain type,Federal agency,Organization name,City,State,AO,AO email,Submitter,Submitter title,Submitter email,Submitter phone,Security Contact Email,Status
adomain2.gov,interstate,,,,, , , , , , , ,ready
cdomain1.gov,federal,World War I Centennial Commission,,,, , , , , , , ,ready
ddomain3.gov,federal,Armed Forces Retirement Home,,,, , , , , , , ,ready
"""
expected_content = (
"Domain name,Domain type,Federal agency,Organization name,City,State,AO,"
"AO email, Submitter,Submitter title,Submitter email,Submitter phone,"
"Security Contact Email,Status\n"
"adomain2.gov,interstate,,,,, , , , , , , ,ready\n"
"cdomain1.gov,federal,World War I Centennial Commission,,,"
", , , , , , , ,ready\n"
"ddomain3.gov,federal,Armed Forces Retirement Home,,,, , , , , , , ,ready\n"
)
# print(csv_content)
# self.maxDiff = None
# Normalize line endings and remove leading/trailing whitespace
csv_content = csv_content.replace('\r\n', '\n').strip()
csv_content = csv_content.replace("\r\n", "\n").strip()
expected_content = expected_content.strip()
self.assertEqual(csv_content, expected_content)
@ -132,7 +138,10 @@ ddomain3.gov,federal,Armed Forces Retirement Home,,,, , , , , , , ,ready
"Security Contact Email",
]
sort_fields = ["domain__name", "federal_agency", "organization_type"]
filter_condition = {"organization_type__icontains": "federal", "domain__state": Domain.State.READY}
filter_condition = {
"organization_type__icontains": "federal",
"domain__state": Domain.State.READY,
}
# Call the export function
export_domains_to_writer(writer, columns, sort_fields, filter_condition)
@ -146,17 +155,17 @@ ddomain3.gov,federal,Armed Forces Retirement Home,,,, , , , , , , ,ready
# We expect READY domains,
# federal only
# sorted alphabetially by domain name
expected_content = """\
Domain name,Domain type,Federal agency,Organization name,City,State,Security Contact Email
cdomain1.gov,federal,World War I Centennial Commission,,,,
ddomain3.gov,federal,Armed Forces Retirement Home,,,,
"""
expected_content = (
"Domain name,Domain type,Federal agency,Organization name,City,"
"State,Security Contact Email\n"
"cdomain1.gov,federal,World War I Centennial Commission,,,,\n"
"ddomain3.gov,federal,Armed Forces Retirement Home,,,,\n"
)
# print(csv_content)
# self.maxDiff = None
# Normalize line endings and remove leading/trailing whitespace
csv_content = csv_content.replace('\r\n', '\n').strip()
csv_content = csv_content.replace("\r\n", "\n").strip()
expected_content = expected_content.strip()
self.assertEqual(csv_content, expected_content)

5
src/registrar/utility/csv_export.py
View file

@ -109,5 +109,8 @@ def export_data_federal_to_csv(csv_file):
"Security Contact Email",
]
sort_fields = ["domain__name", "federal_agency", "organization_type"]
filter_condition = {"organization_type__icontains": "federal", "domain__state": Domain.State.READY}
filter_condition = {
"organization_type__icontains": "federal",
"domain__state": Domain.State.READY,
}
export_domains_to_writer(writer, columns, sort_fields, filter_condition)