Add support for middleware injection on routes

sanic-org · Sep 21, 2022 · cdfd1bc · cdfd1bc
1 parent ca15294
commit cdfd1bc
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 1 deletion.
diff --git a/sanic_testing/__init__.py b/sanic_testing/__init__.py
@@ -1,4 +1,4 @@
 from sanic_testing.manager import TestManager
 
-__version__ = "22.9.0b1"
+__version__ = "22.9.0b2"
 __all__ = ("TestManager",)
diff --git a/sanic_testing/reusable.py b/sanic_testing/reusable.py
@@ -62,6 +62,7 @@ def __init__(
 
     def __enter__(self):
         self.run()
+        return self
 
     def __exit__(self, *_):
         self.stop()
@@ -114,6 +115,10 @@ def _sanic_endpoint_test(
                 _collect_request
             )
 
+            for route in self.app.router.routes:
+                if _collect_request not in route.extra.request_middleware:
+                    route.extra.request_middleware.appendleft(_collect_request)
+
         if uri.startswith(
             ("http:", "https:", "ftp:", "ftps://", "//", "ws:", "wss:")
         ):

diff --git a/sanic_testing/testing.py b/sanic_testing/testing.py
@@ -372,6 +372,11 @@ async def request(  # type: ignore
         await self.sanic_app._startup()  # type: ignore
         await self.sanic_app._server_event("init", "before")
         await self.sanic_app._server_event("init", "after")
+        for route in self.sanic_app.router.routes:
+            if self._collect_request not in route.extra.request_middleware:
+                route.extra.request_middleware.appendleft(
+                    self._collect_request
+                )
 
         if not url.startswith(
             ("http:", "https:", "ftp:", "ftps://", "//", "ws:", "wss:")