Function TestComputeTokenUsageDiff_WithData

pkg/cli/audit_diff_test.go:736–786

(t *testing.T)

}

func TestComputeTokenUsageDiff_WithData(t *testing.T) {
	tu1 := &TokenUsageSummary{
		TotalInputTokens:      10000,
		TotalOutputTokens:     2000,
		TotalCacheReadTokens:  5000,
		TotalCacheWriteTokens: 1000,
		TotalAIC:              0.8,
		TotalRequests:         10,
		CacheEfficiency:       0.333,
	}
	tu2 := &TokenUsageSummary{
		TotalInputTokens:      15000,
		TotalOutputTokens:     3000,
		TotalCacheReadTokens:  7000,
		TotalCacheWriteTokens: 800,
		TotalAIC:              1.4,
		TotalRequests:         14,
		CacheEfficiency:       0.318,
	}

	diff := computeTokenUsageDiff(tu1, tu2)

	require.NotNil(t, diff, "Should produce token usage diff when data is available")
	assert.Equal(t, 10000, diff.Run1InputTokens, "Run1 input tokens should be 10000")
	assert.Equal(t, 15000, diff.Run2InputTokens, "Run2 input tokens should be 15000")
	assert.Equal(t, "+50%", diff.InputTokensChange, "Input tokens should increase by 50%")

	assert.Equal(t, 2000, diff.Run1OutputTokens, "Run1 output tokens should be 2000")
	assert.Equal(t, 3000, diff.Run2OutputTokens, "Run2 output tokens should be 3000")
	assert.Equal(t, "+50%", diff.OutputTokensChange, "Output tokens should increase by 50%")

	assert.Equal(t, 5000, diff.Run1CacheReadTokens, "Run1 cache read tokens should be 5000")
	assert.Equal(t, 7000, diff.Run2CacheReadTokens, "Run2 cache read tokens should be 7000")
	assert.Equal(t, "+40%", diff.CacheReadTokensChange, "Cache read tokens should increase by 40%")

	assert.Equal(t, 1000, diff.Run1CacheWriteTokens, "Run1 cache write tokens should be 1000")
	assert.Equal(t, 800, diff.Run2CacheWriteTokens, "Run2 cache write tokens should be 800")
	assert.Equal(t, "-20%", diff.CacheWriteTokensChange, "Cache write tokens should decrease by 20%")

	assert.InDelta(t, 0.8, diff.Run1AIC, 1e-9, "Run1 AI Credits should be 0.8")
	assert.InDelta(t, 1.4, diff.Run2AIC, 1e-9, "Run2 AI Credits should be 1.4")
	assert.Equal(t, "+0.600", diff.AICChange, "AI Credits delta should be +0.600")

	assert.Equal(t, 10, diff.Run1TotalRequests, "Run1 requests should be 10")
	assert.Equal(t, 14, diff.Run2TotalRequests, "Run2 requests should be 14")
	assert.Equal(t, "+4", diff.RequestsDelta, "Requests delta should be +4")

	assert.InDelta(t, 0.333, diff.Run1CacheEfficiency, 0.001, "Run1 cache efficiency should match")
	assert.InDelta(t, 0.318, diff.Run2CacheEfficiency, 0.001, "Run2 cache efficiency should match")
	assert.Equal(t, "-1.5pp", diff.CacheEfficiencyChange, "Cache efficiency change should be -1.5pp")
}

func TestComputeTokenUsageDiff_Run1Nil(t *testing.T) {
	tu2 := &TokenUsageSummary{