diff --git a/atst/routes/portfolios/task_orders.py b/atst/routes/portfolios/task_orders.py index dd7856a7..6f2be63f 100644 --- a/atst/routes/portfolios/task_orders.py +++ b/atst/routes/portfolios/task_orders.py @@ -126,9 +126,12 @@ def resend_invite(portfolio_id, task_order_id, form=None): invitation = Invitations.lookup_by_portfolio_and_user(portfolio, officer) - if not invitation or (invitation.status is not InvitationStatus.PENDING): + if not invitation: raise NotFoundError("invitation") + if invitation.status is not InvitationStatus.PENDING: + raise NoAccessError("invitation") + Invitations.revoke(token=invitation.token) invite_service = InvitationService( @@ -210,7 +213,7 @@ def task_order_invitations(portfolio_id, task_order_id): form=form, ) else: - raise NotFoundError("task_order") + raise NoAccessError("task_order") @portfolios_bp.route(