File size: 3,397 Bytes
bf48b89 | 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 92 93 94 95 96 97 98 99 100 101 | import { afterAll, afterEach, beforeAll, describe, expect, it, vi } from 'vitest';
import wait from '@/utils/wait';
beforeAll(() => {
process.env.CACHE_EXPIRE = '1';
process.env.CACHE_CONTENT_EXPIRE = '3';
});
afterEach(() => {
delete process.env.CACHE_TYPE;
vi.resetModules();
});
afterAll(() => {
delete process.env.CACHE_EXPIRE;
});
describe('cache', () => {
it('memory', async () => {
process.env.CACHE_TYPE = 'memory';
const cache = (await import('@/utils/cache')).default;
if (!cache.clients.memoryCache || !cache.status.available) {
throw new Error('Memory cache client error');
}
await cache.set('mock', undefined);
expect(await cache.get('mock')).toBe('');
await cache.globalCache.set('mock', undefined);
expect(await cache.globalCache.get('mock')).toBe('');
await cache.globalCache.set('mock', {
mock: 1,
});
expect(await cache.globalCache.get('mock')).toBe('{"mock":1}');
}, 10000);
it('memory get returns null before init', async () => {
const memory = (await import('@/utils/cache/memory')).default;
expect(await memory.get('missing')).toBeNull();
});
it('redis', async () => {
process.env.CACHE_TYPE = 'redis';
const cache = (await import('@/utils/cache')).default;
await wait(500);
if (!cache.clients.redisClient || !cache.status.available) {
throw new Error('Redis client error');
}
await cache.set('mock1', undefined);
expect(await cache.get('mock1')).toBe('');
await cache.set('mock2', '2');
await cache.set('mock2', '2');
expect(await cache.get('mock2')).toBe('2');
await cache.clients.redisClient?.quit();
}, 10000);
it('redis with quit', async () => {
process.env.CACHE_TYPE = 'redis';
const cache = (await import('@/utils/cache')).default;
if (cache.clients.redisClient) {
await cache.clients.redisClient.quit();
} else {
throw new Error('No redis client');
}
await cache.set('mock2', '2');
expect(await cache.get('mock2')).toBe(null);
});
it('redis with error', async () => {
process.env.CACHE_TYPE = 'redis';
process.env.REDIS_URL = 'redis://wrongpath:6379';
const cache = (await import('@/utils/cache')).default;
await cache.set('mock2', '2');
expect(await cache.get('mock2')).toBe(null);
cache.clients.redisClient?.disconnect();
});
it('no cache', async () => {
process.env.CACHE_TYPE = 'NO';
const cache = (await import('@/utils/cache')).default;
await cache.init();
await cache.set('mock2', '2');
expect(await cache.get('mock2')).toBe(null);
expect(await cache.globalCache.get('mock2')).toBeNull();
expect(cache.globalCache.set('mock2', '2')).toBeNull();
});
it('throws TTL key', async () => {
process.env.CACHE_TYPE = 'redis';
const cache = (await import('@/utils/cache')).default;
try {
await cache.get('rsshub:cacheTtl:mock');
} catch (error: any) {
expect(error.message).toContain('reserved for the internal usage');
} finally {
await cache.clients.redisClient?.quit();
}
});
});
|