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,
save_invited_user,
)
from app.models import InvitedUser
from app.models import INVITE_EXPIRED, InvitedUser
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))
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()
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(
@@ -142,9 +146,13 @@ def test_should_not_delete_invitations_less_than_two_days_old(
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()
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"