record expiration time on the invitation
This commit is contained in:
@@ -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):
|
||||
|
@@ -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()
|
||||
|
Reference in New Issue
Block a user