diff --git a/packages/test/test-end-to-end-tests/src/testPersistedCache.ts b/packages/test/test-end-to-end-tests/src/testPersistedCache.ts index 992c13603481..d31ce255bc0e 100644 --- a/packages/test/test-end-to-end-tests/src/testPersistedCache.ts +++ b/packages/test/test-end-to-end-tests/src/testPersistedCache.ts @@ -16,16 +16,19 @@ export interface ValueWithSnapshot { } export class TestPersistedCache implements IPersistedCache { - private readonly cache = new Map(); + private readonly cache = new Map(); private readonly versionCache = new Map(); private readonly versionToCacheKey = new Map(); - public async get(entry: ICacheEntry): Promise { + public async get(entry: ICacheEntry): Promise { const key = getKeyForCacheEntry(entry); return this.cache.get(key); } - public async put(entry: ICacheEntry, value: ValueWithSnapshot): Promise { + public async put(entry: ICacheEntry, value: any): Promise { const key = getKeyForCacheEntry(entry); this.cache.set(key, value); + if (value.value?.snapshotTree?.id === undefined) { + return; + } const versionKey = `${value.value.snapshotTree.id}`; this.versionCache.set(versionKey, value); this.versionToCacheKey.set(versionKey, key);