fix(Litestar): Apply failed_request_status_codes to exceptions raised in middleware

sentry_sdk/integrations/asgi.py

@@ -57,17 +57,6 @@
 TRANSACTION_STYLE_VALUES = ("endpoint", "url")
 
 
-def _capture_exception(exc, mechanism_type="asgi"):
-    # type: (Any, str) -> None
-
-    event, hint = event_from_exception(
-        exc,
-        client_options=sentry_sdk.get_client().options,
-        mechanism={"type": mechanism_type, "handled": False},
-    )
-    sentry_sdk.capture_event(event, hint=hint)
-
-
 def _looks_like_asgi3(app):
     # type: (Any) -> bool
     """
@@ -145,6 +134,23 @@ def __init__(
         else:
             self.__call__ = self._run_asgi2
 
+    def _capture_exception(self, exc):
+        # type: (Exception) -> None
+        event, hint = event_from_exception(
+            exc,
+            client_options=sentry_sdk.get_client().options,
+            mechanism={"type": self.mechanism_type, "handled": False},
+        )
+        sentry_sdk.capture_event(event, hint=hint)
+
+    def _capture_lifespan_exception(self, exc):
+        # type: (Exception) -> None
+        return self._capture_exception(exc)
+
+    def _capture_request_exception(self, exc):
+        # type: (Exception) -> None
+        return self._capture_exception(exc)
+
     def _run_asgi2(self, scope):
         # type: (Any) -> Any
         async def inner(receive, send):
@@ -158,7 +164,7 @@ async def _run_asgi3(self, scope, receive, send):
         return await self._run_app(scope, receive, send, asgi_version=3)
 
     async def _run_app(self, scope, receive, send, asgi_version):
-        # type: (Any, Any, Any, Any, int) -> Any
+        # type: (Any, Any, Any, int) -> Any
         is_recursive_asgi_middleware = _asgi_middleware_applied.get(False)
         is_lifespan = scope["type"] == "lifespan"
         if is_recursive_asgi_middleware or is_lifespan:
@@ -169,7 +175,7 @@ async def _run_app(self, scope, receive, send, asgi_version):
                     return await self.app(scope, receive, send)
 
             except Exception as exc:
-                _capture_exception(exc, mechanism_type=self.mechanism_type)
+                self._capture_lifespan_exception(exc)
                 raise exc from None
 
         _asgi_middleware_applied.set(True)
@@ -255,7 +261,7 @@ async def _sentry_wrapped_send(event):
                                     scope, receive, _sentry_wrapped_send
                                 )
                         except Exception as exc:
-                            _capture_exception(exc, mechanism_type=self.mechanism_type)
+                            self._capture_request_exception(exc)
                             raise exc from None
         finally:
             _asgi_middleware_applied.set(False)

sentry_sdk/integrations/litestar.py

@@ -87,6 +87,10 @@ def __init__(self, app, span_origin=LitestarIntegration.origin):
             span_origin=span_origin,
         )
 
+    def _capture_request_exception(self, exc):
+        # type: (Exception) -> None
+        """Ignore exceptions for requests here: we catch them in Litestar.after_exception handler."""
+
 
 def patch_app_init():
     # type: () -> None

tests/integrations/litestar/test_litestar.py

@@ -16,7 +16,6 @@
 from litestar.middleware.rate_limit import RateLimitConfig
 from litestar.middleware.session.server_side import ServerSideSessionConfig
 from litestar.testing import TestClient
-
 from tests.integrations.conftest import parametrize_test_configurable_status_codes
 
 
@@ -402,7 +401,7 @@ async def __call__(self, scope, receive, send):
 
 
 @parametrize_test_configurable_status_codes
-def test_configurable_status_codes(
+def test_configurable_status_codes_handler(
     sentry_init,
     capture_events,
     failed_request_status_codes,
@@ -427,3 +426,36 @@ async def error() -> None:
     client.get("/error")
 
     assert len(events) == int(expected_error)
+
+
+@parametrize_test_configurable_status_codes
+def test_configurable_status_codes_middleware(
+    sentry_init,
+    capture_events,
+    failed_request_status_codes,
+    status_code,
+    expected_error,
+):
+    integration_kwargs = (
+        {"failed_request_status_codes": failed_request_status_codes}
+        if failed_request_status_codes is not None
+        else {}
+    )
+    sentry_init(integrations=[LitestarIntegration(**integration_kwargs)])
+
+    events = capture_events()
+
+    def create_raising_middleware(app):
+        async def raising_middleware(scope, receive, send):
+            raise HTTPException(status_code=status_code)
+
+        return raising_middleware
+
+    @get("/error")
+    async def error() -> None: ...
+
+    app = Litestar([error], middleware=[create_raising_middleware])
+    client = TestClient(app)
+    client.get("/error")
+
+    assert len(events) == int(expected_error)