Function TestTelemetryMiddleware_WrapToolCall

pkg/middleware/telemetry_test.go:199–268

(t *testing.T)

}

func TestTelemetryMiddleware_WrapToolCall(t *testing.T) {
	tracer := telemetry.NewSimpleTracer()

	m := NewTelemetryMiddleware(&TelemetryMiddlewareConfig{
		Tracer:         tracer,
		AgentID:        "test-agent",
		AgentName:      "Test Agent",
		ConversationID: "conv-123",
	})

	ctx := context.Background()
	req := &ToolCallRequest{
		ToolCallID: "call-123",
		ToolName:   "read_file",
		ToolInput: map[string]any{
			"path": "/tmp/test.txt",
		},
	}

	// Mock handler
	handler := func(ctx context.Context, req *ToolCallRequest) (*ToolCallResponse, error) {
		return &ToolCallResponse{
			Result: map[string]any{
				"content": "file contents",
			},
		}, nil
	}

	resp, err := m.WrapToolCall(ctx, req, handler)
	if err != nil {
		t.Fatalf("unexpected error: %v", err)
	}
	if resp == nil {
		t.Fatal("expected response")
	}

	// Verify spans
	spans := tracer.GetSpans()
	if len(spans) != 1 {
		t.Fatalf("expected 1 span, got %d", len(spans))
	}

	span := spans[0]
	if span.Name() != "execute_tool read_file" {
		t.Errorf("expected span name 'execute_tool read_file', got '%s'", span.Name())
	}

	// Verify attributes
	attrs := span.Attributes()
	attrMap := make(map[string]any)
	for _, attr := range attrs {
		attrMap[attr.Key] = attr.Value
	}

	if attrMap[genai.AttrOperationName] != genai.OpExecuteTool {
		t.Errorf("expected operation name '%s', got '%v'", genai.OpExecuteTool, attrMap[genai.AttrOperationName])
	}
	if attrMap[genai.AttrToolName] != "read_file" {