From 4ea62bc1dd9cc1ae084406daf91ef462fbda93f4 Mon Sep 17 00:00:00 2001 From: Cliff Hill Date: Tue, 14 Nov 2023 12:48:36 -0500 Subject: [PATCH] Got things rolling now. Signed-off-by: Cliff Hill --- tests/app/dao/test_invited_user_dao.py | 18 +++++++++++++----- 1 file changed, 13 insertions(+), 5 deletions(-) diff --git a/tests/app/dao/test_invited_user_dao.py b/tests/app/dao/test_invited_user_dao.py index 4494d7d62..4b49c1948 100644 --- a/tests/app/dao/test_invited_user_dao.py +++ b/tests/app/dao/test_invited_user_dao.py @@ -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"