Function TestGeneratePerformanceMetrics

pkg/cli/audit_report_test.go:555–653

(t *testing.T)

}

func TestGeneratePerformanceMetrics(t *testing.T) {
	tests := []struct {
		name         string
		processedRun ProcessedRun
		metrics      MetricsData
		toolUsage    []ToolUsageInfo
		checkMetrics func(t *testing.T, pm *PerformanceMetrics)
	}{
		{
			name: "tokens per minute calculation",
			processedRun: func() ProcessedRun {
				pr := createTestProcessedRun()
				pr.Run.Duration = 2 * time.Minute
				return pr
			}(),
			metrics: MetricsData{
				TokenUsage: 1000,
			},
			toolUsage: []ToolUsageInfo{},
			checkMetrics: func(t *testing.T, pm *PerformanceMetrics) {
				assert.InDelta(t, 500.0, pm.TokensPerMinute, 0.01,
					"Tokens per minute should be 1000 tokens / 2 minutes = 500")
			},
		},
		{
			name:         "most used tool",
			processedRun: createTestProcessedRun(),
			metrics:      MetricsData{},
			toolUsage: []ToolUsageInfo{
				{Name: "bash", CallCount: 5},
				{Name: "github_issue_read", CallCount: 10},
				{Name: "file_edit", CallCount: 3},
			},
			checkMetrics: func(t *testing.T, pm *PerformanceMetrics) {
				assert.Contains(t, pm.MostUsedTool, "github_issue_read",
					"Most used tool should be github_issue_read")
				assert.Contains(t, pm.MostUsedTool, "10 calls",
					"Most used tool should show 10 calls")
			},
		},
		{
			name:         "average tool duration",
			processedRun: createTestProcessedRun(),
			metrics:      MetricsData{},
			toolUsage: []ToolUsageInfo{
				{Name: "bash", CallCount: 5, MaxDuration: "1s"},
				{Name: "github_issue_read", CallCount: 10, MaxDuration: "3s"},
			},
			checkMetrics: func(t *testing.T, pm *PerformanceMetrics) {
				assert.NotEmpty(t, pm.AvgToolDuration,
					"Average tool duration should be calculated")
			},
		},
		{
			name: "network requests from firewall",
			processedRun: func() ProcessedRun {
				pr := createTestProcessedRun()
				pr.FirewallAnalysis = &FirewallAnalysis{