refactor(current): update

unadlib · Mar 26, 2023 · 8fff243 · 8fff243
1 parent 08f795a
commit 8fff243
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/src/current.ts b/src/current.ts
@@ -20,7 +20,11 @@ export function handleReturnValue<T extends object>(
   forEach(value, (key, item, source) => {
     const proxyDraft = getProxyDraft(item);
     // just handle the draft which is created by the same rootDraft
-    if (proxyDraft && proxyDraft?.finalities === rootDraft?.finalities) {
+    if (
+      proxyDraft &&
+      rootDraft &&
+      proxyDraft.finalities === rootDraft.finalities
+    ) {
       isContainDraft = true;
       if (__DEV__ && warning) {
         console.warn(

diff --git a/test/immer/base.test.ts b/test/immer/base.test.ts
@@ -1522,7 +1522,7 @@ function runBaseTest(
 
           // Modified parent test
           parent.c = 1;
-          expect(produce({}, () => [parent.b])[0]).toBe(parent.b);
+          expect(produce(null, () => [parent.b])[0]).toBe(parent.b);
           parent.b.x; // Ensure proxy not revoked.
         });
       });