canaille-globuzma/tests/app/test_flaskutils.py

import os

import toml

from canaille import create_app
from canaille.app.flask import set_parameter_in_url_query


def test_set_parameter_in_url_query():
    assert (
        set_parameter_in_url_query("https://auth.mydomain.test", foo="bar")
        == "https://auth.mydomain.test?foo=bar"
    )

    assert (
        set_parameter_in_url_query("https://auth.mydomain.test?foo=baz", foo="bar")
        == "https://auth.mydomain.test?foo=bar"
    )

    assert (
        set_parameter_in_url_query("https://auth.mydomain.test?foo=baz", hello="world")
        == "https://auth.mydomain.test?foo=baz&hello=world"
    )


def test_environment_configuration(configuration, tmp_path):
    config_path = os.path.join(tmp_path, "config.toml")
    with open(config_path, "w") as fd:
        toml.dump(configuration, fd)

    os.environ["CONFIG"] = config_path
    app = create_app()
    assert app.config["CANAILLE"]["SMTP"]["FROM_ADDR"] == "admin@mydomain.test"

    del os.environ["CONFIG"]
    os.remove(config_path)