* Fix: Add csrf verification to directory updates * Update templates/dashboard/directory.html * Added csrf for delete account form * Fix tests * Added CSRF check for settings page * Added csrf to batch import * Added CSRF to alias dashboard and alias transfer * Added csrf to contact manager * Added csrf to mailbox * Added csrf for mailbox detail * Added csrf to domain detail * Lint Co-authored-by: Adrià Casajús <adria.casajus@proton.ch>
85 lines
2.6 KiB
Python
85 lines
2.6 KiB
Python
from flask import url_for
|
|
|
|
from app.config import MAX_NB_DIRECTORY
|
|
from app.models import Directory
|
|
from tests.utils import login, random_token
|
|
|
|
|
|
def test_create_directory(flask_client):
|
|
login(flask_client)
|
|
|
|
directory_name = random_token()
|
|
r = flask_client.post(
|
|
url_for("dashboard.directory"),
|
|
data={"form-name": "create", "name": directory_name},
|
|
follow_redirects=True,
|
|
)
|
|
|
|
assert r.status_code == 200
|
|
assert f"Directory {directory_name} is created" in r.data.decode()
|
|
assert Directory.get_by(name=directory_name) is not None
|
|
|
|
|
|
def test_delete_directory(flask_client):
|
|
"""cannot add domain if user personal email uses this domain"""
|
|
user = login(flask_client)
|
|
directory_name = random_token()
|
|
directory = Directory.create(name=directory_name, user_id=user.id, commit=True)
|
|
|
|
r = flask_client.post(
|
|
url_for("dashboard.directory"),
|
|
data={"form-name": "delete", "directory_id": directory.id},
|
|
follow_redirects=True,
|
|
)
|
|
|
|
assert r.status_code == 200
|
|
assert f"Directory {directory_name} has been deleted" in r.data.decode()
|
|
assert Directory.get_by(name=directory_name) is None
|
|
|
|
|
|
def test_create_directory_in_trash(flask_client):
|
|
user = login(flask_client)
|
|
directory_name = random_token()
|
|
|
|
directory = Directory.create(name=directory_name, user_id=user.id, commit=True)
|
|
|
|
# delete the directory
|
|
r = flask_client.post(
|
|
url_for("dashboard.directory"),
|
|
data={"form-name": "delete", "directory_id": directory.id},
|
|
follow_redirects=True,
|
|
)
|
|
assert Directory.get_by(name=directory_name) is None
|
|
|
|
# try to recreate the directory
|
|
r = flask_client.post(
|
|
url_for("dashboard.directory"),
|
|
data={"form-name": "create", "name": directory_name},
|
|
follow_redirects=True,
|
|
)
|
|
|
|
assert r.status_code == 200
|
|
assert (
|
|
f"{directory_name} has been used before and cannot be reused" in r.data.decode()
|
|
)
|
|
|
|
|
|
def test_create_directory_out_of_quota(flask_client):
|
|
user = login(flask_client)
|
|
|
|
for i in range(
|
|
MAX_NB_DIRECTORY - Directory.filter(Directory.user_id == user.id).count()
|
|
):
|
|
Directory.create(name=f"test{i}", user_id=user.id, commit=True)
|
|
|
|
assert Directory.filter(Directory.user_id == user.id).count() == MAX_NB_DIRECTORY
|
|
|
|
flask_client.post(
|
|
url_for("dashboard.directory"),
|
|
data={"form-name": "create", "name": "test"},
|
|
follow_redirects=True,
|
|
)
|
|
|
|
# no new directory is created
|
|
assert Directory.filter(Directory.user_id == user.id).count() == MAX_NB_DIRECTORY
|