feat(empower): support assertion methods with three or more arguments

Author: twada

lib/enhance.js

@@ -78,16 +78,7 @@ function enhance (target, formatter, config) {
         }
         var methodName = detectMethodName(matcher.calleeAst());
         if (methodName && typeof target[methodName] === 'function') {
-            switch (len) {
-            case 1:
-                enhancement[methodName] = decorateArgs(len, target, target[methodName], doPowerAssert);
-                break;
-            case 2:
-                enhancement[methodName] = decorateArgs(len, target, target[methodName], doPowerAssert);
-                break;
-            default:
-                throw new Error('not supported');
-            }
+            enhancement[methodName] = decorateArgs(len, target, target[methodName], doPowerAssert);
         }
     });
 

test/buster_assertions_test.js

@@ -19,8 +19,9 @@
             path: '/path/to/some_test.js',
             patterns: [
                 'assert(actual, [message])',
+                'assert.isNull(object, [message])',
                 'assert.same(actual, expected, [message])',
-                'assert.isNull(object, [message])'
+                'assert.near(actual, expected, delta, [message])'
             ]
         };
         return espowerSource(line, '/path/to/some_test.js', options);
@@ -39,8 +40,9 @@
     var assert = empower(busterAssertions.assert, fakeFormatter, {
         patterns: [
             'assert(actual, [message])',
+            'assert.isNull(object, [message])',
             'assert.same(actual, expected, [message])',
-            'assert.isNull(object, [message])'
+            'assert.near(actual, expected, delta, [message])'
         ]
     });
 
@@ -126,4 +128,37 @@
         });
     });
 
+
+    suite('buster assertion method with three arguments', function () {
+        test('when every argument is Identifier', function () {
+            var actualVal = 10.6, expectedVal = 10, delta = 0.5;
+            try {
+                eval(weave('assert.near(actualVal, expectedVal, delta);'));
+                assert.ok(false, 'AssertionError should be thrown');
+            } catch (e) {
+                baseAssert.equal(e.name, 'AssertionError');
+                baseAssert.equal(e.message, [
+                    '[assert.near] /path/to/some_test.js',
+                    'assert.near(actualVal, expectedVal, delta)',
+                    '[{"value":10.6,"espath":"arguments/0"},{"value":10,"espath":"arguments/1"},{"value":0.5,"espath":"arguments/2"}]: Expected 10.6 to be equal to 10 +/- 0.5'
+                ].join('\n'));
+            }
+        });
+
+        test('optional fourth argument', function () {
+            var actualVal = 10.6, expectedVal = 10, delta = 0.5, messageStr = 'not in delta';
+            try {
+                eval(weave('assert.near(actualVal, expectedVal, delta, messageStr);'));
+                assert.ok(false, 'AssertionError should be thrown');
+            } catch (e) {
+                baseAssert.equal(e.name, 'AssertionError');
+                baseAssert.equal(e.message, [
+                    '[assert.near] not in delta /path/to/some_test.js',
+                    'assert.near(actualVal, expectedVal, delta, messageStr)',
+                    '[{"value":10.6,"espath":"arguments/0"},{"value":10,"espath":"arguments/1"},{"value":0.5,"espath":"arguments/2"}]: Expected 10.6 to be equal to 10 +/- 0.5'
+                ].join('\n'));
+            }
+        });
+    });
+
 }));