Merge pull request #3722 from aryaemami59/createDraftSafeSelector

reduxjs · Oct 29, 2023 · 1016dd9 · 1016dd9
2 parents 2b6d65e + 7306e84
commit 1016dd9
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/packages/toolkit/src/createDraftSafeSelector.ts b/packages/toolkit/src/createDraftSafeSelector.ts
@@ -9,6 +9,7 @@ export const createDraftSafeSelectorCreator: typeof createSelectorCreator = (
     const selector = createSelector(...args)
     const wrappedSelector = (value: unknown, ...rest: unknown[]) =>
       selector(isDraft(value) ? current(value) : value, ...rest)
+    Object.assign(wrappedSelector, selector)
     return wrappedSelector as any
   }
 }

diff --git a/packages/toolkit/src/tests/createDraftSafeSelector.test.ts b/packages/toolkit/src/tests/createDraftSafeSelector.test.ts
@@ -11,6 +11,13 @@ test('handles normal values correctly', () => {
   let state = { value: 1 }
   expect(unsafeSelector(state)).toBe(1)
   expect(draftSafeSelector(state)).toBe(1)
+  expect(draftSafeSelector).toHaveProperty('resultFunc')
+  expect(draftSafeSelector).toHaveProperty('memoizedResultFunc')
+  expect(draftSafeSelector).toHaveProperty('lastResult')
+  expect(draftSafeSelector).toHaveProperty('dependencies')
+  expect(draftSafeSelector).toHaveProperty('recomputations')
+  expect(draftSafeSelector).toHaveProperty('resetRecomputations')
+  expect(draftSafeSelector).toHaveProperty('clearCache')
 
   state = { value: 2 }
   expect(unsafeSelector(state)).toBe(2)