mirror of
https://github.com/simple-login/app.git
synced 2024-11-17 22:21:38 +08:00
83 lines
2.2 KiB
Python
83 lines
2.2 KiB
Python
from flask import url_for
|
|
|
|
from app.db import Session
|
|
from app.models import SLDomain, CustomDomain, Job
|
|
from tests.utils import login
|
|
|
|
|
|
def setup_sl_domain() -> SLDomain:
|
|
"""Take the first SLDomain and set its can_use_subdomain=True"""
|
|
sl_domain: SLDomain = SLDomain.first()
|
|
sl_domain.can_use_subdomain = True
|
|
Session.commit()
|
|
|
|
return sl_domain
|
|
|
|
|
|
def test_create_subdomain(flask_client):
|
|
login(flask_client)
|
|
sl_domain = setup_sl_domain()
|
|
|
|
r = flask_client.post(
|
|
url_for("dashboard.subdomain_route"),
|
|
data={"form-name": "create", "subdomain": "test", "domain": sl_domain.domain},
|
|
follow_redirects=True,
|
|
)
|
|
|
|
assert r.status_code == 200
|
|
assert f"New subdomain test.{sl_domain.domain} is created" in r.data.decode()
|
|
assert CustomDomain.get_by(domain=f"test.{sl_domain.domain}") is not None
|
|
|
|
|
|
def test_delete_subdomain(flask_client):
|
|
user = login(flask_client)
|
|
sl_domain = setup_sl_domain()
|
|
|
|
subdomain = CustomDomain.create(
|
|
domain=f"test.{sl_domain.domain}",
|
|
user_id=user.id,
|
|
is_sl_subdomain=True,
|
|
commit=True,
|
|
)
|
|
|
|
nb_job = Job.count()
|
|
|
|
r = flask_client.post(
|
|
url_for("dashboard.domain_detail", custom_domain_id=subdomain.id),
|
|
data={"form-name": "delete"},
|
|
follow_redirects=True,
|
|
)
|
|
|
|
assert r.status_code == 200
|
|
assert f"test.{sl_domain.domain} scheduled for deletion." in r.data.decode()
|
|
|
|
# a domain deletion job is scheduled
|
|
assert Job.count() == nb_job + 1
|
|
|
|
|
|
def test_create_subdomain_in_trash(flask_client):
|
|
user = login(flask_client)
|
|
sl_domain = setup_sl_domain()
|
|
|
|
subdomain = CustomDomain.create(
|
|
domain=f"test.{sl_domain.domain}",
|
|
user_id=user.id,
|
|
is_sl_subdomain=True,
|
|
commit=True,
|
|
)
|
|
|
|
# delete the subdomain
|
|
CustomDomain.delete(subdomain.id)
|
|
assert CustomDomain.get_by(domain=f"test.{sl_domain.domain}") is None
|
|
|
|
r = flask_client.post(
|
|
url_for("dashboard.subdomain_route"),
|
|
data={"form-name": "create", "subdomain": "test", "domain": sl_domain.domain},
|
|
follow_redirects=True,
|
|
)
|
|
|
|
assert r.status_code == 200
|
|
assert (
|
|
f"test.{sl_domain.domain} has been used before and cannot be reused"
|
|
in r.data.decode()
|
|
)
|