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

fixed unit tests

Browse source
This commit is contained in:
CocoByte 2024-09-26 08:02:13 -10:00
parent 3dbb8f0673
commit 35fec1ed89
No known key found for this signature in database
GPG key ID: BBFAA2526384C97F
1 changed files with 11 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

15
src/registrar/tests/test_views_portfolio.py
View file

@ -703,7 +703,7 @@ class TestPortfolio(WebTest):
# --- admin # --- admin
UserPortfolioPermission.objects.filter(user=self.user, portfolio=self.portfolio).update( UserPortfolioPermission.objects.filter(user=self.user, portfolio=self.portfolio).update(
role=UserPortfolioRoleChoices.ORGANIZATION_ADMIN, roles=[UserPortfolioRoleChoices.ORGANIZATION_ADMIN],
) )
# Verify that the user cannot access the members page # Verify that the user cannot access the members page
@ -723,6 +723,7 @@ class TestPortfolio(WebTest):
UserPortfolioPermission.objects.get_or_create( UserPortfolioPermission.objects.get_or_create(
user=self.user, user=self.user,
portfolio=self.portfolio, portfolio=self.portfolio,
roles=[UserPortfolioRoleChoices.ORGANIZATION_ADMIN],
additional_permissions=[ additional_permissions=[
UserPortfolioPermissionChoices.VIEW_PORTFOLIO, UserPortfolioPermissionChoices.VIEW_PORTFOLIO,
UserPortfolioPermissionChoices.VIEW_MEMBERS, UserPortfolioPermissionChoices.VIEW_MEMBERS,
@ -733,7 +734,7 @@ class TestPortfolio(WebTest):
# This will redirect the user to the members page. # This will redirect the user to the members page.
response = self.app.get(reverse("members")) response = self.app.get(reverse("members"))
# Make sure the page loaded # Make sure the page loaded
self.assertEqual(response.status_code, 200) self.assertContains(response, "Members")
@less_console_noise_decorator @less_console_noise_decorator
@override_flag("organization_feature", active=True) @override_flag("organization_feature", active=True)
@ -747,6 +748,7 @@ class TestPortfolio(WebTest):
UserPortfolioPermission.objects.get_or_create( UserPortfolioPermission.objects.get_or_create(
user=self.user, user=self.user,
portfolio=self.portfolio, portfolio=self.portfolio,
roles=[UserPortfolioRoleChoices.ORGANIZATION_ADMIN],
additional_permissions=[ additional_permissions=[
UserPortfolioPermissionChoices.VIEW_PORTFOLIO, UserPortfolioPermissionChoices.VIEW_PORTFOLIO,
UserPortfolioPermissionChoices.VIEW_MEMBERS, UserPortfolioPermissionChoices.VIEW_MEMBERS,
@ -760,8 +762,11 @@ class TestPortfolio(WebTest):
# Verify that the user can access the members page # Verify that the user can access the members page
# This will redirect the user to the members page. # This will redirect the user to the members page.
self.client.force_login(self.user)
response = self.client.get(reverse("members"), follow=True) response = self.client.get(reverse("members"), follow=True)
# Make sure the page loaded
self.assertEqual(response.status_code, 200)
# Check that the manage settings appear in the DOM
self.assertContains(response, '<use xlink:href="/public/img/sprite.svg#settings"></use>') self.assertContains(response, '<use xlink:href="/public/img/sprite.svg#settings"></use>')
self.assertContains(response, "Manage") self.assertContains(response, "Manage")
@ -790,10 +795,12 @@ class TestPortfolio(WebTest):
# Verify that the user can access the members page # Verify that the user can access the members page
# This will redirect the user to the members page. # This will redirect the user to the members page.
self.client.force_login(self.user)
response = self.client.get(reverse("members"), follow=True) response = self.client.get(reverse("members"), follow=True)
# Make sure the page loaded # Make sure the page loaded
self.assertEqual(response.status_code, 200) self.assertEqual(response.status_code, 200)
# Check that the view-only settings appear in the DOM
self.assertContains(response, '<use xlink:href="/public/img/sprite.svg#visibility"></use>')
self.assertContains(response, "View") self.assertContains(response, "View")
@less_console_noise_decorator @less_console_noise_decorator