Function TestComputeAuditDiff_CombinesAllSections

pkg/cli/audit_diff_test.go:580–627

(t *testing.T)

}

func TestComputeAuditDiff_CombinesAllSections(t *testing.T) {
	summary1 := &RunSummary{
		RunID: 100,
		FirewallAnalysis: &FirewallAnalysis{
			RequestsByDomain: map[string]DomainRequestStats{
				"api.github.com:443": {Allowed: 5, Blocked: 0},
			},
		},
		MCPToolUsage: &MCPToolUsageData{
			Summary: []MCPToolSummary{
				{ServerName: "github", ToolName: "issue_read", CallCount: 3, ErrorCount: 0},
			},
		},
		Run: WorkflowRun{TokenUsage: 2000, Turns: 5},
	}
	summary2 := &RunSummary{
		RunID: 200,
		FirewallAnalysis: &FirewallAnalysis{
			RequestsByDomain: map[string]DomainRequestStats{
				"api.github.com:443":  {Allowed: 5, Blocked: 0},
				"new.example.com:443": {Allowed: 3, Blocked: 0},
			},
		},
		MCPToolUsage: &MCPToolUsageData{
			Summary: []MCPToolSummary{
				{ServerName: "github", ToolName: "issue_read", CallCount: 7, ErrorCount: 0},
				{ServerName: "github", ToolName: "create_issue", CallCount: 2, ErrorCount: 0},
			},
		},
		Run: WorkflowRun{TokenUsage: 3000, Turns: 8},
	}

	diff := computeAuditDiff(100, 200, summary1, summary2)

	assert.Equal(t, int64(100), diff.Run1ID, "Run1ID should match")
	assert.Equal(t, int64(200), diff.Run2ID, "Run2ID should match")

	require.NotNil(t, diff.FirewallDiff, "Should have firewall diff")
	assert.Len(t, diff.FirewallDiff.NewDomains, 1, "Should have 1 new domain")

	require.NotNil(t, diff.MCPToolsDiff, "Should have MCP tools diff")
	assert.Len(t, diff.MCPToolsDiff.NewTools, 1, "Should have 1 new tool")
	assert.Len(t, diff.MCPToolsDiff.ChangedTools, 1, "Should have 1 changed tool")

	require.NotNil(t, diff.RunMetricsDiff, "Should have run metrics diff")
	assert.Equal(t, 2000, diff.RunMetricsDiff.Run1TokenUsage, "Run1 token usage should match")
	assert.Equal(t, 3000, diff.RunMetricsDiff.Run2TokenUsage, "Run2 token usage should match")
}

func TestComputeAuditDiff_NilSummaries(t *testing.T) {
	diff := computeAuditDiff(100, 200, nil, nil)