Add mock X-Total-Count

1 files changed, 7 insertions, 0 deletions

diff --git a/src/main.py b/src/main.py
index 4d20c84..ba2508d 100755
--- a/src/main.py
+++ b/src/main.py
@@ -22,6 +22,13 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
+# TODO: X-Total-Count
+@app.middleware("http")
+async def mock_x_total_count_header(request: Request, call_next):
+    response = await call_next(request)
+    response.headers["X-Total-Count"] = 100
+    return response
+
 for i in range(10):
     try:
         db = DictDB()