Got things rolling now.

Signed-off-by: Cliff Hill <Clifford.hill@gsa.gov>
This commit is contained in:
Cliff Hill
2023-11-14 12:48:36 -05:00
parent 25b203d83d
commit 4ea62bc1dd

View File

@@ -12,7 +12,7 @@ from app.dao.invited_user_dao import (
get_invited_users_for_service, get_invited_users_for_service,
save_invited_user, save_invited_user,
) )
from app.models import InvitedUser from app.models import INVITE_EXPIRED, InvitedUser
from tests.app.db import create_invited_user from tests.app.db import create_invited_user
@@ -121,9 +121,13 @@ 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))
make_invitation(sample_user, sample_service, age=timedelta(hours=48)) make_invitation(sample_user, sample_service, age=timedelta(hours=48))
assert len(InvitedUser.query.all()) == 2 assert (
len(InvitedUser.query.filter(InvitedUser.status != INVITE_EXPIRED).all()) == 2
)
expire_invitations_created_more_than_two_days_ago() expire_invitations_created_more_than_two_days_ago()
assert len(InvitedUser.query.all()) == 0 assert (
len(InvitedUser.query.filter(InvitedUser.status != INVITE_EXPIRED).all()) == 0
)
def test_should_not_delete_invitations_less_than_two_days_old( def test_should_not_delete_invitations_less_than_two_days_old(
@@ -142,9 +146,13 @@ def test_should_not_delete_invitations_less_than_two_days_old(
email_address="expired@1.com", email_address="expired@1.com",
) )
assert len(InvitedUser.query.all()) == 2 assert (
len(InvitedUser.query.filter(InvitedUser.status != INVITE_EXPIRED).all()) == 2
)
expire_invitations_created_more_than_two_days_ago() expire_invitations_created_more_than_two_days_ago()
assert len(InvitedUser.query.all()) == 1 assert (
len(InvitedUser.query.filter(InvitedUser.status != INVITE_EXPIRED).all()) == 1
)
assert InvitedUser.query.first().email_address == "valid@2.com" assert InvitedUser.query.first().email_address == "valid@2.com"