Exemplo n.º 1
0
def test_recipe_book(app: App, client, numbers):
    app.recipe(numbers)

    response = client.get("/numbers/integers/3.5")
    assert response.status_code == 200
    assert response.json() == {"value": 3}

    response = client.get("/numbers/floats/1")
    assert response.status_code == 200
    assert response.json() == {"value": 1.0}
Exemplo n.º 2
0
def test_if_prefix_not_given_then_routes_mounted_at_slash_name(app: App):
    numbers = Recipe("numbers")

    @numbers.route("/real")
    async def real(req, res):
        pass

    app.recipe(numbers)

    assert app.url_for("numbers:real") == "/numbers/real"
Exemplo n.º 3
0
def test_if_prefix_then_routes_mounted_at_prefix(app: App):
    numbers = Recipe("numbers", prefix="/numbers-yo")

    @numbers.route("/real")
    async def real(req, res):
        pass

    app.recipe(numbers)

    assert app.url_for("numbers:real") == "/numbers-yo/real"
Exemplo n.º 4
0
def test_websocket_recipe_route(app: App, client):
    chat = Recipe("chat")

    @chat.websocket_route("/room/{name}",
                          receive_type="json",
                          send_type="text")
    async def chat_room(ws: WebSocket, name: str):
        message = await ws.receive()
        await ws.send(f"[{name}]: {message['text']}")

    app.recipe(chat)

    with client.websocket_connect("/chat/room/test") as ws:
        ws.send_json({"text": "Hello"})
        assert ws.receive_text() == "[test]: Hello"
Exemplo n.º 5
0
def test_redirect(app: App, client):
    numbers = Recipe("numbers")

    @numbers.route("/R")
    async def R(req, res):
        numbers.redirect(name="numbers:real")

    @numbers.route("/real")
    async def real(req, res):
        res.text = "inf"

    app.recipe(numbers)

    response = client.get("/numbers/R")
    assert response.status_code == 200
    assert response.text == "inf"
Exemplo n.º 6
0
# 👇 Executes `requires_token()` before the view
@courses.route("/top")
@hooks.before(requires_token)
async def get_top_courses(req, res):
    try:
        # Query parameters!
        n = int(req.query_params.get("n"))
    except TypeError:
        n = 5
    courses = [storage.get(pk)._asdict() for pk in analytics.top(n=n)]
    res.media = courses


# Mounts the routes of `courses` at `/courses` on `app`
app.recipe(courses)


# Custom error handlers!
@app.error_handler(HTTPError)
def handle_json(req, res, exc):
    res.media = {
        "error": exc.status_phrase,
        "status": exc.status_code,
        "message": "Duh!",
    }
    res.status_code = exc.status_code


if __name__ == "__main__":
    app.run()