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

from app.alias_utils import delete_alias
from app.models import CustomDomain, Alias, Mailbox
from tests.utils import login


def test_get_custom_domains(flask_client):
    user = login(flask_client)

    CustomDomain.create(user_id=user.id, domain="test1.org", verified=True, commit=True)
    CustomDomain.create(
        user_id=user.id, domain="test2.org", verified=False, commit=True
    )

    r = flask_client.get(
        "/api/custom_domains",
    )

    assert r.status_code == 200
    assert len(r.json["custom_domains"]) == 2
    for domain in r.json["custom_domains"]:
        assert domain["domain_name"]
        assert domain["id"]
        assert domain["nb_alias"] == 0
        assert "is_verified" in domain
        assert "catch_all" in domain
        assert "name" in domain
        assert "random_prefix_generation" in domain
        assert domain["creation_date"]
        assert domain["creation_timestamp"]

        assert domain["mailboxes"]
        for mailbox in domain["mailboxes"]:
            assert "id" in mailbox
            assert "email" in mailbox


def test_update_custom_domains(flask_client):
    user = login(flask_client)

    d1 = CustomDomain.create(
        user_id=user.id, domain="test1.org", verified=True, commit=True
    )

    # test update catch all
    assert d1.catch_all is False
    r = flask_client.patch(f"/api/custom_domains/{d1.id}", json={"catch_all": True})
    assert r.status_code == 200
    assert d1.catch_all is True

    # test update random_prefix_generation
    assert d1.random_prefix_generation is False
    r = flask_client.patch(
        f"/api/custom_domains/{d1.id}", json={"random_prefix_generation": True}
    )
    assert r.status_code == 200
    assert d1.random_prefix_generation is True

    # test update name
    assert d1.name is None
    r = flask_client.patch(f"/api/custom_domains/{d1.id}", json={"name": "test name"})
    assert r.status_code == 200
    assert d1.name == "test name"

    # test update mailboxes
    assert d1.mailboxes == [user.default_mailbox]
    mb = Mailbox.create(
        user_id=user.id, email="test@example.org", verified=True, commit=True
    )
    r = flask_client.patch(
        f"/api/custom_domains/{d1.id}", json={"mailbox_ids": [mb.id]}
    )
    assert r.status_code == 200
    assert d1.mailboxes == [mb]


def test_get_custom_domain_trash(flask_client):
    user = login(flask_client)

    cd = CustomDomain.create(
        user_id=user.id, domain="test1.org", verified=True, commit=True
    )

    alias = Alias.create(
        user_id=user.id,
        email="first@test1.org",
        custom_domain_id=cd.id,
        mailbox_id=user.default_mailbox_id,
        commit=True,
    )

    delete_alias(alias, user)

    r = flask_client.get(
        f"/api/custom_domains/{cd.id}/trash",
    )

    for deleted_alias in r.json["aliases"]:
        assert deleted_alias["alias"]
        assert deleted_alias["deletion_timestamp"] > 0