mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-22 00:11:16 -05:00
131 lines
4.5 KiB
Python
131 lines
4.5 KiB
Python
from datetime import datetime, timedelta
|
|
import uuid
|
|
|
|
import pytest
|
|
from sqlalchemy.orm.exc import NoResultFound
|
|
|
|
from app import db
|
|
|
|
from app.models import InvitedUser
|
|
|
|
from app.dao.invited_user_dao import (
|
|
save_invited_user,
|
|
get_invited_user,
|
|
get_invited_users_for_service,
|
|
get_invited_user_by_id,
|
|
delete_invitations_created_more_than_two_days_ago
|
|
)
|
|
|
|
|
|
def test_create_invited_user(notify_db, notify_db_session, sample_service):
|
|
assert InvitedUser.query.count() == 0
|
|
email_address = 'invited_user@service.gov.uk'
|
|
invite_from = sample_service.users[0]
|
|
|
|
data = {
|
|
'service': sample_service,
|
|
'email_address': email_address,
|
|
'from_user': invite_from,
|
|
'permissions': 'send_messages,manage_service'
|
|
}
|
|
|
|
invited_user = InvitedUser(**data)
|
|
save_invited_user(invited_user)
|
|
|
|
assert InvitedUser.query.count() == 1
|
|
assert invited_user.email_address == email_address
|
|
assert invited_user.from_user == invite_from
|
|
permissions = invited_user.get_permissions()
|
|
assert len(permissions) == 2
|
|
assert 'send_messages' in permissions
|
|
assert 'manage_service' in permissions
|
|
|
|
|
|
def test_get_invited_user_by_service_and_id(notify_db, notify_db_session, sample_invited_user):
|
|
from_db = get_invited_user(sample_invited_user.service.id, sample_invited_user.id)
|
|
assert from_db == sample_invited_user
|
|
|
|
|
|
def test_get_invited_user_by_id(notify_db, notify_db_session, sample_invited_user):
|
|
from_db = get_invited_user_by_id(sample_invited_user.id)
|
|
assert from_db == sample_invited_user
|
|
|
|
|
|
def test_get_unknown_invited_user_returns_none(notify_db, notify_db_session, sample_service):
|
|
unknown_id = uuid.uuid4()
|
|
|
|
with pytest.raises(NoResultFound) as e:
|
|
get_invited_user(sample_service.id, unknown_id)
|
|
assert 'No row was found for one()' in str(e.value)
|
|
|
|
|
|
def test_get_invited_users_for_service(notify_db, notify_db_session, sample_service):
|
|
from tests.app.conftest import sample_invited_user
|
|
invites = []
|
|
for i in range(0, 5):
|
|
email = 'invited_user_{}@service.gov.uk'.format(i)
|
|
|
|
invited_user = sample_invited_user(notify_db,
|
|
notify_db_session,
|
|
sample_service,
|
|
email)
|
|
invites.append(invited_user)
|
|
|
|
all_from_db = get_invited_users_for_service(sample_service.id)
|
|
assert len(all_from_db) == 5
|
|
for invite in invites:
|
|
assert invite in all_from_db
|
|
|
|
|
|
def test_get_invited_users_for_service_that_has_no_invites(notify_db, notify_db_session, sample_service):
|
|
invites = get_invited_users_for_service(sample_service.id)
|
|
assert len(invites) == 0
|
|
|
|
|
|
def test_save_invited_user_sets_status_to_cancelled(notify_db, notify_db_session, sample_invited_user):
|
|
assert InvitedUser.query.count() == 1
|
|
saved = InvitedUser.query.get(sample_invited_user.id)
|
|
assert saved.status == 'pending'
|
|
saved.status = 'cancelled'
|
|
save_invited_user(saved)
|
|
assert InvitedUser.query.count() == 1
|
|
cancelled_invited_user = InvitedUser.query.get(sample_invited_user.id)
|
|
assert cancelled_invited_user.status == 'cancelled'
|
|
|
|
|
|
def test_should_delete_all_invitations_more_than_one_day_old(
|
|
sample_user,
|
|
sample_service):
|
|
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
|
|
delete_invitations_created_more_than_two_days_ago()
|
|
assert len(InvitedUser.query.all()) == 0
|
|
|
|
|
|
def test_should_not_delete_invitations_less_than_two_days_old(
|
|
sample_user,
|
|
sample_service):
|
|
make_invitation(sample_user, sample_service, age=timedelta(hours=47, minutes=59, seconds=59),
|
|
email_address="valid@2.com")
|
|
make_invitation(sample_user, sample_service, age=timedelta(hours=48),
|
|
email_address="expired@1.com")
|
|
|
|
assert len(InvitedUser.query.all()) == 2
|
|
delete_invitations_created_more_than_two_days_ago()
|
|
assert len(InvitedUser.query.all()) == 1
|
|
assert InvitedUser.query.first().email_address == "valid@2.com"
|
|
|
|
|
|
def make_invitation(user, service, age=timedelta(hours=0), email_address="test@test.com"):
|
|
verify_code = InvitedUser(
|
|
email_address=email_address,
|
|
from_user=user,
|
|
service=service,
|
|
status='pending',
|
|
created_at=datetime.utcnow() - age,
|
|
permissions='manage_settings'
|
|
)
|
|
db.session.add(verify_code)
|
|
db.session.commit()
|