mirror of
https://github.com/simple-login/app.git
synced 2025-09-12 01:24:20 +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>
94 lines
2.8 KiB
Python
94 lines
2.8 KiB
Python
import csv
|
|
from io import StringIO
|
|
|
|
from flask import url_for
|
|
|
|
from app.alias_delete import delete_alias
|
|
from app.db import Session
|
|
from app.models import Alias, CustomDomain, Mailbox, AliasMailbox
|
|
from tests.utils import login, create_new_user, random_domain, random_token
|
|
|
|
|
|
def alias_export(flask_client, target_url):
|
|
# Create users
|
|
user1 = login(flask_client)
|
|
user2 = create_new_user()
|
|
Session.commit()
|
|
|
|
# Remove onboarding aliases
|
|
for alias in Alias.filter_by(user_id=user1.id).all():
|
|
delete_alias(alias, user1)
|
|
for alias in Alias.filter_by(user_id=user2.id).all():
|
|
delete_alias(alias, user2)
|
|
Session.commit()
|
|
|
|
# Create domains
|
|
ok_domain = CustomDomain.create(
|
|
user_id=user1.id, domain=random_domain(), verified=True
|
|
)
|
|
bad_domain = CustomDomain.create(
|
|
user_id=user2.id, domain=random_domain(), verified=True
|
|
)
|
|
Session.commit()
|
|
|
|
# Create mailboxes
|
|
mailbox1 = Mailbox.create(
|
|
user_id=user1.id, email=f"{random_token()}@{ok_domain.domain}", verified=True
|
|
)
|
|
mailbox2 = Mailbox.create(
|
|
user_id=user1.id, email=f"{random_token()}@{ok_domain.domain}", verified=True
|
|
)
|
|
badmailbox1 = Mailbox.create(
|
|
user_id=user2.id,
|
|
email=f"{random_token()}@{bad_domain.domain}",
|
|
verified=True,
|
|
)
|
|
Session.commit()
|
|
|
|
# Create aliases
|
|
alias1 = Alias.create(
|
|
user_id=user1.id,
|
|
email=f"{random_token()}@my-domain.com",
|
|
note="Used on eBay",
|
|
mailbox_id=mailbox1.id,
|
|
)
|
|
alias2 = Alias.create(
|
|
user_id=user1.id,
|
|
email=f"{random_token()}@my-domain.com",
|
|
note="Used on Facebook, Instagram.",
|
|
mailbox_id=mailbox1.id,
|
|
)
|
|
Alias.create(
|
|
user_id=user2.id,
|
|
email=f"{random_token()}@my-domain.com",
|
|
note="Should not appear",
|
|
mailbox_id=badmailbox1.id,
|
|
)
|
|
Session.commit()
|
|
|
|
# Add second mailbox to an alias
|
|
AliasMailbox.create(
|
|
alias_id=alias2.id,
|
|
mailbox_id=mailbox2.id,
|
|
)
|
|
Session.commit()
|
|
|
|
# Export
|
|
r = flask_client.get(url_for(target_url))
|
|
assert r.status_code == 200
|
|
assert r.mimetype == "text/csv"
|
|
csv_data = csv.DictReader(StringIO(r.data.decode("utf-8")))
|
|
found_aliases = set()
|
|
for row in csv_data:
|
|
found_aliases.add(row["alias"])
|
|
if row["alias"] == alias1.email:
|
|
assert alias1.note == row["note"]
|
|
assert "True" == row["enabled"]
|
|
assert mailbox1.email == row["mailboxes"]
|
|
elif row["alias"] == alias2.email:
|
|
assert alias2.note == row["note"]
|
|
assert "True" == row["enabled"]
|
|
assert f"{mailbox1.email} {mailbox2.email}" == row["mailboxes"]
|
|
else:
|
|
raise AssertionError("Unknown alias")
|
|
assert set((alias1.email, alias2.email)) == found_aliases
|