improve coverage for mcp tool call

Author: ygree

dd-java-agent/instrumentation/openai-java/openai-java-3.0/src/test/groovy/OpenAiTest.groovy

@@ -4,6 +4,7 @@ import com.openai.client.OpenAIClient
 import com.openai.client.okhttp.OkHttpClient
 import com.openai.client.okhttp.OpenAIOkHttpClient
 import com.openai.core.ClientOptions
+import com.openai.core.ObjectMappers
 import com.openai.credential.BearerTokenCredential
 import com.openai.core.JsonValue
 import com.openai.models.ChatModel
@@ -24,6 +25,7 @@ import com.openai.models.responses.ResponseInputItem
 import com.openai.models.responses.ResponsePrompt
 import com.openai.models.responses.FunctionTool
 import com.openai.models.responses.CustomTool
+import com.openai.models.responses.Tool
 import com.openai.models.responses.ToolChoiceCustom
 import datadog.trace.agent.test.InstrumentationSpecification
 import datadog.trace.agent.test.server.http.TestHttpServer
@@ -466,6 +468,21 @@ Alice Johnson majors in mathematics at UCLA.""")
     params
   }
 
+  ResponseCreateParams responseCreateParamsWithMcpToolCall() {
+    def mcpTool = Tool.Mcp.builder()
+    .serverLabel("openai_docs")
+    .serverDescription("OpenAI documentation MCP server.")
+    .serverUrl("https://developers.openai.com/mcp")
+    .requireApproval(Tool.Mcp.RequireApproval.McpToolApprovalSetting.NEVER)
+    .build()
+
+    ResponseCreateParams.builder()
+    .model(ChatModel.GPT_5)
+    .input("Use the OpenAI docs MCP server to find the Responses API MCP tool documentation and summarize the require_approval option.")
+    .addTool(mcpTool)
+    .build()
+  }
+
   ChatCompletionCreateParams chatCompletionCreateParamsMultiChoice(boolean json) {
     if (json) {
       ChatCompletionCreateParams.builder()

dd-java-agent/instrumentation/openai-java/openai-java-3.0/src/test/groovy/ResponseServiceTest.groovy

@@ -318,6 +318,31 @@ class ResponseServiceTest extends OpenAiTest {
     toolDefinitions[0].schema.required == ["name"]
   }
 
+  def "create response with mcp tool call"() {
+    Response resp = runUnderTrace("parent") {
+      openAiClient.responses().create(responseCreateParamsWithMcpToolCall())
+    }
+
+    expect:
+    resp != null
+    and:
+    List<LLMObs.LLMMessage> outputTags = []
+    Map<String, Object> metadata = [:]
+    assertResponseTrace(false, "gpt-5", String, null, null, outputTags, metadata)
+    and:
+    metadata.stream == false
+    outputTags.any {
+      it.toolCalls?.any { toolCall ->
+        toolCall.type == "mcp_call" && toolCall.name
+      }
+    }
+    outputTags.any {
+      it.toolResults?.any { toolResult ->
+        toolResult.type == "mcp_tool_result" && toolResult.name
+      }
+    }
+  }
+
   def "create response error sets model_name and placeholder output"() {
     setup:
     def errorBackend = TestHttpServer.httpServer {