add pydantic test without mocks

Author: shellmayr

tests/integrations/pydantic_ai/test_pydantic_ai.py

@@ -2604,3 +2604,31 @@ async def test_ai_client_span_gets_agent_from_scope(sentry_init, capture_events)
 
     # Should not crash
     assert transaction is not None
+
+
+@pytest.mark.asyncio
+async def test_set_usage_data_with_cache_tokens(sentry_init, capture_events):
+    """Test that cache_read_tokens and cache_write_tokens are tracked."""
+    import sentry_sdk
+    from pydantic_ai.usage import RequestUsage
+    from sentry_sdk.integrations.pydantic_ai.spans.utils import _set_usage_data
+    from sentry_sdk.consts import SPANDATA
+
+    sentry_init(integrations=[PydanticAIIntegration()], traces_sample_rate=1.0)
+    events = capture_events()
+
+    with sentry_sdk.start_transaction(op="test", name="test"):
+        span = sentry_sdk.start_span(op="test_span")
+        usage = RequestUsage(
+            input_tokens=100,
+            output_tokens=50,
+            cache_read_tokens=80,
+            cache_write_tokens=20,
+        )
+        _set_usage_data(span, usage)
+        span.finish()
+
+    (event,) = events
+    (span_data,) = event["spans"]
+    assert span_data["data"][SPANDATA.GEN_AI_USAGE_INPUT_TOKENS_CACHED] == 80
+    assert span_data["data"][SPANDATA.GEN_AI_USAGE_INPUT_TOKENS_CACHE_WRITE] == 20