fix unit test

This commit is contained in:
zandercymatics 2024-09-05 08:53:37 -06:00
parent 47e8934cde
commit 9d4ab22dc4
No known key found for this signature in database
GPG key ID: FF4636ABEC9682B7
2 changed files with 8 additions and 11 deletions

View file

@ -3,6 +3,7 @@ from django.utils import timezone
from django.test import TestCase, RequestFactory, Client from django.test import TestCase, RequestFactory, Client
from django.contrib.admin.sites import AdminSite from django.contrib.admin.sites import AdminSite
from registrar.models.utility.portfolio_helper import UserPortfolioRoleChoices from registrar.models.utility.portfolio_helper import UserPortfolioRoleChoices
from django_webtest import WebTest # type: ignore
from api.tests.common import less_console_noise_decorator from api.tests.common import less_console_noise_decorator
from django.urls import reverse from django.urls import reverse
from registrar.admin import ( from registrar.admin import (
@ -972,7 +973,7 @@ class TestListHeaderAdmin(TestCase):
) )
class TestMyUserAdmin(MockDbForSharedTests): class TestMyUserAdmin(MockDbForSharedTests, WebTest):
"""Tests for the MyUserAdmin class as super or staff user """Tests for the MyUserAdmin class as super or staff user
Notes: Notes:
@ -992,6 +993,7 @@ class TestMyUserAdmin(MockDbForSharedTests):
def setUp(self): def setUp(self):
super().setUp() super().setUp()
self.app.set_user(self.superuser.username)
self.client = Client(HTTP_HOST="localhost:8080") self.client = Client(HTTP_HOST="localhost:8080")
def tearDown(self): def tearDown(self):
@ -1229,16 +1231,11 @@ class TestMyUserAdmin(MockDbForSharedTests):
@less_console_noise_decorator @less_console_noise_decorator
def test_user_can_see_related_portfolios(self): def test_user_can_see_related_portfolios(self):
"""Tests if a user can see the portfolios they are associated with on the user page""" """Tests if a user can see the portfolios they are associated with on the user page"""
portfolio, _ = Portfolio.objects.get_or_create(organization_name="test", creator=self.user) portfolio, _ = Portfolio.objects.get_or_create(organization_name="test", creator=self.superuser)
permission, _ = UserPortfolioPermission.objects.get_or_create( permission, _ = UserPortfolioPermission.objects.get_or_create(
user=self.user, portfolio=portfolio, roles=[UserPortfolioRoleChoices.ORGANIZATION_ADMIN] user=self.superuser, portfolio=portfolio, roles=[UserPortfolioRoleChoices.ORGANIZATION_ADMIN]
)
self.user.refresh_from_db()
self.client.force_login(self.user)
response = self.client.get(
"/admin/registrar/user/{}/change/".format(self.user.id),
follow=True,
) )
response = self.app.get(reverse("admin:registrar_user_change", args=[self.superuser.pk]))
expected_href = reverse("admin:registrar_portfolio_change", args=[portfolio.pk]) expected_href = reverse("admin:registrar_portfolio_change", args=[portfolio.pk])
self.assertContains(response, expected_href) self.assertContains(response, expected_href)
self.assertContains(response, str(portfolio)) self.assertContains(response, str(portfolio))