mirror of
https://github.com/cisagov/manage.get.gov.git
synced 2025-07-27 04:58:42 +02:00
Add unit test
This commit is contained in:
parent
4be1e3e29e
commit
3005158b82
1 changed files with 30 additions and 1 deletions
|
@ -1,7 +1,7 @@
|
||||||
from django.urls import reverse
|
from django.urls import reverse
|
||||||
from api.tests.common import less_console_noise_decorator
|
from api.tests.common import less_console_noise_decorator
|
||||||
from registrar.config import settings
|
from registrar.config import settings
|
||||||
from registrar.models.portfolio import Portfolio
|
from registrar.models import Portfolio, SeniorOfficial
|
||||||
from django_webtest import WebTest # type: ignore
|
from django_webtest import WebTest # type: ignore
|
||||||
from registrar.models import (
|
from registrar.models import (
|
||||||
DomainRequest,
|
DomainRequest,
|
||||||
|
@ -38,6 +38,35 @@ class TestPortfolio(WebTest):
|
||||||
User.objects.all().delete()
|
User.objects.all().delete()
|
||||||
super().tearDown()
|
super().tearDown()
|
||||||
|
|
||||||
|
@less_console_noise_decorator
|
||||||
|
@override_flag("organization_feature", active=True)
|
||||||
|
def test_portfolio_senior_official(self):
|
||||||
|
"""Tests the senior official page on portfolio"""
|
||||||
|
self.app.set_user(self.user.username)
|
||||||
|
|
||||||
|
so = SeniorOfficial.objects.create(
|
||||||
|
first_name="Saturn", last_name="Enceladus", title="Planet/Moon", email="spacedivision@igorville.com"
|
||||||
|
)
|
||||||
|
|
||||||
|
self.portfolio.senior_official = so
|
||||||
|
self.portfolio.save()
|
||||||
|
self.portfolio.refresh_from_db()
|
||||||
|
|
||||||
|
self.user.portfolio = self.portfolio
|
||||||
|
self.user.portfolio_additional_permissions = [UserPortfolioPermissionChoices.VIEW_PORTFOLIO]
|
||||||
|
self.user.save()
|
||||||
|
self.user.refresh_from_db()
|
||||||
|
|
||||||
|
so_portfolio_page = self.app.get(reverse("senior-official"))
|
||||||
|
# Assert that we're on the right page
|
||||||
|
self.assertContains(so_portfolio_page, "Senior official")
|
||||||
|
self.assertContains(so_portfolio_page, "Saturn Enceladus")
|
||||||
|
self.assertContains(so_portfolio_page, "Planet/Moon")
|
||||||
|
self.assertContains(so_portfolio_page, "spacedivision@igorville.com")
|
||||||
|
|
||||||
|
self.portfolio.delete()
|
||||||
|
so.delete()
|
||||||
|
|
||||||
@less_console_noise_decorator
|
@less_console_noise_decorator
|
||||||
def test_middleware_does_not_redirect_if_no_permission(self):
|
def test_middleware_does_not_redirect_if_no_permission(self):
|
||||||
"""Test that user with no portfolio permission is not redirected when attempting to access home"""
|
"""Test that user with no portfolio permission is not redirected when attempting to access home"""
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue