mirror of
https://github.com/GSA/notifications-api.git
synced 2026-09-05 21:38:25 -04:00
@@ -12,7 +12,8 @@ from app.dao.invited_user_dao import (
|
||||
get_invited_users_for_service,
|
||||
save_invited_user,
|
||||
)
|
||||
from app.models import INVITE_EXPIRED, InvitedUser
|
||||
from app.enums import InvitedUserStatus
|
||||
from app.models import InvitedUser
|
||||
from tests.app.db import create_invited_user
|
||||
|
||||
|
||||
@@ -122,11 +123,11 @@ def test_should_delete_all_invitations_more_than_one_day_old(
|
||||
make_invitation(sample_user, sample_service, age=timedelta(hours=48))
|
||||
make_invitation(sample_user, sample_service, age=timedelta(hours=48))
|
||||
assert (
|
||||
len(InvitedUser.query.filter(InvitedUser.status != INVITE_EXPIRED).all()) == 2
|
||||
len(InvitedUser.query.filter(InvitedUser.status != InvitedUserStatus.EXPIRED).all()) == 2
|
||||
)
|
||||
expire_invitations_created_more_than_two_days_ago()
|
||||
assert (
|
||||
len(InvitedUser.query.filter(InvitedUser.status != INVITE_EXPIRED).all()) == 0
|
||||
len(InvitedUser.query.filter(InvitedUser.status != InvitedUserStatus.EXPIRED).all()) == 0
|
||||
)
|
||||
|
||||
|
||||
@@ -149,20 +150,20 @@ def test_should_not_delete_invitations_less_than_two_days_old(
|
||||
)
|
||||
|
||||
assert (
|
||||
len(InvitedUser.query.filter(InvitedUser.status != INVITE_EXPIRED).all()) == 2
|
||||
len(InvitedUser.query.filter(InvitedUser.status != InvitedUserStatus.EXPIRED).all()) == 2
|
||||
)
|
||||
expire_invitations_created_more_than_two_days_ago()
|
||||
assert (
|
||||
len(InvitedUser.query.filter(InvitedUser.status != INVITE_EXPIRED).all()) == 1
|
||||
len(InvitedUser.query.filter(InvitedUser.status != InvitedUserStatus.EXPIRED).all()) == 1
|
||||
)
|
||||
assert (
|
||||
InvitedUser.query.filter(InvitedUser.status != INVITE_EXPIRED)
|
||||
InvitedUser.query.filter(InvitedUser.status != InvitedUserStatus.EXPIRED)
|
||||
.first()
|
||||
.email_address
|
||||
== "valid@2.com"
|
||||
)
|
||||
assert (
|
||||
InvitedUser.query.filter(InvitedUser.status == INVITE_EXPIRED)
|
||||
InvitedUser.query.filter(InvitedUser.status == InvitedUserStatus.EXPIRED)
|
||||
.first()
|
||||
.email_address
|
||||
== "expired@1.com"
|
||||
|
||||
Reference in New Issue
Block a user