91 lines
2.7 KiB
TypeScript
91 lines
2.7 KiB
TypeScript
|
import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
|
||
|
|
import { RegistryCache } from '../../src/registry/cache.js';
|
||
|
|
import type { RegistryServer, SearchOptions } from '../../src/registry/types.js';
|
||
|
|
|
||
|
|
function makeServer(name: string): RegistryServer {
|
||
|
|
return {
|
||
|
|
name,
|
||
|
|
description: `${name} server`,
|
||
|
|
packages: {},
|
||
|
|
envTemplate: [],
|
||
|
|
transport: 'stdio',
|
||
|
|
popularityScore: 0,
|
||
|
|
verified: false,
|
||
|
|
sourceRegistry: 'official',
|
||
|
|
};
|
||
|
|
}
|
||
|
|
|
||
|
|
const defaultOptions: SearchOptions = { query: 'test' };
|
||
|
|
|
||
|
|
describe('RegistryCache', () => {
|
||
|
|
beforeEach(() => {
|
||
|
|
vi.useFakeTimers();
|
||
|
|
});
|
||
|
|
afterEach(() => {
|
||
|
|
vi.useRealTimers();
|
||
|
|
});
|
||
|
|
|
||
|
|
it('returns null for cache miss', () => {
|
||
|
|
const cache = new RegistryCache();
|
||
|
|
expect(cache.get('unknown', defaultOptions)).toBeNull();
|
||
|
|
});
|
||
|
|
|
||
|
|
it('returns data for cache hit within TTL', () => {
|
||
|
|
const cache = new RegistryCache();
|
||
|
|
const data = [makeServer('test')];
|
||
|
|
cache.set('test', defaultOptions, data);
|
||
|
|
expect(cache.get('test', defaultOptions)).toEqual(data);
|
||
|
|
});
|
||
|
|
|
||
|
|
it('returns null after TTL expires', () => {
|
||
|
|
const cache = new RegistryCache(1000); // 1 second TTL
|
||
|
|
cache.set('test', defaultOptions, [makeServer('test')]);
|
||
|
|
|
||
|
|
vi.advanceTimersByTime(1001);
|
||
|
|
expect(cache.get('test', defaultOptions)).toBeNull();
|
||
|
|
});
|
||
|
|
|
||
|
|
it('generates deterministic cache keys', () => {
|
||
|
|
const cache = new RegistryCache();
|
||
|
|
const data = [makeServer('test')];
|
||
|
|
cache.set('query', { query: 'query', limit: 10 }, data);
|
||
|
|
expect(cache.get('query', { query: 'query', limit: 10 })).toEqual(data);
|
||
|
|
});
|
||
|
|
|
||
|
|
it('generates different keys for different queries', () => {
|
||
|
|
const cache = new RegistryCache();
|
||
|
|
cache.set('a', { query: 'a' }, [makeServer('a')]);
|
||
|
|
expect(cache.get('b', { query: 'b' })).toBeNull();
|
||
|
|
});
|
||
|
|
|
||
|
|
it('tracks hits and misses correctly', () => {
|
||
|
|
const cache = new RegistryCache();
|
||
|
|
cache.set('test', defaultOptions, [makeServer('test')]);
|
||
|
|
|
||
|
|
cache.get('test', defaultOptions); // hit
|
||
|
|
cache.get('test', defaultOptions); // hit
|
||
|
|
cache.get('miss', { query: 'miss' }); // miss
|
||
|
|
|
||
|
|
const ratio = cache.getHitRatio();
|
||
|
|
expect(ratio.hits).toBe(2);
|
||
|
|
expect(ratio.misses).toBe(1);
|
||
|
|
expect(ratio.ratio).toBeCloseTo(2 / 3);
|
||
|
|
});
|
||
|
|
|
||
|
|
it('returns 0 ratio when no accesses', () => {
|
||
|
|
const cache = new RegistryCache();
|
||
|
|
expect(cache.getHitRatio().ratio).toBe(0);
|
||
|
|
});
|
||
|
|
|
||
|
|
it('clears all entries and resets metrics', () => {
|
||
|
|
const cache = new RegistryCache();
|
||
|
|
cache.set('a', { query: 'a' }, [makeServer('a')]);
|
||
|
|
cache.get('a', { query: 'a' }); // hit
|
||
|
|
cache.clear();
|
||
|
|
|
||
|
|
expect(cache.get('a', { query: 'a' })).toBeNull();
|
||
|
|
expect(cache.size).toBe(0);
|
||
|
|
expect(cache.getHitRatio().hits).toBe(0);
|
||
|
|
});
|
||
|
|
});
|