app-MAIL-temp/tests/api/test_setting.py

from app.models import (
    CustomDomain,
    AliasGeneratorEnum,
    SenderFormatEnum,
    AliasSuffixEnum,
)
from tests.utils import login


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": "random_string",
    }


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)
    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"


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)
    CustomDomain.create(user_id=user.id, domain="ab.cd", 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)
    CustomDomain.create(user_id=user.id, domain="ab.cd", 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.random_string.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": "word"})
    assert r.status_code == 200
    assert user.random_alias_suffix == AliasSuffixEnum.word.value