Add tests for initialization span tracing

Author: jboolean

contrib/mark3labs/mcp-go/hooks_test.go

@@ -87,6 +87,35 @@ func TestIntegrationSessionInitialize(t *testing.T) {
 	assert.Equal(t, "2.0", resp["jsonrpc"])
 	assert.Equal(t, float64(1), resp["id"])
 	assert.NotNil(t, resp["result"])
+
+	// Verify LLMObs task span was created for initialization
+	spans := tt.WaitForLLMObsSpans(t, 1)
+	require.Len(t, spans, 1)
+
+	taskSpan := spans[0]
+	assert.Equal(t, "mcp.initialize", taskSpan.Name)
+	assert.Equal(t, "task", taskSpan.Meta["span.kind"])
+
+	// Verify input/output annotations exist
+	assert.Contains(t, taskSpan.Meta, "input")
+	assert.Contains(t, taskSpan.Meta, "output")
+
+	// Verify input contains the initialize request data
+	inputMeta := taskSpan.Meta["input"]
+	assert.NotNil(t, inputMeta)
+	inputJSON, err := json.Marshal(inputMeta)
+	require.NoError(t, err)
+	inputStr := string(inputJSON)
+	assert.Contains(t, inputStr, "2024-11-05") // protocol version
+	assert.Contains(t, inputStr, "test-client")
+
+	// Verify output contains the result
+	outputMeta := taskSpan.Meta["output"]
+	assert.NotNil(t, outputMeta)
+	outputJSON, err := json.Marshal(outputMeta)
+	require.NoError(t, err)
+	outputStr := string(outputJSON)
+	assert.Contains(t, outputStr, "serverInfo")
 }
 
 func TestIntegrationToolCallSuccess(t *testing.T) {