def test_add_rule_decorators(client, flask_app): def hello_world(): return "hello world" api = Api(flask_app, request_decorators=[auth_required]) api.add_url_rule("/", "index", hello_world, request_decorators={"POST": [post_hook]}) resp = client.get("/") assert resp.status_code == 403 resp = client.post("/", headers={"auth": True}) assert resp.status_code == 200 assert resp.data == b"hello worldpost_hook"
def test_url_rule(flask_app, client): def hello_world(name): return f"hello {name}" api = Api(flask_app) api.add_url_rule("/<string:name>", "index", hello_world) resp = client.get("/raynor") assert resp.status_code == 200 assert resp.data == b"hello raynor" resp = client.post("/kerrigan") assert resp.status_code == 200 assert resp.data == b"hello kerrigan" resp = client.put("/artanis") assert resp.status_code == 200 assert resp.data == b"hello artanis" resp = client.delete("/zeratul") assert resp.status_code == 200 assert resp.data == b"hello zeratul"