remove tests of private details

Author: alexander-alderman-webb

tests/integrations/openai/test_openai.py

@@ -44,7 +44,6 @@
 from sentry_sdk.consts import SPANDATA, OP
 from sentry_sdk.integrations.openai import (
     OpenAIIntegration,
-    _calculate_token_usage,
 )
 from sentry_sdk.utils import safe_serialize
 
@@ -1780,162 +1779,6 @@ async def test_span_origin_embeddings_async(sentry_init, capture_events):
     assert event["spans"][0]["origin"] == "auto.ai.openai"
 
 
-def test_calculate_token_usage_a():
-    span = mock.MagicMock()
-
-    def count_tokens(msg):
-        return len(str(msg))
-
-    response = mock.MagicMock()
-    response.usage = mock.MagicMock()
-    response.usage.completion_tokens = 10
-    response.usage.prompt_tokens = 20
-    response.usage.total_tokens = 30
-    messages = []
-    streaming_message_responses = []
-
-    with mock.patch(
-        "sentry_sdk.integrations.openai.record_token_usage"
-    ) as mock_record_token_usage:
-        _calculate_token_usage(
-            messages, response, span, streaming_message_responses, count_tokens
-        )
-        mock_record_token_usage.assert_called_once_with(
-            span,
-            input_tokens=20,
-            input_tokens_cached=None,
-            output_tokens=10,
-            output_tokens_reasoning=None,
-            total_tokens=30,
-        )
-
-
-def test_calculate_token_usage_b():
-    span = mock.MagicMock()
-
-    def count_tokens(msg):
-        return len(str(msg))
-
-    response = mock.MagicMock()
-    response.usage = mock.MagicMock()
-    response.usage.completion_tokens = 10
-    response.usage.total_tokens = 10
-    messages = [
-        {"content": "one"},
-        {"content": "two"},
-        {"content": "three"},
-    ]
-    streaming_message_responses = []
-
-    with mock.patch(
-        "sentry_sdk.integrations.openai.record_token_usage"
-    ) as mock_record_token_usage:
-        _calculate_token_usage(
-            messages, response, span, streaming_message_responses, count_tokens
-        )
-        mock_record_token_usage.assert_called_once_with(
-            span,
-            input_tokens=11,
-            input_tokens_cached=None,
-            output_tokens=10,
-            output_tokens_reasoning=None,
-            total_tokens=10,
-        )
-
-
-def test_calculate_token_usage_c():
-    span = mock.MagicMock()
-
-    def count_tokens(msg):
-        return len(str(msg))
-
-    response = mock.MagicMock()
-    response.usage = mock.MagicMock()
-    response.usage.prompt_tokens = 20
-    response.usage.total_tokens = 20
-    messages = []
-    streaming_message_responses = [
-        "one",
-        "two",
-        "three",
-    ]
-
-    with mock.patch(
-        "sentry_sdk.integrations.openai.record_token_usage"
-    ) as mock_record_token_usage:
-        _calculate_token_usage(
-            messages, response, span, streaming_message_responses, count_tokens
-        )
-        mock_record_token_usage.assert_called_once_with(
-            span,
-            input_tokens=20,
-            input_tokens_cached=None,
-            output_tokens=11,
-            output_tokens_reasoning=None,
-            total_tokens=20,
-        )
-
-
-def test_calculate_token_usage_d():
-    span = mock.MagicMock()
-
-    def count_tokens(msg):
-        return len(str(msg))
-
-    response = mock.MagicMock()
-    response.usage = mock.MagicMock()
-    response.usage.prompt_tokens = 20
-    response.usage.total_tokens = 20
-    response.choices = [
-        mock.MagicMock(message="one"),
-        mock.MagicMock(message="two"),
-        mock.MagicMock(message="three"),
-    ]
-    messages = []
-    streaming_message_responses = []
-
-    with mock.patch(
-        "sentry_sdk.integrations.openai.record_token_usage"
-    ) as mock_record_token_usage:
-        _calculate_token_usage(
-            messages, response, span, streaming_message_responses, count_tokens
-        )
-        mock_record_token_usage.assert_called_once_with(
-            span,
-            input_tokens=20,
-            input_tokens_cached=None,
-            output_tokens=None,
-            output_tokens_reasoning=None,
-            total_tokens=20,
-        )
-
-
-def test_calculate_token_usage_e():
-    span = mock.MagicMock()
-
-    def count_tokens(msg):
-        return len(str(msg))
-
-    response = mock.MagicMock()
-    messages = []
-    streaming_message_responses = None
-
-    with mock.patch(
-        "sentry_sdk.integrations.openai.record_token_usage"
-    ) as mock_record_token_usage:
-        _calculate_token_usage(
-            messages, response, span, streaming_message_responses, count_tokens
-        )
-        mock_record_token_usage.assert_called_once_with(
-            span,
-            input_tokens=None,
-            input_tokens_cached=None,
-            output_tokens=None,
-            output_tokens_reasoning=None,
-            total_tokens=None,
-        )
-
-
 @pytest.mark.skipif(SKIP_RESPONSES_TESTS, reason="Responses API not available")
 def test_ai_client_span_responses_api_no_pii(sentry_init, capture_events):
     sentry_init(