Merge pull request #3722 from aryaemami59/createDraftSafeSelector

Author: markerikson

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
   }
 }

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)