record expiration time on the invitation

This commit is contained in:
dandds
2018-10-26 13:43:40 -04:00
parent d5998ed370
commit 5c5f9c6c9c
5 changed files with 44 additions and 8 deletions

View File

@@ -30,11 +30,11 @@ def test_accept_expired_invitation():
workspace = WorkspaceFactory.create()
user = UserFactory.create()
increment = Invitations.EXPIRATION_LIMIT_MINUTES + 1
created_at = datetime.datetime.now() - datetime.timedelta(minutes=increment)
expiration_time = datetime.datetime.now() - datetime.timedelta(minutes=increment)
invite = InvitationFactory.create(
workspace_id=workspace.id,
user_id=user.id,
time_created=created_at,
expiration_time=expiration_time,
status=Status.PENDING,
)
with pytest.raises(InvitationError):

View File

@@ -22,6 +22,7 @@ from atst.models.workspace_role import WorkspaceRole
from atst.models.environment_role import EnvironmentRole
from atst.models.invitation import Invitation, Status as InvitationStatus
from atst.domain.workspaces import Workspaces
from atst.domain.invitations import Invitations
class Base(factory.alchemy.SQLAlchemyModelFactory):
@@ -341,3 +342,4 @@ class InvitationFactory(Base):
model = Invitation
status = InvitationStatus.PENDING
expiration_time = Invitations.current_expiration_time()