feat(metrics): Add source context to code locations (#2539)

getsentry · Nov 29, 2023 · bd68a3e · bd68a3e
1 parent 044ce0a
commit bd68a3e
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 1 deletion.
diff --git a/sentry_sdk/metrics.py b/sentry_sdk/metrics.py
@@ -76,7 +76,7 @@ def get_code_location(stacklevel):
         return None
 
     return serialize_frame(
-        frm, include_local_variables=False, include_source_context=False
+        frm, include_local_variables=False, include_source_context=True
     )
 
 

diff --git a/tests/test_metrics.py b/tests/test_metrics.py
@@ -85,6 +85,9 @@ def test_incr(sentry_init, capture_envelopes):
                     "function": sys._getframe().f_code.co_name,
                     "module": __name__,
                     "lineno": mock.ANY,
+                    "pre_context": mock.ANY,
+                    "context_line": mock.ANY,
+                    "post_context": mock.ANY,
                 }
             ]
         },
@@ -133,6 +136,9 @@ def test_timing(sentry_init, capture_envelopes):
                     "function": sys._getframe().f_code.co_name,
                     "module": __name__,
                     "lineno": mock.ANY,
+                    "pre_context": mock.ANY,
+                    "context_line": mock.ANY,
+                    "post_context": mock.ANY,
                 }
             ]
         },
@@ -200,6 +206,9 @@ def amazing_nano():
                     "function": sys._getframe().f_code.co_name,
                     "module": __name__,
                     "lineno": mock.ANY,
+                    "pre_context": mock.ANY,
+                    "context_line": mock.ANY,
+                    "post_context": mock.ANY,
                 }
             ],
             "d:whatever-2@nanosecond": [
@@ -210,6 +219,9 @@ def amazing_nano():
                     "function": sys._getframe().f_code.co_name,
                     "module": __name__,
                     "lineno": mock.ANY,
+                    "pre_context": mock.ANY,
+                    "context_line": mock.ANY,
+                    "post_context": mock.ANY,
                 }
             ],
         },
@@ -261,6 +273,9 @@ def test_timing_basic(sentry_init, capture_envelopes):
                     "function": sys._getframe().f_code.co_name,
                     "module": __name__,
                     "lineno": mock.ANY,
+                    "pre_context": mock.ANY,
+                    "context_line": mock.ANY,
+                    "post_context": mock.ANY,
                 }
             ]
         },
@@ -311,6 +326,9 @@ def test_distribution(sentry_init, capture_envelopes):
                     "function": sys._getframe().f_code.co_name,
                     "module": __name__,
                     "lineno": mock.ANY,
+                    "pre_context": mock.ANY,
+                    "context_line": mock.ANY,
+                    "post_context": mock.ANY,
                 }
             ]
         },
@@ -360,6 +378,9 @@ def test_set(sentry_init, capture_envelopes):
                     "function": sys._getframe().f_code.co_name,
                     "module": __name__,
                     "lineno": mock.ANY,
+                    "pre_context": mock.ANY,
+                    "context_line": mock.ANY,
+                    "post_context": mock.ANY,
                 }
             ]
         },