diff --git a/src/testbench_tuna/main.py b/src/testbench_tuna/main.py index aa4e4d6..ffdf4f8 100644 --- a/src/testbench_tuna/main.py +++ b/src/testbench_tuna/main.py @@ -1,6 +1,6 @@ from typing import Union -from fastapi import FastAPI +from fastapi import FastAPI, HTTPException from . import utils @@ -15,3 +15,13 @@ async def get_root(): @app.get("/pizza-size") async def get_pizza_size(diameter: Union[int, float]): return {"area": utils.pizza_size(diameter)} + + +@app.get("/users/{username}/email") +async def get_users_username_email(username: str): + if username == "foo": + return {"username": username, "email": "foo@example.com"} + elif username == "bar": + return {"username": username, "email": "bar@example.com"} + else: + raise HTTPException(status_code=404, detail="User not found")