Extract wrap matcher logic into function

mattphillips · Aug 7, 2018 · d1a57b4 · d1a57b4
1 parent 0fd13a0
commit d1a57b4
Showing 1 changed file with 21 additions and 14 deletions.
diff --git a/src/withMessage.js b/src/withMessage.js
@@ -9,26 +9,33 @@ class JestAssertionError extends Error {
   }
 }
 
+const wrapMatcher = (matcher, customMessage) => {
+  const newMatcher = (...args) => {
+    try {
+      matcher(...args);
+    } catch (error) {
+      if (typeof customMessage !== 'string' || customMessage.length < 1 || !error.matcherResult) {
+        throw error;
+      }
+
+      const { matcherResult } = error;
+      const message = () => 'Custom message:\n  ' + customMessage + '\n\n' + matcherResult.message();
+
+      throw new JestAssertionError({ ...matcherResult, message }, newMatcher);
+    }
+  };
+  return newMatcher;
+};
+
 const wrapMatchers = (matchers, customMessage) => {
   return Object.keys(matchers).reduce((acc, name) => {
     const matcher = matchers[name];
 
     if (typeof matcher === 'function') {
-      const newMatcher = (...args) => {
-        try {
-          matcher(...args);
-        } catch (error) {
-          if (typeof customMessage !== 'string' || customMessage.length < 1 || !error.matcherResult) {
-            throw error;
-          }
-
-          const { matcherResult } = error;
-          const message = () => 'Custom message:\n  ' + customMessage + '\n\n' + matcherResult.message();
-
-          throw new JestAssertionError(Object.assign({}, matcherResult, { message }), newMatcher);
-        }
+      return {
+        ...acc,
+        [name]: wrapMatcher(matcher, customMessage)
       };
-      return { ...acc, [name]: newMatcher };
     }
 
     return {