diff --git a/src/registrar/tests/common.py b/src/registrar/tests/common.py index f3fd6709f..ceb3b6e92 100644 --- a/src/registrar/tests/common.py +++ b/src/registrar/tests/common.py @@ -906,7 +906,7 @@ def completed_domain_request( # noqa federal_agency=None, federal_type=None, action_needed_reason=None, - last_selected_portfolio=None, + portfolio=None, ): """A completed domain request.""" if not user: @@ -977,8 +977,8 @@ def completed_domain_request( # noqa if action_needed_reason: domain_request_kwargs["action_needed_reason"] = action_needed_reason - if last_selected_portfolio: - domain_request_kwargs["last_selected_portfolio"] = last_selected_portfolio + if portfolio: + domain_request_kwargs["portfolio"] = portfolio domain_request, _ = DomainRequest.objects.get_or_create(**domain_request_kwargs) diff --git a/src/registrar/tests/test_admin.py b/src/registrar/tests/test_admin.py index 49d2b4802..144605b87 100644 --- a/src/registrar/tests/test_admin.py +++ b/src/registrar/tests/test_admin.py @@ -41,6 +41,7 @@ from registrar.models import ( TransitionDomain, Portfolio, Suborganization, + UserPortfolioPermission, ) from registrar.models.portfolio_invitation import PortfolioInvitation from registrar.models.senior_official import SeniorOfficial