Something went wrong. Try again.
Monorepo for wisp.place. A static site hosting service built on top of the AT Protocol.
Something went wrong. Try again.
18 kB · 578 lines
TypeScript
at main
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579import { describe, expect, test } from 'bun:test'import { CacheManager, estimateJsonBytes } from './cache-manager'
type TestNS = 'ttl' | 'lru' | 'sized' | 'combo'
function createTestCache() { return new CacheManager<TestNS>({ ttl: { ttl: 100, maxEntries: 100 }, lru: { maxEntries: 3 }, sized: { maxEntries: 100, maxSize: 300, estimateSize: (v) => (v as string).length }, combo: { ttl: 100, maxEntries: 3, maxSize: 500, estimateSize: (v) => (v as string).length }, })}
function createDeferred<T>() { let resolve!: (value: T | PromiseLike<T>) => void let reject!: (reason?: unknown) => void const promise = new Promise<T>((resolvePromise, rejectPromise) => { resolve = resolvePromise reject = rejectPromise }) return { promise, resolve, reject }}
describe('CacheManager', () => { describe('get / set basics', () => { test('returns undefined for missing key', () => { const c = createTestCache() expect(c.get('ttl', 'missing')).toBeUndefined() })
test('stores and retrieves a value', () => { const c = createTestCache() c.set('ttl', 'k', 42) expect(c.get<number>('ttl', 'k')).toBe(42) })
test('namespaces are isolated', () => { const c = createTestCache() c.set('ttl', 'k', 'from-ttl') c.set('lru', 'k', 'from-lru') expect(c.get<string>('ttl', 'k')).toBe('from-ttl') expect(c.get<string>('lru', 'k')).toBe('from-lru') })
test('overwrites existing key', () => { const c = createTestCache() c.set('lru', 'k', 'v1') c.set('lru', 'k', 'v2') expect(c.get<string>('lru', 'k')).toBe('v2') }) })
describe('TTL expiry', () => { test('returns value within TTL', () => { const c = createTestCache() c.set('ttl', 'k', 'fresh') expect(c.get<string>('ttl', 'k')).toBe('fresh') })
test('expires stale entries after the safety TTL and releases their stats', async () => { const c = createTestCache() c.set('ttl', 'k', 'stale') await Bun.sleep(150) expect(c.get('ttl', 'k')).toBeUndefined() expect(c.getStats().ttl).toMatchObject({ entries: 0, sizeBytes: 0, misses: 1 }) }) })
describe('LRU eviction by maxEntries', () => { test('evicts oldest entry when maxEntries exceeded', () => { const c = createTestCache() c.set('lru', 'a', 1) c.set('lru', 'b', 2) c.set('lru', 'c', 3) // At capacity (3). Adding a 4th should evict 'a' (oldest). c.set('lru', 'd', 4) expect(c.getStats().lru).toMatchObject({ entries: 3, sizeBytes: 0, evictions: 1 }) expect(c.get<number>('lru', 'a')).toBeUndefined() expect(c.get<number>('lru', 'b')).toBe(2) expect(c.get<number>('lru', 'd')).toBe(4) })
test('accessing a key refreshes its LRU position', () => { const c = createTestCache() c.set('lru', 'a', 1) c.set('lru', 'b', 2) c.set('lru', 'c', 3) // Touch 'a' so it's no longer the oldest c.get('lru', 'a') // Now 'b' is the oldest c.set('lru', 'd', 4) expect(c.get<number>('lru', 'b')).toBeUndefined() expect(c.get<number>('lru', 'a')).toBe(1) }) })
describe('LRU eviction by maxSize', () => { test('evicts entries when maxSize exceeded', () => { const c = createTestCache() // sized ns: maxSize=300, estimateSize = string length c.set('sized', 'a', 'x'.repeat(100)) c.set('sized', 'b', 'x'.repeat(100)) c.set('sized', 'c', 'x'.repeat(100)) // At 300 bytes. Adding 150 more should evict until it fits. c.set('sized', 'd', 'x'.repeat(150)) expect(c.getStats().sized).toMatchObject({ entries: 2, sizeBytes: 250, evictions: 2 }) expect(c.get('sized', 'a')).toBeUndefined() expect(c.get('sized', 'd')).toBeDefined() }) })
describe('delete', () => { test('removes a key', () => { const c = createTestCache() c.set('lru', 'k', 'val') c.delete('lru', 'k') expect(c.get('lru', 'k')).toBeUndefined() })
test('delete on missing key is a no-op', () => { const c = createTestCache() c.delete('lru', 'nonexistent') // should not throw }) })
describe('clear', () => { test('removes all entries in a namespace', () => { const c = createTestCache() c.set('lru', 'a', 1) c.set('lru', 'b', 2) c.set('ttl', 'x', 3) c.clear('lru') expect(c.get<number>('lru', 'a')).toBeUndefined() expect(c.get<number>('lru', 'b')).toBeUndefined() // Other namespace untouched expect(c.get<number>('ttl', 'x')).toBe(3) })
test('clearAll clears every namespace while preserving cumulative stats', () => { const c = createTestCache() c.set('ttl', 'a', 1) c.set('lru', 'b', 2) c.set('sized', 'c', 'size') c.get('lru', 'b')
c.clearAll()
const stats = c.getStats() expect(stats.ttl).toMatchObject({ entries: 0, sizeBytes: 0, hits: 0, misses: 0 }) expect(stats.lru).toMatchObject({ entries: 0, sizeBytes: 0, hits: 1, misses: 0 }) expect(stats.sized).toMatchObject({ entries: 0, sizeBytes: 0, hits: 0, misses: 0 }) })
test('clearAll fences pending fetches so stale completions cannot repopulate a namespace', async () => { const c = createTestCache() const stale = createDeferred<string>() const fresh = createDeferred<string>() let calls = 0 const staleRequest = c.getOrFetch('ttl', 'pending', () => { calls++ return stale.promise })
c.clearAll() const freshRequest = c.getOrFetch('ttl', 'pending', () => { calls++ return fresh.promise }) expect(calls).toBe(2)
stale.resolve('stale') expect(await staleRequest).toBe('stale') expect(c.get('ttl', 'pending')).toBeUndefined() fresh.resolve('fresh') expect(await freshRequest).toBe('fresh') expect(c.get<string>('ttl', 'pending')).toBe('fresh') }) })
describe('getOrFetch', () => { test('calls fetcher on cache miss', async () => { const c = createTestCache() let called = 0 const val = await c.getOrFetch('lru', 'k', () => { called++ return 'fetched' }) expect(val).toBe('fetched') expect(called).toBe(1) })
test('returns cached value without calling fetcher', async () => { const c = createTestCache() c.set('lru', 'k', 'cached') let called = 0 const val = await c.getOrFetch('lru', 'k', () => { called++ return 'fetched' }) expect(val).toBe('cached') expect(called).toBe(0) })
test('works with async fetcher', async () => { const c = createTestCache() const val = await c.getOrFetch('lru', 'k', async () => { await Bun.sleep(5) return 'async-result' }) expect(val).toBe('async-result') // Second call should be from cache expect(c.get<string>('lru', 'k')).toBe('async-result') })
test('shares one in-flight fetch across concurrent cache misses', async () => { const c = createTestCache() const deferred = createDeferred<string>() let calls = 0 const first = c.getOrFetch('lru', 'k', () => { calls++ return deferred.promise }) const second = c.getOrFetch('lru', 'k', () => { calls++ return 'second fetch should not run' })
expect(calls).toBe(1) deferred.resolve('shared') expect(await first).toBe('shared') expect(await second).toBe('shared') expect(c.get<string>('lru', 'k')).toBe('shared') })
test('keeps in-flight fetches isolated by namespace and key', async () => { const c = createTestCache() const firstValue = createDeferred<string>() const secondValue = createDeferred<string>() const thirdValue = createDeferred<string>() let calls = 0 const first = c.getOrFetch('lru', 'same-key', () => { calls++ return firstValue.promise }) const second = c.getOrFetch('lru', 'other-key', () => { calls++ return secondValue.promise }) const third = c.getOrFetch('ttl', 'same-key', () => { calls++ return thirdValue.promise })
expect(calls).toBe(3) firstValue.resolve('first') secondValue.resolve('second') thirdValue.resolve('third') expect(await first).toBe('first') expect(await second).toBe('second') expect(await third).toBe('third') })
test('shares a rejected fetch and removes it so a later request can retry', async () => { const c = createTestCache() const deferred = createDeferred<string>() const error = new Error('fetch failed') let calls = 0 const first = c.getOrFetch('lru', 'k', () => { calls++ return deferred.promise }) const second = c.getOrFetch('lru', 'k', () => { calls++ return 'second fetch should not run' }) const outcomes = Promise.allSettled([first, second])
expect(calls).toBe(1) deferred.reject(error) const [firstOutcome, secondOutcome] = await outcomes expect(firstOutcome).toEqual({ status: 'rejected', reason: error }) expect(secondOutcome).toEqual({ status: 'rejected', reason: error })
expect( await c.getOrFetch('lru', 'k', () => { calls++ return 'retried' }), ).toBe('retried') expect(calls).toBe(2) })
test('delete fences an in-flight fetch and keeps a newer pending fetch joinable', async () => { const c = createTestCache() const stale = createDeferred<string>() const fresh = createDeferred<string>() let calls = 0 const staleRequest = c.getOrFetch('lru', 'k', () => { calls++ return stale.promise })
c.delete('lru', 'k') const freshRequest = c.getOrFetch('lru', 'k', () => { calls++ return fresh.promise }) expect(calls).toBe(2)
stale.resolve('stale') expect(await staleRequest).toBe('stale') expect(c.get('lru', 'k')).toBeUndefined()
const joinedFreshRequest = c.getOrFetch('lru', 'k', () => { calls++ return 'third fetch should not run' }) expect(calls).toBe(2) fresh.resolve('fresh') expect(await freshRequest).toBe('fresh') expect(await joinedFreshRequest).toBe('fresh') expect(c.get<string>('lru', 'k')).toBe('fresh') })
test('deletePrefix fences matching in-flight fetches', async () => { const c = createTestCache() const stale = createDeferred<string>() const fresh = createDeferred<string>() let calls = 0 const staleRequest = c.getOrFetch('lru', 'site:page', () => { calls++ return stale.promise })
c.deletePrefix('lru', 'site:') const freshRequest = c.getOrFetch('lru', 'site:page', () => { calls++ return fresh.promise }) expect(calls).toBe(2)
stale.resolve('stale') expect(await staleRequest).toBe('stale') expect(c.get('lru', 'site:page')).toBeUndefined()
const joinedFreshRequest = c.getOrFetch('lru', 'site:page', () => { calls++ return 'third fetch should not run' }) expect(calls).toBe(2) fresh.resolve('fresh') expect(await freshRequest).toBe('fresh') expect(await joinedFreshRequest).toBe('fresh') expect(c.get<string>('lru', 'site:page')).toBe('fresh') })
test('clear fences all in-flight fetches in a namespace', async () => { const c = createTestCache() const staleFirst = createDeferred<string>() const staleSecond = createDeferred<string>() const fresh = createDeferred<string>() let calls = 0 const firstRequest = c.getOrFetch('lru', 'first', () => { calls++ return staleFirst.promise }) const secondRequest = c.getOrFetch('lru', 'second', () => { calls++ return staleSecond.promise })
c.clear('lru') const freshRequest = c.getOrFetch('lru', 'first', () => { calls++ return fresh.promise }) expect(calls).toBe(3)
staleFirst.resolve('stale first') staleSecond.resolve('stale second') expect(await firstRequest).toBe('stale first') expect(await secondRequest).toBe('stale second') expect(c.get('lru', 'first')).toBeUndefined() expect(c.get('lru', 'second')).toBeUndefined()
const joinedFreshRequest = c.getOrFetch('lru', 'first', () => { calls++ return 'fourth fetch should not run' }) expect(calls).toBe(3) fresh.resolve('fresh') expect(await freshRequest).toBe('fresh') expect(await joinedFreshRequest).toBe('fresh') expect(c.get<string>('lru', 'first')).toBe('fresh') })
test('explicit set fences an in-flight fetch without letting it overwrite the set value', async () => { const c = createTestCache() const stale = createDeferred<string>() let calls = 0 const staleRequest = c.getOrFetch('lru', 'k', () => { calls++ return stale.promise })
c.set('lru', 'k', 'explicit') expect( await c.getOrFetch('lru', 'k', () => { calls++ return 'fetch should not run' }), ).toBe('explicit') expect(calls).toBe(1)
stale.resolve('stale') expect(await staleRequest).toBe('stale') expect(c.get<string>('lru', 'k')).toBe('explicit') })
test('cacheIf: false skips caching', async () => { const c = createTestCache() const val = await c.getOrFetch('lru', 'k', () => null, { cacheIf: (v) => v !== null, }) expect(val).toBeNull() // Should NOT be cached expect(c.get('lru', 'k')).toBeUndefined() })
test('cacheIf: true caches normally', async () => { const c = createTestCache() const val = await c.getOrFetch('lru', 'k', () => 'good', { cacheIf: (v) => v !== null, }) expect(val).toBe('good') expect(c.get<string>('lru', 'k')).toBe('good') }) })
describe('stats', () => { test('tracks hits and misses', () => { const c = createTestCache() c.get('lru', 'miss1') c.get('lru', 'miss2') c.set('lru', 'k', 'v') c.get('lru', 'k') const stats = c.getStats() expect(stats.lru.misses).toBe(2) expect(stats.lru.hits).toBe(1) })
test('tracks evictions', () => { const c = createTestCache() c.set('lru', 'a', 1) c.set('lru', 'b', 2) c.set('lru', 'c', 3) c.set('lru', 'd', 4) // evicts 'a' const stats = c.getStats() expect(stats.lru.evictions).toBe(1) })
test('tracks entries count', () => { const c = createTestCache() c.set('lru', 'a', 1) c.set('lru', 'b', 2) expect(c.getStats().lru.entries).toBe(2) c.delete('lru', 'a') expect(c.getStats().lru.entries).toBe(1) })
test('tracks sizeBytes with estimateSize', () => { const c = createTestCache() c.set('sized', 'a', 'hello') // 5 bytes c.set('sized', 'b', 'world!') // 6 bytes expect(c.getStats().sized.sizeBytes).toBe(11) c.delete('sized', 'a') expect(c.getStats().sized.sizeBytes).toBe(6) })
test('does not retain a value larger than the namespace byte cap', () => { const c = createTestCache() c.set('sized', 'key', 'small') c.set('sized', 'key', 'x'.repeat(301))
expect(c.get('sized', 'key')).toBeUndefined() expect(c.getStats().sized).toMatchObject({ entries: 0, sizeBytes: 0 }) })
test('returns independent stats per namespace', () => { const c = createTestCache() c.set('ttl', 'a', 1) c.set('lru', 'b', 2) const stats = c.getStats() expect(stats.ttl.entries).toBe(1) expect(stats.lru.entries).toBe(1) expect(stats.sized.entries).toBe(0) }) })
describe('cleanup', () => { test('startCleanup / stopCleanup do not throw', () => { const c = createTestCache() c.startCleanup(50) c.stopCleanup() })
test('cleanup sweeps expired TTL entries', async () => { const c = createTestCache() c.set('ttl', 'k', 'val') c.startCleanup(50) // Wait for TTL (100ms) + cleanup interval (50ms) + buffer await Bun.sleep(200) c.stopCleanup() expect(c.get('ttl', 'k')).toBeUndefined() })
test('cleanup does not touch non-TTL namespaces', async () => { const c = createTestCache() c.set('lru', 'k', 'val') c.startCleanup(50) await Bun.sleep(200) c.stopCleanup() expect(c.get<string>('lru', 'k')).toBe('val') }) })})
function manifestRow(files: number) { const fileCids: Record<string, string> = {} const fileObjects: Record<string, string> = {} for (let index = 0; index < files; index++) { const path = `posts/post-${index}/index.html` fileCids[path] = `bafkreiexample${index}`.padEnd(59, 'a') fileObjects[path] = `cas/bafkreiexample${index}.0123abcd.html` } return { did: 'did:plc:example', rkey: 'site', file_cids: fileCids, file_objects: fileObjects }}
describe('estimateJsonBytes', () => { test('grows with every path, CID and CAS key a manifest row holds', () => { const small = estimateJsonBytes(manifestRow(10)) const large = estimateJsonBytes(manifestRow(1000)) // Each path appears in both maps: two keys, a CID and a CAS key. const perFile = (large - small) / 990 expect(perFile).toBeGreaterThan(2 * 'posts/post-100/index.html'.length + 59 + 33) expect(perFile).toBeLessThan(1024) })
test('counts arrays, nested values and scalars', () => { expect(estimateJsonBytes([['a.html', 'cid']])).toBeGreaterThan(estimateJsonBytes([])) expect(estimateJsonBytes({ a: null, b: 1, c: true })).toBeGreaterThan(estimateJsonBytes({})) expect(estimateJsonBytes('x'.repeat(100))).toBeGreaterThanOrEqual(100) })})
describe('site manifest cache budget', () => { test('evicts the oldest manifests once their estimated bytes exceed MANIFEST_CACHE_SIZE', async () => { const previous = process.env.MANIFEST_CACHE_SIZE const row = manifestRow(200) const rowBytes = estimateJsonBytes(row) process.env.MANIFEST_CACHE_SIZE = String(rowBytes * 3) try { const budgetedModule = './cache-manager?manifest-budget' const { cache } = (await import(budgetedModule)) as typeof import('./cache-manager') for (let site = 0; site < 10; site++) cache.set('siteCache', `did:plc:example:site-${site}`, manifestRow(200))
const stats = cache.getStats().siteCache expect(stats.entries).toBe(3) expect(stats.sizeBytes).toBeLessThanOrEqual(rowBytes * 3) expect(cache.get('siteCache', 'did:plc:example:site-0')).toBeUndefined() expect(cache.get('siteCache', 'did:plc:example:site-9')).toBeDefined() } finally { if (previous === undefined) delete process.env.MANIFEST_CACHE_SIZE else process.env.MANIFEST_CACHE_SIZE = previous } })})