59 lines
1.7 KiB
Python
59 lines
1.7 KiB
Python
|
from flask import url_for
|
||
|
|
||
|
from app.models import Directory
|
||
|
from tests.utils import login
|
||
|
|
||
|
|
||
|
def test_create_directory(flask_client):
|
||
|
login(flask_client)
|
||
|
|
||
|
r = flask_client.post(
|
||
|
url_for("dashboard.directory"),
|
||
|
data={"form-name": "create", "name": "test"},
|
||
|
follow_redirects=True,
|
||
|
)
|
||
|
|
||
|
assert r.status_code == 200
|
||
|
assert f"Directory test is created" in r.data.decode()
|
||
|
assert Directory.get_by(name="test") is not None
|
||
|
|
||
|
|
||
|
def test_delete_directory(flask_client):
|
||
|
"""cannot add domain if user personal email uses this domain"""
|
||
|
user = login(flask_client)
|
||
|
directory = Directory.create(name="test", user_id=user.id, commit=True)
|
||
|
|
||
|
r = flask_client.post(
|
||
|
url_for("dashboard.directory"),
|
||
|
data={"form-name": "delete", "dir-id": directory.id},
|
||
|
follow_redirects=True,
|
||
|
)
|
||
|
|
||
|
assert r.status_code == 200
|
||
|
assert f"Directory test has been deleted" in r.data.decode()
|
||
|
assert Directory.get_by(name="test") is None
|
||
|
|
||
|
|
||
|
def test_create_directory_in_trash(flask_client):
|
||
|
user = login(flask_client)
|
||
|
|
||
|
directory = Directory.create(name="test", user_id=user.id, commit=True)
|
||
|
|
||
|
# delete the directory
|
||
|
r = flask_client.post(
|
||
|
url_for("dashboard.directory"),
|
||
|
data={"form-name": "delete", "dir-id": directory.id},
|
||
|
follow_redirects=True,
|
||
|
)
|
||
|
assert Directory.get_by(name="test") is None
|
||
|
|
||
|
# try to recreate the directory
|
||
|
r = flask_client.post(
|
||
|
url_for("dashboard.directory"),
|
||
|
data={"form-name": "create", "name": "test"},
|
||
|
follow_redirects=True,
|
||
|
)
|
||
|
|
||
|
assert r.status_code == 200
|
||
|
assert "test has been used before and cannot be reused" in r.data.decode()
|