mirror of
https://github.com/simple-login/app.git
synced 2024-11-16 08:58:30 +01:00
dd0598a4dd
* Send welcome email when user created by login with proton * Add dedicated test to user.created_by_partner
31 lines
1 KiB
Python
31 lines
1 KiB
Python
from app import config
|
|
from app.db import Session
|
|
from app.models import User, Job
|
|
from tests.utils import create_new_user, random_email
|
|
|
|
|
|
def test_available_sl_domains(flask_client):
|
|
user = create_new_user()
|
|
|
|
assert set(user.available_sl_domains()) == {"d1.test", "d2.test", "sl.local"}
|
|
|
|
|
|
def test_create_from_partner(flask_client):
|
|
user = User.create(email=random_email(), from_partner=True)
|
|
assert User.FLAG_CREATED_FROM_PARTNER == (
|
|
user.flags & User.FLAG_CREATED_FROM_PARTNER
|
|
)
|
|
assert user.notification is False
|
|
assert user.trial_end is None
|
|
job = Session.query(Job).order_by(Job.id.desc()).first()
|
|
assert job is not None
|
|
assert job.name == config.JOB_SEND_PROTON_WELCOME_1
|
|
assert job.payload.get("user_id") == user.id
|
|
|
|
|
|
def test_user_created_by_partner(flask_client):
|
|
user_from_partner = User.create(email=random_email(), from_partner=True)
|
|
assert user_from_partner.created_by_partner is True
|
|
|
|
regular_user = User.create(email=random_email())
|
|
assert regular_user.created_by_partner is False
|