fix: allow to use customEqual

azu · Dec 13, 2017 · be22e31 · be22e31
1 parent 92d9e9d
commit be22e31
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/src/shallow-equal-props.ts b/src/shallow-equal-props.ts
@@ -28,8 +28,9 @@ export const shallowEqualProps = (
         console?: Pick<Console, "log" | "group" | "groupEnd">;
     }
 ): boolean => {
+    const customEqual = options && options.customEqual;
     return shallowEqual(propsA, propsB, {
-        customEqual: isEqualProps,
+        customEqual: customEqual || isEqualProps,
         debug: options && options.debug,
         console: options && options.console
     });

diff --git a/test/shallow-equal-props-test.ts b/test/shallow-equal-props-test.ts
@@ -51,4 +51,19 @@ describe("shallow-equal-props", () => {
             "elementA is not equal elementB"
         );
     });
+    it("pass customEqual", () => {
+        assert.equal(
+            shallowEqualProps(
+                { a: 1 },
+                { a: 2 },
+                {
+                    customEqual() {
+                        return true;
+                    }
+                }
+            ),
+            true,
+            "should be equal"
+        );
+    });
 });