Update test

Author: antonpirker

tests/integrations/anthropic/test_anthropic.py

@@ -1,4 +1,3 @@
-import json
 from unittest import mock
 
 try:
@@ -20,7 +19,7 @@ async def __call__(self, *args, **kwargs):
 from anthropic.types.message_delta_event import MessageDeltaEvent
 from anthropic.types.message_start_event import MessageStartEvent
 
-from sentry_sdk.utils import package_version
+from sentry_sdk.utils import _serialize_span_attribute, package_version
 
 try:
     from anthropic.types import InputJSONDelta
@@ -116,8 +115,8 @@ def test_nonstreaming_create_message(
     assert span["data"][SPANDATA.AI_MODEL_ID] == "model"
 
     if send_default_pii and include_prompts:
-        assert span["data"][SPANDATA.AI_INPUT_MESSAGES] == json.dumps(messages)
-        assert span["data"][SPANDATA.AI_RESPONSES] == json.dumps(
+        assert span["data"][SPANDATA.AI_INPUT_MESSAGES] == _serialize_span_attribute(messages)
+        assert span["data"][SPANDATA.AI_RESPONSES] == _serialize_span_attribute(
             [{"type": "text", "text": "Hi, I'm Claude."}]
         )
     else:
@@ -184,8 +183,8 @@ async def test_nonstreaming_create_message_async(
     assert span["data"][SPANDATA.AI_MODEL_ID] == "model"
 
     if send_default_pii and include_prompts:
-        assert span["data"][SPANDATA.AI_INPUT_MESSAGES] == json.dumps(messages)
-        assert span["data"][SPANDATA.AI_RESPONSES] == json.dumps(
+        assert span["data"][SPANDATA.AI_INPUT_MESSAGES] == _serialize_span_attribute(messages)
+        assert span["data"][SPANDATA.AI_RESPONSES] == _serialize_span_attribute(
             [{"type": "text", "text": "Hi, I'm Claude."}]
         )
     else:
@@ -283,8 +282,8 @@ def test_streaming_create_message(
     assert span["data"][SPANDATA.AI_MODEL_ID] == "model"
 
     if send_default_pii and include_prompts:
-        assert span["data"][SPANDATA.AI_INPUT_MESSAGES] == json.dumps(messages)
-        assert span["data"][SPANDATA.AI_RESPONSES] == json.dumps(
+        assert span["data"][SPANDATA.AI_INPUT_MESSAGES] == _serialize_span_attribute(messages)
+        assert span["data"][SPANDATA.AI_RESPONSES] == _serialize_span_attribute(
             [{"type": "text", "text": "Hi! I'm Claude!"}]
         )
 
@@ -386,8 +385,8 @@ async def test_streaming_create_message_async(
     assert span["data"][SPANDATA.AI_MODEL_ID] == "model"
 
     if send_default_pii and include_prompts:
-        assert span["data"][SPANDATA.AI_INPUT_MESSAGES] == json.dumps(messages)
-        assert span["data"][SPANDATA.AI_RESPONSES] == json.dumps(
+        assert span["data"][SPANDATA.AI_INPUT_MESSAGES] == _serialize_span_attribute(messages)
+        assert span["data"][SPANDATA.AI_RESPONSES] == _serialize_span_attribute(
             [{"type": "text", "text": "Hi! I'm Claude!"}]
         )
 
@@ -516,8 +515,8 @@ def test_streaming_create_message_with_input_json_delta(
     assert span["data"][SPANDATA.AI_MODEL_ID] == "model"
 
     if send_default_pii and include_prompts:
-        assert span["data"][SPANDATA.AI_INPUT_MESSAGES] == json.dumps(messages)
-        assert span["data"][SPANDATA.AI_RESPONSES] == json.dumps(
+        assert span["data"][SPANDATA.AI_INPUT_MESSAGES] == _serialize_span_attribute(messages)
+        assert span["data"][SPANDATA.AI_RESPONSES] == _serialize_span_attribute(
             [{"type": "text", "text": ""}]
         )  # we do not record InputJSONDelta because it could contain PII
 
@@ -653,8 +652,8 @@ async def test_streaming_create_message_with_input_json_delta_async(
     assert span["data"][SPANDATA.AI_MODEL_ID] == "model"
 
     if send_default_pii and include_prompts:
-        assert span["data"][SPANDATA.AI_INPUT_MESSAGES] == json.dumps(messages)
-        assert span["data"][SPANDATA.AI_RESPONSES] == json.dumps(
+        assert span["data"][SPANDATA.AI_INPUT_MESSAGES] == _serialize_span_attribute(messages)
+        assert span["data"][SPANDATA.AI_RESPONSES] == _serialize_span_attribute(
             [{"type": "text", "text": ""}]
         )  # we do not record InputJSONDelta because it could contain PII