Skip to content

Commit f2786c1

Browse files
committed
fix tests for exact date benchmark query param
1 parent bfb148e commit f2786c1

2 files changed

Lines changed: 23 additions & 3 deletions

File tree

packages/app/src/app/api/v1/benchmarks/route.test.ts

Lines changed: 17 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -52,15 +52,30 @@ describe('GET /api/v1/benchmarks', () => {
5252
expect(res.status).toBe(200);
5353
const body = await res.json();
5454
expect(body).toEqual(mockRows);
55-
expect(mockGetLatestBenchmarks).toHaveBeenCalledWith('mock-sql', 'dsr1', undefined);
55+
expect(mockGetLatestBenchmarks).toHaveBeenCalledWith('mock-sql', 'dsr1', undefined, undefined);
5656
});
5757

5858
it('passes date param to query when provided', async () => {
5959
mockGetLatestBenchmarks.mockResolvedValueOnce([]);
6060

6161
const res = await GET(req('/api/v1/benchmarks?model=DeepSeek-R1-0528&date=2026-03-01'));
6262
expect(res.status).toBe(200);
63-
expect(mockGetLatestBenchmarks).toHaveBeenCalledWith('mock-sql', 'dsr1', '2026-03-01');
63+
expect(mockGetLatestBenchmarks).toHaveBeenCalledWith(
64+
'mock-sql',
65+
'dsr1',
66+
'2026-03-01',
67+
undefined,
68+
);
69+
});
70+
71+
it('passes exact=true when query param set', async () => {
72+
mockGetLatestBenchmarks.mockResolvedValueOnce([]);
73+
74+
const res = await GET(
75+
req('/api/v1/benchmarks?model=DeepSeek-R1-0528&date=2026-03-01&exact=true'),
76+
);
77+
expect(res.status).toBe(200);
78+
expect(mockGetLatestBenchmarks).toHaveBeenCalledWith('mock-sql', 'dsr1', '2026-03-01', true);
6479
});
6580

6681
it('returns 500 when query throws', async () => {

packages/app/src/hooks/api/use-benchmarks.test.ts

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,12 @@ import { benchmarkQueryOptions } from '@/hooks/api/use-benchmarks';
55
describe('benchmarkQueryOptions', () => {
66
it('builds query key from model and date', () => {
77
const opts = benchmarkQueryOptions('DeepSeek-R1-0528', '2026-03-01');
8-
expect(opts.queryKey).toEqual(['benchmarks', 'DeepSeek-R1-0528', '2026-03-01']);
8+
expect(opts.queryKey).toEqual(['benchmarks', 'DeepSeek-R1-0528', '2026-03-01', 'latest']);
9+
});
10+
11+
it('builds exact query key when exact=true', () => {
12+
const opts = benchmarkQueryOptions('DeepSeek-R1-0528', '2026-03-01', true, true);
13+
expect(opts.queryKey).toEqual(['benchmarks', 'DeepSeek-R1-0528', '2026-03-01', 'exact']);
914
});
1015

1116
it('produces distinct keys for different models', () => {

0 commit comments

Comments
 (0)