mirror of
https://github.com/simple-login/app.git
synced 2024-11-18 01:40:38 +01:00
58 lines
1.8 KiB
Python
58 lines
1.8 KiB
Python
import arrow
|
|
|
|
from app.config import EMAIL_DOMAIN, MAX_NB_EMAIL_FREE_PLAN
|
|
from app.extensions import db
|
|
from app.models import generate_email, User, GenEmail
|
|
|
|
|
|
def test_generate_email(flask_client):
|
|
email = generate_email()
|
|
assert email.endswith("@" + EMAIL_DOMAIN)
|
|
|
|
|
|
def test_profile_picture_url(flask_client):
|
|
user = User.create(
|
|
email="a@b.c", password="password", name="Test User", activated=True
|
|
)
|
|
db.session.commit()
|
|
|
|
assert user.profile_picture_url() == "http://sl.test/static/default-avatar.png"
|
|
|
|
|
|
def test_suggested_emails_for_user_who_can_create_new_email(flask_client):
|
|
user = User.create(
|
|
email="a@b.c", password="password", name="Test User", activated=True
|
|
)
|
|
db.session.commit()
|
|
|
|
suggested_email, other_emails = user.suggested_emails()
|
|
assert suggested_email
|
|
assert len(other_emails) == 1
|
|
|
|
# the suggested email is new and not exist in GenEmail
|
|
assert GenEmail.get_by(email=suggested_email) is None
|
|
|
|
# all other emails are generated emails
|
|
assert GenEmail.get_by(email=other_emails[0])
|
|
|
|
|
|
def test_suggested_emails_for_user_who_cannot_create_new_email(flask_client):
|
|
user = User.create(
|
|
email="a@b.c", password="password", name="Test User", activated=True
|
|
)
|
|
user.trial_expiration = arrow.now().shift(days=-1)
|
|
db.session.commit()
|
|
|
|
# make sure user runs out of quota to create new email
|
|
for i in range(MAX_NB_EMAIL_FREE_PLAN):
|
|
GenEmail.create_new_gen_email(user_id=user.id)
|
|
db.session.commit()
|
|
|
|
suggested_email, other_emails = user.suggested_emails()
|
|
|
|
# the suggested email is chosen from existing GenEmail
|
|
assert GenEmail.get_by(email=suggested_email)
|
|
|
|
# all other emails are generated emails
|
|
for email in other_emails:
|
|
assert GenEmail.get_by(email=email)
|