From 8234037b1121628b6cb1f832a40e76f06f83a616 Mon Sep 17 00:00:00 2001 From: Montana Date: Wed, 20 Mar 2019 15:05:50 -0400 Subject: [PATCH] Add tests for has_permission_sets --- tests/models/test_portfolio_role.py | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) diff --git a/tests/models/test_portfolio_role.py b/tests/models/test_portfolio_role.py index 198408c8..333375ff 100644 --- a/tests/models/test_portfolio_role.py +++ b/tests/models/test_portfolio_role.py @@ -307,3 +307,23 @@ def test_can_list_all_permissions(): port_role = PortfolioRoleFactory.create(permission_sets=[role_one, role_two]) expected_perms = role_one.permissions + role_two.permissions assert expected_perms == expected_perms + + +def test_has_permission_set(): + perm_sets = PermissionSets.get_many( + [PermissionSets.VIEW_PORTFOLIO_FUNDING, PermissionSets.VIEW_PORTFOLIO_REPORTS] + ) + port_role = PortfolioRoleFactory.create(permission_sets=perm_sets) + + assert port_role.has_permission_set(PermissionSets.VIEW_PORTFOLIO_REPORTS) + + +def test_does_not_have_permission_set(): + perm_sets = PermissionSets.get_many( + [PermissionSets.VIEW_PORTFOLIO_FUNDING, PermissionSets.VIEW_PORTFOLIO_REPORTS] + ) + port_role = PortfolioRoleFactory.create(permission_sets=perm_sets) + + assert not port_role.has_permission_set( + PermissionSets.EDIT_PORTFOLIO_APPLICATION_MANAGEMENT + )