feat: add test case for setStorageStateRecorder

alibaba · Oct 5, 2024 · b031aae · b031aae
1 parent 47146da
commit b031aae
Show file tree

Hide file tree

Showing 2 changed files with 42 additions and 3 deletions.
diff --git a/packages/hooks/src/usePageCacheState/__tests__/index.test.ts b/packages/hooks/src/usePageCacheState/__tests__/index.test.ts
@@ -179,7 +179,9 @@ describe('usePageCacheState', () => {
     // remove over count data
     expect(localStorage.getItem(getRealityStorageKey(LOCAL_STORAGE_KEY))).toBe(null);
 
-    secondHook.result.current.setState('B2');
+    act(() => {
+      secondHook.result.current.setState('B2');
+    });
     await act(async () => {
       jest.advanceTimersByTime(6000);
     });
@@ -198,5 +200,33 @@ describe('usePageCacheState', () => {
     });
     expect(secondHook.result.current.state).toBe('A2');
     expect(thirdHook.result.current.state).toBe('B3');
+
+    const thirdHookOtherVersion = setUp(LOCAL_STORAGE_KEY, 'A3', {
+      expire: 5,
+      version: '4',
+      subKey: 'test3',
+    });
+
+    // test delete all versions in a subKey
+    let recorder = thirdHookOtherVersion.result.current.operations.storageStateRecorder || {};
+    let versionsMap = recorder['test3'];
+    expect(Object.keys(versionsMap).toString()).toBe('3,4');
+    act(() => {
+      thirdHookOtherVersion.result.current.operations.delete('test3');
+    });
+    recorder = thirdHookOtherVersion.result.current.operations.storageStateRecorder || {};
+    versionsMap = recorder['test3'];
+    // only keep own version
+    expect(Object.keys(versionsMap || {}).toString()).toBe('4');
+
+    // test `setStorageStateRecorder`
+    act(() => {
+      thirdHookOtherVersion.result.current.operations.setStorageStateRecorder({});
+    });
+    expect(
+      Object.keys(
+        thirdHookOtherVersion.result.current.operations.storageStateRecorder || {},
+      ).toString(),
+    ).toBe('');
   });
 });
diff --git a/packages/hooks/src/usePageCacheState/index.ts b/packages/hooks/src/usePageCacheState/index.ts
@@ -93,7 +93,7 @@ export default function <T>(key: string, options?: Options<T>) {
   // ============================= hanlde recorder =============================
   const [pageCacheKeysRecorder, setPageCacheKeysRecorder] = useStorageState<
     StorageStateRecorder<T>
-  >(`${key}_all_keys_recorder`, {});
+  >(`${key}_storage_state_recorder`, {});
 
   // remove data when expired or over count
   useEffect(() => {
@@ -195,12 +195,21 @@ export default function <T>(key: string, options?: Options<T>) {
     setPageCache(undefined);
 
     // ===================== find all keys and remove =====================
-    const curSubKeyStorageRecorder = pageCacheKeysRecorder?.[key]?.[deleteSubKey] || {};
+    if (!pageCacheKeysRecorder) {
+      return;
+    }
+    const curSubKeyStorageRecorder = pageCacheKeysRecorder?.[deleteSubKey] || {};
     const versionKeys = Object.keys(curSubKeyStorageRecorder);
 
     versionKeys.forEach((versionItem) => {
       getStorage()?.removeItem(getRealityStorageKey(key, versionItem, deleteSubKey));
     });
+
+    delete pageCacheKeysRecorder[deleteSubKey];
+
+    setPageCacheKeysRecorder({
+      ...pageCacheKeysRecorder,
+    });
   });
 
   return [