mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-20 15:31:15 -05:00
Adjusting some DAO stuff to send expired invites now.
Signed-off-by: Cliff Hill <Clifford.hill@gsa.gov>
This commit is contained in:
@@ -13,26 +13,19 @@ def get_invited_user_by_service_and_id(service_id, invited_user_id):
|
|||||||
return InvitedUser.query.filter(
|
return InvitedUser.query.filter(
|
||||||
InvitedUser.service_id == service_id,
|
InvitedUser.service_id == service_id,
|
||||||
InvitedUser.id == invited_user_id,
|
InvitedUser.id == invited_user_id,
|
||||||
InvitedUser.status != INVITE_EXPIRED,
|
|
||||||
).one()
|
).one()
|
||||||
|
|
||||||
|
|
||||||
def get_invited_user_by_id(invited_user_id):
|
def get_invited_user_by_id(invited_user_id):
|
||||||
return InvitedUser.query.filter(
|
return InvitedUser.query.filter(InvitedUser.id == invited_user_id).one()
|
||||||
InvitedUser.id == invited_user_id, InvitedUser.status != INVITE_EXPIRED
|
|
||||||
).one()
|
|
||||||
|
|
||||||
|
|
||||||
def get_expired_invited_users_for_service(service_id):
|
def get_expired_invited_users_for_service(service_id):
|
||||||
return InvitedUser.query.filter(
|
return InvitedUser.query.filter(InvitedUser.service_id == service_id).all()
|
||||||
InvitedUser.service_id == service_id, InvitedUser.status == INVITE_EXPIRED
|
|
||||||
).all()
|
|
||||||
|
|
||||||
|
|
||||||
def get_invited_users_for_service(service_id):
|
def get_invited_users_for_service(service_id):
|
||||||
return InvitedUser.query.filter(
|
return InvitedUser.query.filter(InvitedUser.service_id == service_id).all()
|
||||||
InvitedUser.service_id == service_id, InvitedUser.status != INVITE_EXPIRED
|
|
||||||
).all()
|
|
||||||
|
|
||||||
|
|
||||||
def expire_invitations_created_more_than_two_days_ago():
|
def expire_invitations_created_more_than_two_days_ago():
|
||||||
|
|||||||
@@ -155,7 +155,18 @@ def test_should_not_delete_invitations_less_than_two_days_old(
|
|||||||
assert (
|
assert (
|
||||||
len(InvitedUser.query.filter(InvitedUser.status != INVITE_EXPIRED).all()) == 1
|
len(InvitedUser.query.filter(InvitedUser.status != INVITE_EXPIRED).all()) == 1
|
||||||
)
|
)
|
||||||
assert InvitedUser.query.first().email_address == "valid@2.com"
|
assert (
|
||||||
|
InvitedUser.query.filter(InvitedUser.status != INVITE_EXPIRED)
|
||||||
|
.first()
|
||||||
|
.email_address
|
||||||
|
== "valid@2.com"
|
||||||
|
)
|
||||||
|
assert (
|
||||||
|
InvitedUser.query.filter(InvitedUser.status == INVITE_EXPIRED)
|
||||||
|
.first()
|
||||||
|
.email_address
|
||||||
|
== "expired@1.com"
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
def make_invitation(user, service, age=None, email_address="test@test.com"):
|
def make_invitation(user, service, age=None, email_address="test@test.com"):
|
||||||
|
|||||||
Reference in New Issue
Block a user