mirror of
https://github.com/simple-login/app.git
synced 2025-10-06 05:17:41 +08:00
a70baad478
* wip: start implementing alias trash * Added alias trash dashboard page * test: delete_alias changes * Format html * fix: mailbox deletion * feat: add delete_alias_action setting in dashboard settings * chore: disable alias when trashing it * Add restore tests * Move tras/restore to alias_actions * rename alias_actions to alias_delete * Remove alias_actions * Send events and alias audit log on alias restore * feat: adapt queries to trashed alias * chore: add metrics on alias trash actions * fix: missing empty arg * Add rate limit for restore and restore all * fix: mailbox alias count * feat: properly handle alias deletion for custom domain deletion * chore: add error logs * chore: update alias trash copy + change Trash location * feat: make can_create_new_alias not take trashed aliases into account * chore: update mailbox deletion dialog copy --------- Co-authored-by: Adrià Casajús <adria.casajus@proton.ch>
71 lines
2.3 KiB
Python
71 lines
2.3 KiB
Python
import arrow
|
|
|
|
from app.alias_delete import move_alias_to_trash
|
|
from app.constants import JobType
|
|
from app.db import Session
|
|
from app.models import (
|
|
User,
|
|
Job,
|
|
PartnerSubscription,
|
|
PartnerUser,
|
|
ManualSubscription,
|
|
UserAliasDeleteAction,
|
|
Alias,
|
|
)
|
|
from app.proton.proton_partner import get_proton_partner
|
|
from tests.utils import random_email, random_token, create_new_user
|
|
|
|
|
|
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
|
|
assert user.newsletter_alias_id is None
|
|
job = Session.query(Job).order_by(Job.id.desc()).first()
|
|
assert job is not None
|
|
assert job.name == JobType.SEND_PROTON_WELCOME_1.value
|
|
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
|
|
|
|
|
|
def test_user_is_premium(flask_client):
|
|
user = User.create(email=random_email(), from_partner=True)
|
|
assert not user.is_premium()
|
|
partner_user = PartnerUser.create(
|
|
user_id=user.id,
|
|
partner_id=get_proton_partner().id,
|
|
partner_email=user.email,
|
|
external_user_id=random_token(),
|
|
flush=True,
|
|
)
|
|
ps = PartnerSubscription.create(
|
|
partner_user_id=partner_user.id, end_at=arrow.now().shift(years=1), flush=True
|
|
)
|
|
assert user.is_premium()
|
|
assert not user.is_premium(include_partner_subscription=False)
|
|
ManualSubscription.create(user_id=user.id, end_at=ps.end_at)
|
|
assert user.is_premium()
|
|
assert user.is_premium(include_partner_subscription=False)
|
|
|
|
|
|
def test_user_can_create_alias_does_not_count_trashed_ones():
|
|
user = create_new_user(alias_delete_action=UserAliasDeleteAction.MoveToTrash)
|
|
|
|
aliases = []
|
|
while user.can_create_new_alias():
|
|
aliases.append(Alias.create_new_random(user))
|
|
|
|
assert user.can_create_new_alias() is False
|
|
|
|
move_alias_to_trash(aliases[0], user, commit=True)
|
|
assert user.can_create_new_alias() is True
|