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

Cleanup

Browse source
This commit is contained in:
zandercymatics 2024-07-31 08:43:49 -06:00
parent 1c7cf928d1
commit f3489ae571
No known key found for this signature in database
GPG key ID: FF4636ABEC9682B7
2 changed files with 6 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

1
src/registrar/tests/common.py
View file

@ -794,7 +794,6 @@ class MockDb(TestCase):
UserDomainRole.objects.all().delete() UserDomainRole.objects.all().delete()
User.objects.all().delete() User.objects.all().delete()
DomainInvitation.objects.all().delete() DomainInvitation.objects.all().delete()
Portfolio.objects.all().delete()
cls.federal_agency_1.delete() cls.federal_agency_1.delete()
cls.federal_agency_2.delete() cls.federal_agency_2.delete()

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

@ -361,16 +361,18 @@ class ExportDataTest(MockDbForIndividualTests, MockEppLib):
self.assertNotIn(self.domain_2.name, csv_content) self.assertNotIn(self.domain_2.name, csv_content)
# Get the csv content # Get the csv content
csv_content = self._run_domain_data_type_user_export(request)
self.user.portfolio_roles = [User.UserPortfolioRoleChoices.ORGANIZATION_MEMBER] self.user.portfolio_roles = [User.UserPortfolioRoleChoices.ORGANIZATION_MEMBER]
self.user.save() self.user.save()
csv_content = self._run_domain_data_type_user_export(request)
self.assertNotIn(self.domain_1.name, csv_content) self.assertNotIn(self.domain_1.name, csv_content)
self.assertNotIn(self.domain_3.name, csv_content) self.assertNotIn(self.domain_3.name, csv_content)
self.assertIn(self.domain_2.name, csv_content) self.assertIn(self.domain_2.name, csv_content)
self.domain_1.delete()
# Get the csv content self.domain_2.delete()
csv_content = self._run_domain_data_type_user_export(request) self.domain_3.delete()
portfolio.delete()
def _run_domain_data_type_user_export(self, request): def _run_domain_data_type_user_export(self, request):
"""Helper function to run the export_data_to_csv function on DomainDataTypeUser""" """Helper function to run the export_data_to_csv function on DomainDataTypeUser"""