app-MAIL-temp/tests/dashboard/test_custom_alias.py

43 lines
1.2 KiB
Python
Raw Normal View History

2020-02-05 11:35:38 +01:00
from flask import url_for
from app.config import EMAIL_DOMAIN
from app.extensions import db
from app.models import Mailbox
2020-02-05 11:35:38 +01:00
from app.utils import random_word
from tests.utils import login
def test_add_alias_success(flask_client):
2020-02-10 17:30:04 +01:00
user = login(flask_client)
2020-02-05 11:35:38 +01:00
db.session.commit()
word = random_word()
r = flask_client.post(
url_for("dashboard.custom_alias"),
2020-02-10 17:30:04 +01:00
data={
"prefix": "prefix",
"suffix": f".{word}@{EMAIL_DOMAIN}",
"mailbox": user.email,
},
2020-02-05 11:35:38 +01:00
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)