52 lines
1.9 KiB
Python
52 lines
1.9 KiB
Python
from flask import url_for
|
|
|
|
from tests.factories import UserFactory, PortfolioFactory
|
|
from atst.domain.permission_sets import PermissionSets
|
|
from atst.queue import queue
|
|
|
|
_DEFAULT_PERMS_FORM_DATA = {
|
|
"permission_sets-perms_app_mgmt": PermissionSets.VIEW_PORTFOLIO_APPLICATION_MANAGEMENT,
|
|
"permission_sets-perms_funding": PermissionSets.VIEW_PORTFOLIO_FUNDING,
|
|
"permission_sets-perms_reporting": PermissionSets.VIEW_PORTFOLIO_REPORTS,
|
|
"permission_sets-perms_portfolio_mgmt": PermissionSets.VIEW_PORTFOLIO_ADMIN,
|
|
}
|
|
|
|
|
|
def test_user_with_permission_has_add_member_link(client, user_session):
|
|
portfolio = PortfolioFactory.create()
|
|
user_session(portfolio.owner)
|
|
response = client.get(url_for("portfolios.admin", portfolio_id=portfolio.id))
|
|
assert response.status_code == 200
|
|
assert (
|
|
url_for("portfolios.create_member", portfolio_id=portfolio.id).encode()
|
|
in response.data
|
|
)
|
|
|
|
|
|
def test_create_member(client, user_session):
|
|
user = UserFactory.create()
|
|
portfolio = PortfolioFactory.create()
|
|
user_session(portfolio.owner)
|
|
queue_length = len(queue.get_queue())
|
|
|
|
response = client.post(
|
|
url_for("portfolios.create_member", portfolio_id=portfolio.id),
|
|
data={
|
|
"user_data-dod_id": user.dod_id,
|
|
"user_data-first_name": "user_data-Wilbur",
|
|
"user_data-last_name": "user_data-Zuckerman",
|
|
"user_data-email": "user_data-some_pig@zuckermans.com",
|
|
"user_data-portfolio_role": "user_data-developer",
|
|
**_DEFAULT_PERMS_FORM_DATA,
|
|
},
|
|
follow_redirects=True,
|
|
)
|
|
|
|
assert response.status_code == 200
|
|
assert user.full_name in response.data.decode()
|
|
assert user.has_portfolios
|
|
assert user.portfolio_invitations
|
|
assert len(queue.get_queue()) == queue_length + 1
|
|
portfolio_role = user.portfolio_roles[0]
|
|
assert len(portfolio_role.permission_sets) == 5
|