feat: update middleware

DotzInc · May 21, 2024 · 5cd8859 · 5cd8859
1 parent 18817cc
commit 5cd8859
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/fastapi_extras/middlewares/remove_path.py b/fastapi_extras/middlewares/remove_path.py
@@ -10,6 +10,6 @@ async def __call__(self, scope: Scope, receive: Receive, send: Send) -> None:
         if scope["type"] != "http":
             return await self.app(scope, receive, send)
 
-        scope["path"] = scope["path"].replace(self.path, "", 1)
+        scope["path"] = scope["path"].replace(self.path, "/", 1)
 
         await self.app(scope, receive, send)
diff --git a/tests/middlewares/test_remove_path.py b/tests/middlewares/test_remove_path.py
@@ -4,7 +4,7 @@
 from fastapi_extras.middlewares.remove_path import RemovePathMiddleware
 
 app = FastAPI()
-app.add_middleware(RemovePathMiddleware, path="test")
+app.add_middleware(RemovePathMiddleware, path="/test")
 
 
 @app.get("/foo")