43 lines
1.2 KiB
Python
43 lines
1.2 KiB
Python
from flask import url_for
|
|
|
|
from app.config import EMAIL_DOMAIN
|
|
from app.extensions import db
|
|
from app.models import Mailbox
|
|
from app.utils import random_word
|
|
from tests.utils import login
|
|
|
|
|
|
def test_add_alias_success(flask_client):
|
|
user = login(flask_client)
|
|
db.session.commit()
|
|
|
|
word = random_word()
|
|
|
|
r = flask_client.post(
|
|
url_for("dashboard.custom_alias"),
|
|
data={
|
|
"prefix": "prefix",
|
|
"suffix": f".{word}@{EMAIL_DOMAIN}",
|
|
"mailbox": user.email,
|
|
},
|
|
follow_redirects=True,
|
|
)
|
|
|
|
assert r.status_code == 200
|
|
assert f"Alias prefix.{word}@{EMAIL_DOMAIN} has been created" in str(r.data)
|
|
|
|
|
|
def test_not_show_unverified_mailbox(flask_client):
|
|
"""make sure user unverified mailbox is not shown to user"""
|
|
user = login(flask_client)
|
|
db.session.commit()
|
|
|
|
Mailbox.create(user_id=user.id, email="m1@example.com", verified=True)
|
|
Mailbox.create(user_id=user.id, email="m2@example.com", verified=False)
|
|
db.session.commit()
|
|
|
|
r = flask_client.get(url_for("dashboard.custom_alias"))
|
|
|
|
assert "m1@example.com" in str(r.data)
|
|
assert "m2@example.com" not in str(r.data)
|