From f3489ae5714ea6aa91f631bba28bb1178a1ab30b Mon Sep 17 00:00:00 2001 From: zandercymatics <141044360+zandercymatics@users.noreply.github.com> Date: Wed, 31 Jul 2024 08:43:49 -0600 Subject: [PATCH] Cleanup --- src/registrar/tests/common.py | 1 - src/registrar/tests/test_reports.py | 10 ++++++---- 2 files changed, 6 insertions(+), 5 deletions(-) diff --git a/src/registrar/tests/common.py b/src/registrar/tests/common.py index d9c9001ce..69e3fa174 100644 --- a/src/registrar/tests/common.py +++ b/src/registrar/tests/common.py @@ -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() diff --git a/src/registrar/tests/test_reports.py b/src/registrar/tests/test_reports.py index e18439d88..06e6f264d 100644 --- a/src/registrar/tests/test_reports.py +++ b/src/registrar/tests/test_reports.py @@ -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"""