Less API using power-assert

lib/assert.js

@@ -120,3 +120,5 @@ x.regexTest = function (regex, contents, msg) {
 x.ifError = x.error = function (err, msg) {
 	test(!err, create(err, 'Error', '!==', msg, x.ifError));
 };
+
+require('./enhance-assert')(x);

lib/babel.js

@@ -1,13 +1,19 @@
 'use strict';
 
 var resolveFrom = require('resolve-from');
+var createEspowerPlugin = require('babel-plugin-espower/create');
 
 var hasGenerators = parseInt(process.version.slice(1), 10) > 0;
 
 var options = {
 	only: /(test|test\-.+|test\/.+)\.js$/,
 	blacklist: hasGenerators ? ['regenerator'] : [],
-	optional: hasGenerators ? ['asyncToGenerator'] : []
+	optional: hasGenerators ? ['asyncToGenerator'] : [],
+	plugins: [
+		createEspowerPlugin(require('babel-core'), {
+			patterns: require('./enhance-assert').PATTERNS
+		})
+	]
 };
 
 try {

lib/enhance-assert.js

@@ -0,0 +1,34 @@
+module.exports = enhanceAssert;
+
+module.exports.PATTERNS = [
+	't.ok(value, [message])',
+	't.notOk(value, [message])',
+	't.true(value, [message])',
+	't.false(value, [message])',
+	't.is(value, expected, [message])',
+	't.not(value, expected, [message])',
+	't.same(value, expected, [message])',
+	't.notSame(value, expected, [message])',
+	't.regexTest(regex, contents, [message])'
+];
+
+function enhanceAssert(assert) {
+	var empower = require('empower');
+	var powerAssertFormatter = require('power-assert-formatter');
+	var powerAssertRenderers = require('power-assert-renderers');
+
+	empower(assert,
+		powerAssertFormatter({
+			renderers: [
+				powerAssertRenderers.AssertionRenderer,
+				powerAssertRenderers.SuccinctRenderer
+			]
+		}),
+		{
+			destructive: true,
+			modifyMessageOnRethrow: true,
+			saveContextOnRethrow: false,
+			patterns: module.exports.PATTERNS
+		}
+	);
+}

lib/test.js

@@ -70,6 +70,11 @@ Object.keys(assert).forEach(function (el) {
 	};
 });
 
+// Workaround for power-assert
+// `t` must be capturable for decorated assert output
+Test.prototype._capt = assert._capt;
+Test.prototype._expr = assert._expr;
+
 Test.prototype.plan = function (count) {
 	if (typeof count !== 'number') {
 		throw new TypeError('Expected a number');

package.json

@@ -62,17 +62,21 @@
     "arr-flatten": "^1.0.1",
     "ava-init": "^0.1.0",
     "babel-core": "^5.8.23",
+    "babel-plugin-espower": "^1.0.0",
     "bluebird": "^2.9.34",
     "chalk": "^1.0.0",
     "co": "^4.6.0",
     "core-assert": "^0.1.0",
+    "empower": "^1.0.2",
     "figures": "^1.4.0",
     "fn-name": "^2.0.0",
     "globby": "^3.0.1",
     "is-generator": "^1.0.2",
     "meow": "^3.3.0",
     "object-assign": "^4.0.1",
     "plur": "^2.0.0",
+    "power-assert-formatter": "^1.1.0",
+    "power-assert-renderers": "^0.1.0",
     "pretty-ms": "^2.0.0",
     "resolve-from": "^1.0.0",
     "set-immediate-shim": "^1.0.1",

test/fixture/power-assert.js

@@ -0,0 +1,8 @@
+import test from '../../';
+
+test(t => {
+	const a = 'foo';
+
+	t.ok(a === 'bar');
+	t.end();
+});

test/test.js

@@ -837,3 +837,16 @@ test('fail-fast mode', function (t) {
 		t.end();
 	});
 });
+
+test('power-assert support', function (t) {
+	t.plan(2);
+
+	execCli('fixture/power-assert.js', function (err, stdout, stderr) {
+		t.ok(err);
+
+		// t.ok(a === 'bar')
+		//      |
+		//      "foo"
+		t.true((/t\.ok\(a === 'bar'\)\s*\n\s+\|\s*\n\s+"foo"/m).test(stderr));
+	});
+});