Function TestNewTelemetryMiddleware

pkg/middleware/telemetry_test.go:13–59

(t *testing.T)

)

func TestNewTelemetryMiddleware(t *testing.T) {
	tests := []struct {
		name   string
		config *TelemetryMiddlewareConfig
	}{
		{
			name:   "nil config",
			config: nil,
		},
		{
			name: "basic config",
			config: &TelemetryMiddlewareConfig{
				AgentID:   "test-agent",
				AgentName: "Test Agent",
				Provider:  genai.ProviderAnthropic,
				Model:     "claude-3-5-sonnet",
			},
		},
		{
			name: "full config",
			config: &TelemetryMiddlewareConfig{
				AgentID:           "test-agent",
				AgentName:         "Test Agent",
				Provider:          genai.ProviderOpenAI,
				Model:             "gpt-4",
				ConversationID:    "conv-123",
				RecordPrompts:     true,
				RecordCompletions: true,
			},
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			m := NewTelemetryMiddleware(tt.config)
			if m == nil {
				t.Fatal("expected middleware to be created")
			}
			if m.Name() != "telemetry" {
				t.Errorf("expected name 'telemetry', got '%s'", m.Name())
			}
			if m.Priority() != 5 {
				t.Errorf("expected priority 5, got %d", m.Priority())
			}
		})
	}
}

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