|
@@ -2,6 +2,7 @@ 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
|
|
|
|
|
@@ -24,3 +25,18 @@ def test_add_alias_success(flask_client):
|
|
|
|
|
|
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)
|