Check that user has portfolio and invite perms to revoke or resend invites

This commit is contained in:
leigh-mil
2019-03-25 15:54:40 -04:00
parent d152034e1b
commit 5d2b8556ed
2 changed files with 57 additions and 0 deletions

View File

@@ -169,6 +169,58 @@ def test_revoke_invitation(client, user_session):
assert invite.is_revoked
def test_user_can_only_revoke_invites_in_their_portfolio(client, user_session):
portfolio = PortfolioFactory.create()
other_portfolio = PortfolioFactory.create()
user = UserFactory.create()
ws_role = PortfolioRoleFactory.create(
user=user, portfolio=other_portfolio, status=PortfolioRoleStatus.PENDING
)
invite = InvitationFactory.create(
user_id=user.id,
portfolio_role=ws_role,
status=InvitationStatus.REJECTED_EXPIRED,
expiration_time=datetime.datetime.now() - datetime.timedelta(seconds=1),
)
user_session(portfolio.owner)
response = client.post(
url_for(
"portfolios.revoke_invitation",
portfolio_id=portfolio.id,
token=invite.token,
)
)
assert response.status_code == 404
assert not invite.is_revoked
def test_user_can_only_resend_invites_in_their_portfolio(client, user_session, queue):
portfolio = PortfolioFactory.create()
other_portfolio = PortfolioFactory.create()
user = UserFactory.create()
ws_role = PortfolioRoleFactory.create(
user=user, portfolio=other_portfolio, status=PortfolioRoleStatus.PENDING
)
invite = InvitationFactory.create(
user_id=user.id,
portfolio_role=ws_role,
status=InvitationStatus.REJECTED_EXPIRED,
expiration_time=datetime.datetime.now() - datetime.timedelta(seconds=1),
)
user_session(portfolio.owner)
response = client.post(
url_for(
"portfolios.resend_invitation",
portfolio_id=portfolio.id,
token=invite.token,
)
)
assert response.status_code == 404
assert len(queue.get_queue()) == 0
def test_resend_invitation_sends_email(client, user_session, queue):
user = UserFactory.create()
portfolio = PortfolioFactory.create()