From a9fac1ca5945c0257a6aab1077322f73a12608f9 Mon Sep 17 00:00:00 2001 From: George Drummond Date: Tue, 21 May 2019 11:09:57 -0400 Subject: [PATCH] Add in missing test --- tests/domain/test_portfolios.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/tests/domain/test_portfolios.py b/tests/domain/test_portfolios.py index 6819b36b..bf6a415e 100644 --- a/tests/domain/test_portfolios.py +++ b/tests/domain/test_portfolios.py @@ -7,9 +7,12 @@ from atst.domain.portfolio_roles import PortfolioRoles from atst.domain.applications import Applications from atst.domain.environments import Environments from atst.domain.permission_sets import PermissionSets, PORTFOLIO_PERMISSION_SETS +from atst.models.application_role import Status as ApplicationRoleStatus from atst.models.portfolio_role import Status as PortfolioRoleStatus from tests.factories import ( + ApplicationFactory, + ApplicationRoleFactory, UserFactory, PortfolioRoleFactory, PortfolioFactory, @@ -164,6 +167,17 @@ def test_scoped_portfolio_returns_all_applications_for_portfolio_owner( assert len(scoped_portfolio.applications[0].environments) == 3 +def test_for_user_returns_portfolios_for_applications_user_invited_to(): + bob = UserFactory.create() + portfolio = PortfolioFactory.create() + application = ApplicationFactory.create(portfolio=portfolio) + ApplicationRoleFactory.create( + application=application, user=bob, status=ApplicationRoleStatus.ACTIVE + ) + + assert portfolio in Portfolios.for_user(user=bob) + + def test_for_user_returns_active_portfolios_for_user(portfolio, portfolio_owner): bob = UserFactory.create() PortfolioRoleFactory.create(