Spaces:
Running
Running
File size: 2,426 Bytes
b034029 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 | import { describe, expect, it } from 'vitest';
import { buildStatCardMetrics } from './StatCards';
import type { UsagePayload } from './hooks/useUsageData';
const usageWithBackendSummary: UsagePayload = {
total_requests: 9,
success_count: 8,
failure_count: 1,
total_tokens: 900,
requests_by_day: {},
requests_by_hour: {},
tokens_by_day: {},
tokens_by_hour: {},
apis: {
'provider-a': {
total_requests: 1,
success_count: 1,
failure_count: 0,
total_tokens: 100,
models: {
'claude-sonnet': {
total_requests: 1,
success_count: 1,
failure_count: 0,
total_tokens: 100,
details: [
{
timestamp: '2026-04-23T00:00:00.000Z',
latency_ms: 100,
source: 'source-a',
auth_index: '1',
failed: false,
tokens: {
input_tokens: 20,
output_tokens: 30,
reasoning_tokens: 4,
cached_tokens: 5,
total_tokens: 50,
},
},
],
},
},
},
},
summary: {
request_count: 3,
token_count: 777,
window_minutes: 120,
rpm: 0.025,
tpm: 6.475,
total_cost: 1.234,
cost_available: true,
cached_tokens: 22,
reasoning_tokens: 33,
},
};
describe('buildStatCardMetrics', () => {
it('prefers backend summary values over detail-derived metrics', () => {
const metrics = buildStatCardMetrics({
usage: usageWithBackendSummary,
});
expect(metrics.rateStats.requestCount).toBe(3);
expect(metrics.rateStats.tokenCount).toBe(777);
expect(metrics.rateStats.windowMinutes).toBe(120);
expect(metrics.rateStats.rpm).toBe(0.025);
expect(metrics.rateStats.tpm).toBe(6.475);
expect(metrics.tokenBreakdown.cachedTokens).toBe(22);
expect(metrics.tokenBreakdown.reasoningTokens).toBe(33);
expect(metrics.totalCost).toBe(1.234);
});
it('keeps priced total cost visible when availability is partial', () => {
const metrics = buildStatCardMetrics({
usage: {
...usageWithBackendSummary,
summary: {
...usageWithBackendSummary.summary!,
total_cost: 4.56,
cost_available: false,
},
},
});
expect(metrics.totalCost).toBe(4.56);
expect(metrics.costAvailable).toBe(false);
});
});
|