add a test route with varying arguments

Signed-off-by: Grant Ramsay <[email protected]>
seapagan · Mar 30, 2024 · 1a6ffa3 · 1a6ffa3
1 parent f982112
commit 1a6ffa3
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/tests/live_test.py b/tests/live_test.py
@@ -94,3 +94,13 @@ def cache_invalid_type(request: Request, response: Response) -> logging.Logger:
     logger = logging.getLogger(__name__)
     logger.setLevel(logging.INFO)
     return logger
+
+
+@app.get("/cache_with_args/{user}")
+@cache_one_hour(tag="user_tag")
+def cache_with_args(user: int) -> dict[str, Union[bool, str]]:
+    """Have a varying cache key based on the user argument."""
+    return {
+        "success": True,
+        "message": f"this data is for user {user}",
+    }
diff --git a/tests/main.py b/tests/main.py
@@ -63,3 +63,13 @@ def cache_invalid_type(request: Request, response: Response) -> logging.Logger:
     logger = logging.getLogger(__name__)
     logger.setLevel(logging.INFO)
     return logger
+
+
+@app.get("/cache_with_args/{user}")
+@cache_one_hour(tag="user_tag")
+def cache_with_args(user: int) -> dict[str, Union[bool, str]]:
+    """Have a varying cache key based on the user argument."""
+    return {
+        "success": True,
+        "message": f"this data is for user {user}",
+    }