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

61 lines
1.6 KiB
Python
Raw Normal View History

2020-01-28 09:13:03 +01:00
from flask import url_for
from app.db import Session
from app.email_utils import get_email_domain_part
from app.models import Mailbox
2020-01-28 09:13:03 +01:00
from tests.utils import login
def test_add_domain_success(flask_client):
user = login(flask_client)
user.lifetime = True
Session.commit()
2020-01-28 09:13:03 +01:00
r = flask_client.post(
url_for("dashboard.custom_domain"),
data={"form-name": "create", "domain": "ab.cd"},
follow_redirects=True,
)
assert r.status_code == 200
assert b"New domain ab.cd is created" in r.data
def test_add_domain_same_as_user_email(flask_client):
"""cannot add domain if user personal email uses this domain"""
user = login(flask_client)
user.lifetime = True
Session.commit()
r = flask_client.post(
url_for("dashboard.custom_domain"),
data={"form-name": "create", "domain": get_email_domain_part(user.email)},
follow_redirects=True,
)
assert r.status_code == 200
assert (
b"You cannot add a domain that you are currently using for your personal email"
in r.data
)
def test_add_domain_used_in_mailbox(flask_client):
"""cannot add domain if it has been used in a verified mailbox"""
user = login(flask_client)
user.lifetime = True
Session.commit()
Mailbox.create(
user_id=user.id, email="mailbox@new-domain.com", verified=True, commit=True
)
r = flask_client.post(
url_for("dashboard.custom_domain"),
data={"form-name": "create", "domain": "new-domain.com"},
follow_redirects=True,
)
assert r.status_code == 200
assert b"new-domain.com already used in a SimpleLogin mailbox" in r.data