mirror of
https://github.com/simple-login/app.git
synced 2024-11-15 13:14:36 +08:00
63 lines
1.8 KiB
Python
63 lines
1.8 KiB
Python
from typing import Tuple
|
|
|
|
from app.alias_mailbox_utils import (
|
|
set_mailboxes_for_alias,
|
|
CannotSetMailboxesForAliasCause,
|
|
)
|
|
from app.models import Alias, Mailbox, User, AliasMailbox
|
|
from tests.utils import create_new_user, random_email
|
|
|
|
|
|
def setup() -> Tuple[User, Alias]:
|
|
user = create_new_user()
|
|
alias = Alias.create(
|
|
user_id=user.id,
|
|
email=random_email(),
|
|
mailbox_id=user.default_mailbox_id,
|
|
commit=True,
|
|
)
|
|
return user, alias
|
|
|
|
|
|
def test_set_mailboxes_for_alias_empty_list():
|
|
user, alias = setup()
|
|
err = set_mailboxes_for_alias(user.id, alias, [])
|
|
assert err is CannotSetMailboxesForAliasCause.EmptyMailboxes
|
|
|
|
|
|
def test_set_mailboxes_for_alias_mailbox_for_other_user():
|
|
user, alias = setup()
|
|
another_user = create_new_user()
|
|
err = set_mailboxes_for_alias(user.id, alias, [another_user.default_mailbox_id])
|
|
assert err is CannotSetMailboxesForAliasCause.Forbidden
|
|
|
|
|
|
def test_set_mailboxes_for_alias_mailbox_not_exists():
|
|
user, alias = setup()
|
|
err = set_mailboxes_for_alias(user.id, alias, [9999999])
|
|
assert err is CannotSetMailboxesForAliasCause.Forbidden
|
|
|
|
|
|
def test_set_mailboxes_for_alias_mailbox_success():
|
|
user, alias = setup()
|
|
mb1 = Mailbox.create(
|
|
user_id=user.id,
|
|
email=random_email(),
|
|
verified=True,
|
|
)
|
|
mb2 = Mailbox.create(
|
|
user_id=user.id,
|
|
email=random_email(),
|
|
verified=True,
|
|
commit=True,
|
|
)
|
|
err = set_mailboxes_for_alias(user.id, alias, [mb1.id, mb2.id])
|
|
assert err is None
|
|
|
|
db_alias = Alias.get_by(id=alias.id)
|
|
assert db_alias is not None
|
|
assert db_alias.mailbox_id == mb1.id
|
|
|
|
alias_mailboxes = AliasMailbox.filter_by(alias_id=alias.id).all()
|
|
assert len(alias_mailboxes) == 1
|
|
assert alias_mailboxes[0].mailbox_id == mb2.id
|