66 lines
2.1 KiB
Python
66 lines
2.1 KiB
Python
from atst.domain.portfolio_roles import PortfolioRoles
|
|
from atst.domain.users import Users
|
|
from atst.models.portfolio_role import Status as PortfolioRoleStatus
|
|
from atst.domain.roles import Roles
|
|
|
|
from tests.factories import (
|
|
PortfolioFactory,
|
|
UserFactory,
|
|
InvitationFactory,
|
|
PortfolioRoleFactory,
|
|
)
|
|
|
|
|
|
def test_can_create_new_portfolio_role():
|
|
portfolio = PortfolioFactory.create()
|
|
new_user = UserFactory.create()
|
|
|
|
portfolio_role_dicts = [{"id": new_user.id, "portfolio_role": "owner"}]
|
|
portfolio_roles = PortfolioRoles.add_many(portfolio.id, portfolio_role_dicts)
|
|
|
|
assert portfolio_roles[0].user_id == new_user.id
|
|
assert portfolio_roles[0].user.atat_role.name == new_user.atat_role.name
|
|
assert portfolio_roles[0].role.name == new_user.portfolio_roles[0].role.name
|
|
|
|
|
|
def test_can_update_existing_portfolio_role():
|
|
portfolio = PortfolioFactory.create()
|
|
new_user = UserFactory.create()
|
|
|
|
PortfolioRoles.add_many(
|
|
portfolio.id, [{"id": new_user.id, "portfolio_role": "owner"}]
|
|
)
|
|
portfolio_roles = PortfolioRoles.add_many(
|
|
portfolio.id, [{"id": new_user.id, "portfolio_role": "developer"}]
|
|
)
|
|
|
|
assert portfolio_roles[0].user.atat_role.name == new_user.atat_role.name
|
|
assert portfolio_roles[0].role.name == new_user.portfolio_roles[0].role.name
|
|
|
|
|
|
def test_portfolio_role_permissions():
|
|
portfolio_one = PortfolioFactory.create()
|
|
portfolio_two = PortfolioFactory.create()
|
|
new_user = UserFactory.create()
|
|
PortfolioRoleFactory.create(
|
|
portfolio=portfolio_one,
|
|
user=new_user,
|
|
role=Roles.get("developer"),
|
|
status=PortfolioRoleStatus.ACTIVE,
|
|
)
|
|
PortfolioRoleFactory.create(
|
|
portfolio=portfolio_two,
|
|
user=new_user,
|
|
role=Roles.get("developer"),
|
|
status=PortfolioRoleStatus.PENDING,
|
|
)
|
|
|
|
default_perms = set(new_user.atat_role.permissions)
|
|
assert len(
|
|
PortfolioRoles.portfolio_role_permissions(portfolio_one, new_user)
|
|
) > len(default_perms)
|
|
assert (
|
|
PortfolioRoles.portfolio_role_permissions(portfolio_two, new_user)
|
|
== default_perms
|
|
)
|