squash! Add endpoint

trallnag · Aug 16, 2022 · 4cf10d2 · 4cf10d2
1 parent 4421038
commit 4cf10d2
Showing 1 changed file with 11 additions and 1 deletion.
diff --git 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")