mirror of
https://github.com/cisagov/manage.get.gov.git
synced 2025-07-06 19:23:23 +02:00
Cleanup
This commit is contained in:
parent
1c7cf928d1
commit
f3489ae571
2 changed files with 6 additions and 5 deletions
|
@ -794,7 +794,6 @@ class MockDb(TestCase):
|
|||
UserDomainRole.objects.all().delete()
|
||||
User.objects.all().delete()
|
||||
DomainInvitation.objects.all().delete()
|
||||
Portfolio.objects.all().delete()
|
||||
cls.federal_agency_1.delete()
|
||||
cls.federal_agency_2.delete()
|
||||
|
||||
|
|
|
@ -361,16 +361,18 @@ class ExportDataTest(MockDbForIndividualTests, MockEppLib):
|
|||
self.assertNotIn(self.domain_2.name, 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.save()
|
||||
|
||||
csv_content = self._run_domain_data_type_user_export(request)
|
||||
|
||||
self.assertNotIn(self.domain_1.name, csv_content)
|
||||
self.assertNotIn(self.domain_3.name, csv_content)
|
||||
self.assertIn(self.domain_2.name, csv_content)
|
||||
|
||||
# Get the csv content
|
||||
csv_content = self._run_domain_data_type_user_export(request)
|
||||
self.domain_1.delete()
|
||||
self.domain_2.delete()
|
||||
self.domain_3.delete()
|
||||
portfolio.delete()
|
||||
|
||||
def _run_domain_data_type_user_export(self, request):
|
||||
"""Helper function to run the export_data_to_csv function on DomainDataTypeUser"""
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue