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

updating tests from org to portfolio

Browse source
This commit is contained in:
David Kennedy 2024-06-20 16:43:57 -04:00
parent 7072125992
commit aa652fac1b
No known key found for this signature in database
GPG key ID: 6528A5386E66B96B
1 changed files with 13 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

26
src/registrar/tests/test_views.py
View file

@ -908,7 +908,7 @@ class UserProfileTests(TestWithUser, WebTest):
self.assertContains(profile_page, "Your profile has been updated") self.assertContains(profile_page, "Your profile has been updated")
class OrganizationsTests(TestWithUser, WebTest): class PortfoliosTests(TestWithUser, WebTest):
"""A series of tests that target the organizations""" """A series of tests that target the organizations"""
# csrf checks do not work well with WebTest. # csrf checks do not work well with WebTest.
@ -939,33 +939,33 @@ class OrganizationsTests(TestWithUser, WebTest):
self.app.set_cookie(settings.SESSION_COOKIE_NAME, session_id) self.app.set_cookie(settings.SESSION_COOKIE_NAME, session_id)
@less_console_noise_decorator @less_console_noise_decorator
def test_middleware_redirects_to_organization_homepage(self): def test_middleware_redirects_to_portfolio_homepage(self):
"""Tests that a user is redirected to the org homepage when organization_feature is on and """Tests that a user is redirected to the portfolio homepage when organization_feature is on and
a portfolio belongs to the user, test for the special h1s which only exist in that version a portfolio belongs to the user, test for the special h1s which only exist in that version
of the homepage""" of the homepage"""
self.app.set_user(self.user.username) self.app.set_user(self.user.username)
with override_flag("organization_feature", active=True): with override_flag("organization_feature", active=True):
# This will redirect the user to the org page. # This will redirect the user to the portfolio page.
# Follow implicity checks if our redirect is working. # Follow implicity checks if our redirect is working.
org_page = self.app.get(reverse("home")).follow() portfolio_page = self.app.get(reverse("home")).follow()
self._set_session_cookie() self._set_session_cookie()
# Assert that we're on the right page # Assert that we're on the right page
self.assertContains(org_page, self.portfolio.organization_name) self.assertContains(portfolio_page, self.portfolio.organization_name)
self.assertContains(org_page, "<h1>Domains</h1>") self.assertContains(portfolio_page, "<h1>Domains</h1>")
@less_console_noise_decorator @less_console_noise_decorator
def test_no_redirect_when_org_flag_false(self): def test_no_redirect_when_org_flag_false(self):
"""No redirect so no follow, """No redirect so no follow,
implicitely test for the presense of the h2 by looking up its id""" implicitely test for the presense of the h2 by looking up its id"""
self.app.set_user(self.user.username) self.app.set_user(self.user.username)
org_page = self.app.get(reverse("home")) home_page = self.app.get(reverse("home"))
self._set_session_cookie() self._set_session_cookie()
self.assertNotContains(org_page, self.portfolio.organization_name) self.assertNotContains(home_page, self.portfolio.organization_name)
self.assertContains(org_page, 'id="domain-requests-header"') self.assertContains(home_page, 'id="domain-requests-header"')
@less_console_noise_decorator @less_console_noise_decorator
def test_no_redirect_when_user_has_no_portfolios(self): def test_no_redirect_when_user_has_no_portfolios(self):
@ -974,9 +974,9 @@ class OrganizationsTests(TestWithUser, WebTest):
self.portfolio.delete() self.portfolio.delete()
self.app.set_user(self.user.username) self.app.set_user(self.user.username)
with override_flag("organization_feature", active=True): with override_flag("organization_feature", active=True):
org_page = self.app.get(reverse("home")) home_page = self.app.get(reverse("home"))
self._set_session_cookie() self._set_session_cookie()
self.assertNotContains(org_page, self.portfolio.organization_name) self.assertNotContains(home_page, self.portfolio.organization_name)
self.assertContains(org_page, 'id="domain-requests-header"') self.assertContains(home_page, 'id="domain-requests-header"')