From 16fc4462de84c9c522da9eee7bab3e66347a808a Mon Sep 17 00:00:00 2001 From: Rachid Mrad Date: Thu, 12 Sep 2024 12:52:16 -0400 Subject: [PATCH] unit tests for the extra column in the template --- src/registrar/tests/test_views_portfolio.py | 38 +++++++++++++++++++++ 1 file changed, 38 insertions(+) diff --git a/src/registrar/tests/test_views_portfolio.py b/src/registrar/tests/test_views_portfolio.py index 069ff8785..aaafa3262 100644 --- a/src/registrar/tests/test_views_portfolio.py +++ b/src/registrar/tests/test_views_portfolio.py @@ -750,6 +750,44 @@ def test_main_nav_when_user_has_view_but_not_edit_permissions(self): # create new request btn self.assertNotContains(requests_page, "Start a new domain request") + @less_console_noise_decorator + @override_flag("organization_feature", active=True) + @override_flag("organization_requests", active=True) + def test_organization_requests_additional_column(self): + """The requests table has a column for created at""" + self.app.set_user(self.user.username) + + UserPortfolioPermission.objects.get_or_create( + user=self.user, + portfolio=self.portfolio, + additional_permissions=[ + UserPortfolioPermissionChoices.VIEW_PORTFOLIO, + UserPortfolioPermissionChoices.EDIT_REQUESTS, + UserPortfolioPermissionChoices.VIEW_ALL_REQUESTS, + UserPortfolioPermissionChoices.EDIT_REQUESTS, + ], + ) + + home = self.app.get(reverse("home")).follow() + + self.assertContains(home, "Hotel California") + self.assertContains(home, "Domain requests") + + domain_requests = self.app.get(reverse("domain-requests")) + self.assertEqual(domain_requests.status_code, 200) + + self.assertContains(domain_requests, "Created by") + + @less_console_noise_decorator + def test_no_org_requests_no_additional_column(self): + """The requests table does not have a column for created at""" + self.app.set_user(self.user.username) + + home = self.app.get(reverse("home")) + + self.assertContains(home, "Domain requests") + self.assertNotContains(home, "Created by") + @less_console_noise_decorator def test_portfolio_cache_updates_when_modified(self): """Test that the portfolio in session updates when the portfolio is modified"""