HH-217412 upgrade sentry

frontik/integrations/sentry.py

@@ -1,9 +1,17 @@
 from __future__ import annotations
 
 from typing import TYPE_CHECKING, Optional
+import logging
 
 import sentry_sdk
 from sentry_sdk.integrations.fastapi import FastApiIntegration, StarletteIntegration
+from sentry_sdk.integrations.aiohttp import AioHttpIntegration
+from sentry_sdk.integrations.atexit import AtexitIntegration
+from sentry_sdk.integrations.dedupe import DedupeIntegration
+from sentry_sdk.integrations.stdlib import StdlibIntegration
+from sentry_sdk.integrations.excepthook import ExcepthookIntegration
+from sentry_sdk.integrations.modules import ModulesIntegration
+from sentry_sdk.integrations.logging import LoggingIntegration
 
 from frontik.integrations import Integration, integrations_logger
 from frontik.options import options
@@ -25,7 +33,21 @@ def initialize_app(self, app: FrontikApplication) -> Optional[Future]:
             max_breadcrumbs=options.sentry_max_breadcrumbs,
             default_integrations=False,
             auto_enabling_integrations=False,
-            integrations=[FastApiIntegration(), StarletteIntegration()],
+            integrations=[
+                AioHttpIntegration(),
+                FastApiIntegration(),
+                StarletteIntegration(),
+                AtexitIntegration(),
+                DedupeIntegration(),
+                ExcepthookIntegration(),
+                ModulesIntegration(),
+                StdlibIntegration(),
+                LoggingIntegration(level=None, event_level=logging.WARNING),
+            ],
+            sample_rate=options.sentry_sample_rate,
+            enable_tracing=options.sentry_enable_tracing,
+            traces_sample_rate=options.sentry_traces_sample_rate,
+            in_app_include=list(filter(None, options.sentry_in_app_include.split(','))),
         )
 
         return None

frontik/options.py

@@ -79,6 +79,10 @@ class Options:
 
     sentry_dsn: Optional[str] = None
     sentry_max_breadcrumbs: int = 0
+    sentry_sample_rate: float = 1.0
+    sentry_enable_tracing: Optional[bool] = None
+    sentry_traces_sample_rate: Optional[float] = None
+    sentry_in_app_include: str = ''
 
     send_timeout_stats_interval_ms: int = 60000
 

tests/test_sentry_integration.py

@@ -20,7 +20,7 @@ def initialize_sentry_logger(self):
 
 
 @router.get('/sentry_error', cls=Page)
-async def get_page(handler=get_current_handler()):
+async def get_page(handler: Page = get_current_handler()):
     ip = handler.get_query_argument('ip', None)
     extra = handler.get_query_argument('extra_key', None)
     a = 155
@@ -59,7 +59,7 @@ async def test_sentry_exception(self):
         assert len(sentry_events) == 1
         event = sentry_events[0]
         assert len(event['breadcrumbs']['values']) == 0
-        assert event.get('modules') is None
+        assert event.get('modules') is not None
         assert event['request']['query_string'] == 'ip=127.0.0.77&extra_key=extra_val'
         assert event['user']['real_ip'] == '127.0.0.77'
         assert event['extra']['extra_key'] == 'extra_val'