Add tests

Author: vrslev

tests/integrations/litestar/test_litestar.py

@@ -1,6 +1,7 @@
 from __future__ import annotations
 import functools
 
+from litestar.exceptions import HTTPException
 import pytest
 
 from sentry_sdk import capture_message
@@ -16,6 +17,8 @@
 from litestar.middleware.session.server_side import ServerSideSessionConfig
 from litestar.testing import TestClient
 
+from tests.integrations.starlette import test_starlette
+
 
 def litestar_app_factory(middleware=None, debug=True, exception_handlers=None):
     class MyController(Controller):
@@ -396,3 +399,25 @@ async def __call__(self, scope, receive, send):
         }
     else:
         assert "user" not in event
+@test_starlette.parametrize_test_configurable_status_codes
+def test_configurable_status_codes(
+    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()
+
+    @get("/error")
+    async def error()-> None:
+        raise HTTPException(status_code=status_code)
+
+    app = Litestar([error])
+    client = TestClient(app)
+    client.get("/error")
+
+    assert len(events) == int(expected_error)