fix(aiomysql): Use record_sql_queries (renamed back in upstream)

Upstream renamed record_sql_queries_supporting_streaming back to
record_sql_queries since it now natively supports streaming.
Update imports and mock targets accordingly.

Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>

Author: tonal

sentry_sdk/integrations/aiomysql.py

@@ -10,7 +10,7 @@
 from sentry_sdk.tracing_utils import (
     add_query_source,
     has_span_streaming_enabled,
-    record_sql_queries_supporting_streaming,
+    record_sql_queries,
 )
 from sentry_sdk.utils import (
     capture_internal_exceptions,
@@ -85,7 +85,7 @@ async def _inner(*args: Any, **kwargs: Any) -> T:
         params_list = params if integration and integration._record_params else None
         param_style = "pyformat" if params_list else None
 
-        with record_sql_queries_supporting_streaming(
+        with record_sql_queries(
             cursor=None,
             query=query_str,
             params_list=params_list,
@@ -132,7 +132,7 @@ async def _inner(*args: Any, **kwargs: Any) -> T:
         # Prevent double-recording: _do_execute_many calls self.execute internally
         cursor._sentry_skip_next_execute = True
         try:
-            with record_sql_queries_supporting_streaming(
+            with record_sql_queries(
                 cursor=None,
                 query=query_str,
                 params_list=params_list,

tests/integrations/aiomysql/test_aiomysql.py

@@ -22,7 +22,7 @@
 from sentry_sdk import capture_message, start_transaction
 from sentry_sdk.consts import SPANDATA
 from sentry_sdk.integrations.aiomysql import AioMySQLIntegration
-from sentry_sdk.tracing_utils import record_sql_queries_supporting_streaming
+from sentry_sdk.tracing_utils import record_sql_queries
 from tests.conftest import ApproxDict
 
 MYSQL_HOST = os.getenv("SENTRY_PYTHON_TEST_MYSQL_HOST", "localhost")
@@ -570,15 +570,15 @@ async def test_no_query_source_if_duration_too_short(sentry_init, capture_events
 
         @contextmanager
         def fake_record_sql_queries(*args, **kwargs):
-            with record_sql_queries_supporting_streaming(*args, **kwargs) as span:
+            with record_sql_queries(*args, **kwargs) as span:
                 pass
             span.start_timestamp = datetime.datetime(2024, 1, 1, microsecond=0)
             span.timestamp = datetime.datetime(2024, 1, 1, microsecond=99999)
             yield span
 
         async with conn.cursor() as cur:
             with mock.patch(
-                "sentry_sdk.integrations.aiomysql.record_sql_queries_supporting_streaming",
+                "sentry_sdk.integrations.aiomysql.record_sql_queries",
                 fake_record_sql_queries,
             ):
                 await cur.execute(
@@ -616,15 +616,15 @@ async def test_query_source_if_duration_over_threshold(sentry_init, capture_even
 
         @contextmanager
         def fake_record_sql_queries(*args, **kwargs):
-            with record_sql_queries_supporting_streaming(*args, **kwargs) as span:
+            with record_sql_queries(*args, **kwargs) as span:
                 pass
             span.start_timestamp = datetime.datetime(2024, 1, 1, microsecond=0)
             span.timestamp = datetime.datetime(2024, 1, 1, microsecond=100001)
             yield span
 
         async with conn.cursor() as cur:
             with mock.patch(
-                "sentry_sdk.integrations.aiomysql.record_sql_queries_supporting_streaming",
+                "sentry_sdk.integrations.aiomysql.record_sql_queries",
                 fake_record_sql_queries,
             ):
                 await cur.execute(