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"""