91 lines
2.4 KiB
TypeScript
91 lines
2.4 KiB
TypeScript
import { afterEach, describe, expect, it, vi } from 'vitest';
|
|
|
|
import {
|
|
__resetShortLivedProviderPrepareCacheForTests,
|
|
getShortLivedProviderPrepareModelResults,
|
|
storeShortLivedProviderPrepareModelResults,
|
|
} from '@renderer/components/team/dialogs/providerPrepareShortLivedCache';
|
|
|
|
describe('providerPrepareShortLivedCache', () => {
|
|
afterEach(() => {
|
|
__resetShortLivedProviderPrepareCacheForTests();
|
|
vi.useRealTimers();
|
|
});
|
|
|
|
it('stores only successful OpenCode deep verification results', () => {
|
|
storeShortLivedProviderPrepareModelResults({
|
|
providerId: 'opencode',
|
|
cacheKey: 'key-1',
|
|
modelResultsById: {
|
|
'opencode/minimax-m2.5-free': {
|
|
status: 'ready',
|
|
line: 'minimax-m2.5-free - verified',
|
|
warningLine: null,
|
|
},
|
|
'opencode/nemotron-3-super-free': {
|
|
status: 'notes',
|
|
line: 'nemotron-3-super-free - check failed - timed out',
|
|
warningLine: 'nemotron-3-super-free - check failed - timed out',
|
|
},
|
|
},
|
|
});
|
|
|
|
expect(
|
|
getShortLivedProviderPrepareModelResults({
|
|
providerId: 'opencode',
|
|
cacheKey: 'key-1',
|
|
})
|
|
).toEqual({
|
|
'opencode/minimax-m2.5-free': {
|
|
status: 'ready',
|
|
line: 'minimax-m2.5-free - verified',
|
|
warningLine: null,
|
|
},
|
|
});
|
|
});
|
|
|
|
it('expires cached OpenCode results after the short-lived TTL', () => {
|
|
vi.useFakeTimers();
|
|
storeShortLivedProviderPrepareModelResults({
|
|
providerId: 'opencode',
|
|
cacheKey: 'key-2',
|
|
modelResultsById: {
|
|
'opencode/minimax-m2.5-free': {
|
|
status: 'ready',
|
|
line: 'minimax-m2.5-free - verified',
|
|
warningLine: null,
|
|
},
|
|
},
|
|
});
|
|
|
|
vi.advanceTimersByTime(45_001);
|
|
|
|
expect(
|
|
getShortLivedProviderPrepareModelResults({
|
|
providerId: 'opencode',
|
|
cacheKey: 'key-2',
|
|
})
|
|
).toEqual({});
|
|
});
|
|
|
|
it('does not store short-lived cache for non-OpenCode providers', () => {
|
|
storeShortLivedProviderPrepareModelResults({
|
|
providerId: 'codex',
|
|
cacheKey: 'key-3',
|
|
modelResultsById: {
|
|
'gpt-5.4': {
|
|
status: 'ready',
|
|
line: '5.4 - verified',
|
|
warningLine: null,
|
|
},
|
|
},
|
|
});
|
|
|
|
expect(
|
|
getShortLivedProviderPrepareModelResults({
|
|
providerId: 'codex',
|
|
cacheKey: 'key-3',
|
|
})
|
|
).toEqual({});
|
|
});
|
|
});
|