addMethod: Prevent unnecessary transferFlags() calls

chaijs · Mar 31, 2016 · 2fd19ca · 2fd19ca
1 parent 218dd94
commit 2fd19ca
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/lib/chai/utils/addMethod.js b/lib/chai/utils/addMethod.js
@@ -40,11 +40,13 @@ module.exports = function (ctx, name, method) {
     var old_ssfi = flag(this, 'ssfi');
     if (old_ssfi && config.includeStack === false)
       flag(this, 'ssfi', ctx[name]);
+
     var result = method.apply(this, arguments);
+    if (result !== undefined)
+      return result;
 
     var newAssertion = new chai.Assertion();
     transferFlags(this, newAssertion);
-
-    return result === undefined ? newAssertion: result;
+    return newAssertion;
   };
 };