Add tests

farhatnawaz · farhatnawaz · commit cf3cc2259c50 · 2023-04-18T05:09:31.000+05:00
diff --git a/tests/test_utils.py b/tests/test_utils.py
@@ -1,7 +1,14 @@
 import pytest
 import re
+import sys
 
-from sentry_sdk.utils import is_valid_sample_rate, logger, parse_url, sanitize_url
+from sentry_sdk.utils import (
+    is_valid_sample_rate,
+    logger,
+    parse_url,
+    sanitize_url,
+    serialize_frame,
+)
 
 try:
     from unittest import mock  # python 3.3 and above
@@ -221,3 +228,21 @@ def test_warns_on_invalid_sample_rate(rate, StringContaining):  # noqa: N803
         result = is_valid_sample_rate(rate, source="Testing")
         logger.warning.assert_any_call(StringContaining("Given sample rate is invalid"))
         assert result is False
+
+
+def test_include_source_context_when_serializing_frame():
+    frame = sys._getframe()
+    result = serialize_frame(frame)
+
+    assert "pre_context" in result
+    assert "context_line" in result
+    assert "post_context" in result
+
+
+def test_exclude_source_context_when_serializing_frame():
+    frame = sys._getframe()
+    result = serialize_frame(frame, include_source_context=False)
+
+    assert "pre_context" not in result
+    assert "context_line" not in result
+    assert "post_context" not in result
