mirror of
https://github.com/simple-login/app.git
synced 2024-11-10 17:35:27 +08:00
a7aec0c37a
* Move set default alias to a separate method to reuse it * Add tests * Find domains by domain not by id * Revert models and setting changes * Remove non required function
116 lines
3.6 KiB
Python
116 lines
3.6 KiB
Python
from app.models import (
|
|
CustomDomain,
|
|
AliasGeneratorEnum,
|
|
SenderFormatEnum,
|
|
AliasSuffixEnum,
|
|
)
|
|
from tests.utils import login, random_domain
|
|
|
|
|
|
def test_get_setting(flask_client):
|
|
login(flask_client)
|
|
|
|
r = flask_client.get("/api/setting")
|
|
assert r.status_code == 200
|
|
assert r.json == {
|
|
"alias_generator": "word",
|
|
"notification": True,
|
|
"random_alias_default_domain": "sl.local",
|
|
"sender_format": "AT",
|
|
"random_alias_suffix": "word",
|
|
}
|
|
|
|
|
|
def test_update_settings_notification(flask_client):
|
|
user = login(flask_client)
|
|
assert user.notification
|
|
|
|
r = flask_client.patch("/api/setting", json={"notification": False})
|
|
assert r.status_code == 200
|
|
assert not user.notification
|
|
|
|
|
|
def test_update_settings_alias_generator(flask_client):
|
|
user = login(flask_client)
|
|
assert user.alias_generator == AliasGeneratorEnum.word.value
|
|
|
|
r = flask_client.patch("/api/setting", json={"alias_generator": "invalid"})
|
|
assert r.status_code == 400
|
|
|
|
r = flask_client.patch("/api/setting", json={"alias_generator": "uuid"})
|
|
assert r.status_code == 200
|
|
assert user.alias_generator == AliasGeneratorEnum.uuid.value
|
|
|
|
|
|
def test_update_settings_random_alias_default_domain(flask_client):
|
|
user = login(flask_client)
|
|
custom_domain = CustomDomain.create(
|
|
domain=random_domain(), verified=True, user_id=user.id, flush=True
|
|
)
|
|
assert user.default_random_alias_domain() == "sl.local"
|
|
|
|
r = flask_client.patch(
|
|
"/api/setting", json={"random_alias_default_domain": "invalid"}
|
|
)
|
|
assert r.status_code == 400
|
|
|
|
r = flask_client.patch(
|
|
"/api/setting", json={"random_alias_default_domain": "d1.test"}
|
|
)
|
|
assert r.status_code == 200
|
|
assert user.default_random_alias_domain() == "d1.test"
|
|
|
|
r = flask_client.patch(
|
|
"/api/setting", json={"random_alias_default_domain": custom_domain.domain}
|
|
)
|
|
assert r.status_code == 200
|
|
assert user.default_random_alias_domain() == custom_domain.domain
|
|
|
|
|
|
def test_update_settings_sender_format(flask_client):
|
|
user = login(flask_client)
|
|
assert user.sender_format == SenderFormatEnum.AT.value
|
|
|
|
r = flask_client.patch("/api/setting", json={"sender_format": "invalid"})
|
|
assert r.status_code == 400
|
|
|
|
r = flask_client.patch("/api/setting", json={"sender_format": "A"})
|
|
assert r.status_code == 200
|
|
assert user.sender_format == SenderFormatEnum.A.value
|
|
|
|
r = flask_client.patch("/api/setting", json={"sender_format": "NAME_ONLY"})
|
|
assert r.status_code == 200
|
|
assert user.sender_format == SenderFormatEnum.NAME_ONLY.value
|
|
|
|
|
|
def test_get_setting_domains(flask_client):
|
|
user = login(flask_client)
|
|
domain = random_domain()
|
|
CustomDomain.create(user_id=user.id, domain=domain, verified=True, commit=True)
|
|
|
|
r = flask_client.get("/api/setting/domains")
|
|
assert r.status_code == 200
|
|
|
|
|
|
def test_get_setting_domains_v2(flask_client):
|
|
user = login(flask_client)
|
|
domain = random_domain()
|
|
CustomDomain.create(user_id=user.id, domain=domain, verified=True, commit=True)
|
|
|
|
r = flask_client.get("/api/v2/setting/domains")
|
|
assert r.status_code == 200
|
|
|
|
|
|
def test_update_settings_random_alias_suffix(flask_client):
|
|
user = login(flask_client)
|
|
# default random_alias_suffix is random_string
|
|
assert user.random_alias_suffix == AliasSuffixEnum.word.value
|
|
|
|
r = flask_client.patch("/api/setting", json={"random_alias_suffix": "invalid"})
|
|
assert r.status_code == 400
|
|
|
|
r = flask_client.patch(
|
|
"/api/setting", json={"random_alias_suffix": "random_string"}
|
|
)
|
|
assert r.status_code == 200
|
|
assert user.random_alias_suffix == AliasSuffixEnum.random_string.value
|