fix(tracing): Fix InvalidOperation (#4179)

`InvalidOperation` can occur when using tracing if the `Decimal` class's
global context has been modified to set the precision below 6. This
change fixes this bug by setting a custom context for our `quantize`
call.

Fixes #4177

Author: sentrivana

sentry_sdk/tracing_utils.py

@@ -6,6 +6,7 @@
 import uuid
 from collections.abc import Mapping
 from datetime import datetime, timedelta, timezone
+from decimal import ROUND_DOWN, Context, Decimal
 from functools import wraps
 from random import Random
 from urllib.parse import quote, unquote
@@ -699,8 +700,6 @@ def _generate_sample_rand(
 
     The pseudorandom number generator is seeded with the trace ID.
     """
-    import decimal
-
     lower, upper = interval
     if not lower < upper:  # using `if lower >= upper` would handle NaNs incorrectly
         raise ValueError("Invalid interval: lower must be less than upper")
@@ -711,8 +710,10 @@ def _generate_sample_rand(
         sample_rand = rng.uniform(lower, upper)
 
     # Round down to exactly six decimal-digit precision.
-    return decimal.Decimal(sample_rand).quantize(
-        decimal.Decimal("0.000001"), rounding=decimal.ROUND_DOWN
+    # Setting the context is needed to avoid an InvalidOperation exception
+    # in case the user has changed the default precision.
+    return Decimal(sample_rand).quantize(
+        Decimal("0.000001"), rounding=ROUND_DOWN, context=Context(prec=6)
     )
 
 

tests/tracing/test_sample_rand.py

@@ -1,3 +1,4 @@
+import decimal
 from unittest import mock
 
 import pytest
@@ -53,3 +54,28 @@ def test_transaction_uses_incoming_sample_rand(
     # Transaction event captured if sample_rand < sample_rate, indicating that
     # sample_rand is used to make the sampling decision.
     assert len(events) == int(sample_rand < sample_rate)
+
+
+def test_decimal_context(sentry_init, capture_events):
+    """
+    Ensure that having a decimal context with a precision below 6
+    does not cause an InvalidOperation exception.
+    """
+    sentry_init(traces_sample_rate=1.0)
+    events = capture_events()
+
+    old_prec = decimal.getcontext().prec
+    decimal.getcontext().prec = 2
+
+    try:
+        with mock.patch(
+            "sentry_sdk.tracing_utils.Random.uniform", return_value=0.123456789
+        ):
+            with sentry_sdk.start_transaction() as transaction:
+                assert (
+                    transaction.get_baggage().sentry_items["sample_rand"] == "0.123456"
+                )
+    finally:
+        decimal.getcontext().prec = old_prec
+
+    assert len(events) == 1