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