unleash.unleash/frontend/src/hooks/useClearSWRCache.test.ts

import { clearCacheEntries } from './useClearSWRCache.js';

describe('manageCacheEntries', () => {
    it('should clear old cache entries and keep the current one when SWR_CACHE_SIZE is not provided', () => {
        const cacheMock = new Map();
        cacheMock.set('prefix-1', {});
        cacheMock.set('prefix-2', {});
        cacheMock.set('prefix-3', {});

        clearCacheEntries({
            cache: cacheMock,
            currentKey: 'prefix-3',
            clearPrefix: 'prefix-',
        });

        expect(cacheMock.has('prefix-1')).toBe(false);
        expect(cacheMock.has('prefix-2')).toBe(false);
        expect(cacheMock.has('prefix-3')).toBe(true);
    });

    it('should keep the SWR_CACHE_SIZE entries and delete the rest, keeping the most recent entries', () => {
        const cacheMock = new Map();
        cacheMock.set('prefix-1', {});
        cacheMock.set('prefix-2', {});
        cacheMock.set('prefix-3', {});
        cacheMock.set('prefix-4', {});

        clearCacheEntries({
            cache: cacheMock,
            currentKey: 'prefix-4',
            clearPrefix: 'prefix-',
            cacheSize: 2,
        });

        expect([...cacheMock.keys()]).toStrictEqual(['prefix-3', 'prefix-4']);
    });

    it('should not delete the current key, even if it would be cleared as part of the cache size limit', () => {
        const cacheMock = new Map();
        cacheMock.set('prefix-1', {});
        cacheMock.set('prefix-2', {});
        cacheMock.set('prefix-3', {});
        cacheMock.set('prefix-4', {});

        clearCacheEntries({
            cache: cacheMock,
            currentKey: 'prefix-2',
            clearPrefix: 'prefix-',
            cacheSize: 2,
        });

        expect([...cacheMock.keys()]).toStrictEqual(['prefix-2', 'prefix-4']);
    });

    it('should handle case when SWR_CACHE_SIZE is larger than number of entries', () => {
        const cacheMock = new Map();
        cacheMock.set('prefix-1', {});
        cacheMock.set('prefix-2', {});

        clearCacheEntries({
            cache: cacheMock,
            currentKey: 'prefix-2',
            clearPrefix: 'prefix-',
            cacheSize: 5,
        });

        expect(cacheMock.has('prefix-1')).toBe(true);
        expect(cacheMock.has('prefix-2')).toBe(true);
    });

    it('should not delete entries that do not match the prefix', () => {
        const cacheMock = new Map();
        cacheMock.set('prefix-1', {});
        cacheMock.set('other-2', {});
        cacheMock.set('prefix-3', {});

        clearCacheEntries({
            cache: cacheMock,
            currentKey: 'prefix-3',
            clearPrefix: 'prefix-',
            cacheSize: 2,
        });

        expect(cacheMock.has('prefix-1')).toBe(true);
        expect(cacheMock.has('other-2')).toBe(true);
        expect(cacheMock.has('prefix-3')).toBe(true);
    });

    it('treats a negative cache size as zero', () => {
        const cacheMock = new Map();
        cacheMock.set('prefix-1', {});
        cacheMock.set('prefix-2', {});
        cacheMock.set('prefix-3', {});
        cacheMock.set('prefix-4', {});

        clearCacheEntries({
            cache: cacheMock,
            currentKey: 'prefix-3',
            clearPrefix: 'prefix-',
            cacheSize: -1,
        });

        expect([...cacheMock.keys()]).toStrictEqual(['prefix-3']);
    });
});