import connexion.apis.flask_utils as flask_utils


def test_flaskify_path():
    assert flask_utils.flaskify_path("{test-path}") == "<test_path>"
    assert flask_utils.flaskify_path("api/{test-path}") == "api/<test_path>"
    assert flask_utils.flaskify_path("my-api/{test-path}") == "my-api/<test_path>"
    assert flask_utils.flaskify_path("foo_bar/{a-b}/{c_d}") == "foo_bar/<a_b>/<c_d>"
    assert (
        flask_utils.flaskify_path("foo/{a}/{b}", {"a": "integer"}) == "foo/<int:a>/<b>"
    )
    assert (
        flask_utils.flaskify_path("foo/{a}/{b}", {"a": "number"}) == "foo/<float:a>/<b>"
    )
    assert flask_utils.flaskify_path("foo/{a}/{b}", {"a": "path"}) == "foo/<path:a>/<b>"
    assert flask_utils.flaskify_path("foo/{a}", {"a": "path"}) == "foo/<path:a>"


def test_flaskify_endpoint():
    assert flask_utils.flaskify_endpoint("module.function") == "module_function"
    assert flask_utils.flaskify_endpoint("function") == "function"

    name = "module.function"
    randlen = 6
    res = flask_utils.flaskify_endpoint(name, randlen)
    assert res.startswith("module_function")
    assert len(res) == len(name) + 1 + randlen