more fine-grained errors for invalid invitations

This commit is contained in:
dandds
2018-10-31 12:06:42 -04:00
parent cdef2d8803
commit 87baa1f873
4 changed files with 90 additions and 36 deletions

View File

@@ -2,7 +2,12 @@ import datetime
import pytest
import re
from atst.domain.invitations import Invitations, InvitationError, WrongUserError
from atst.domain.invitations import (
Invitations,
InvitationError,
WrongUserError,
ExpiredError,
)
from atst.models.invitation import Status
from tests.factories import (
@@ -42,7 +47,7 @@ def test_accept_expired_invitation():
invite = InvitationFactory.create(
user_id=user.id, expiration_time=expiration_time, status=Status.PENDING
)
with pytest.raises(InvitationError):
with pytest.raises(ExpiredError):
Invitations.accept(user, invite.token)
assert invite.is_rejected
@@ -69,3 +74,12 @@ def test_wrong_user_accepts_invitation():
with pytest.raises(WrongUserError):
Invitations.accept(wrong_user, invite.token)
def test_accept_invitation_twice():
workspace = WorkspaceFactory.create()
user = UserFactory.create()
ws_role = WorkspaceRoleFactory.create(user=user, workspace=workspace)
invite = Invitations.create(ws_role, workspace.owner, user)
Invitations.accept(user, invite.token)
with pytest.raises(InvitationError):
Invitations.accept(user, invite.token)