feat: simplify mocha error stack

.gitignore

@@ -3,6 +3,7 @@ coverage/
 !test/fixtures/custom-framework-app/node_modules/
 !test/fixtures/demo-app/node_modules/aliyun-egg/node_modules/
 !test/fixtures/test-files-glob/**
+!test/fixtures/test-files-stack/node_modules/
 .tmp
 .vscode
 *.log

README.md

@@ -115,6 +115,7 @@ You can pass any mocha argv.
 - `--require` require the given module
 - `--grep` only run tests matching <pattern>
 - `--timeout` milliseconds, default to 30000
+- `--full-trace` display the full stack trace, default to false.
 - see more at https://mochajs.org/#usage
 
 #### environment

lib/cmd/test.js

@@ -26,6 +26,9 @@ class TestCommand extends Command {
         alias: 't',
         type: 'number',
       },
+      'full-trace': {
+        description: 'display the full stack trace',
+      },
     };
   }
 
@@ -69,6 +72,9 @@ class TestCommand extends Command {
     /* istanbul ignore next */
     if (!Array.isArray(requireArr)) requireArr = [ requireArr ];
 
+    // clean mocha stack, inspired by https://github.com/rstacruz/mocha-clean
+    if (!testArgv.fullTrace) requireArr.unshift(require.resolve('../mocha-clean'));
+
     requireArr.push(require.resolve('co-mocha'));
 
     if (requireArr.includes('intelli-espower-loader')) {

lib/mocha-clean.js

@@ -0,0 +1,39 @@
+'use strict';
+
+const mocha = require('mocha');
+const internal = [
+  '(timers.js:',
+  '(node.js:',
+  '(module.js:',
+  '(domain.js:',
+  'GeneratorFunctionPrototype.next (native)',
+  'at Generator.next',
+  // node 8.x
+  'at Promise (<anonymous>)',
+  'at next (native)',
+  '__mocha_internal__',
+  /node_modules\/.*empower-core\//,
+  /node_modules\/.*mocha\//,
+  /node_modules\/.*co\//,
+  /node_modules\/.*co-mocha\//,
+];
+
+// monkey-patch `Runner#fail` to modify stack
+const originFn = mocha.Runner.prototype.fail;
+mocha.Runner.prototype.fail = function(test, err) {
+  /* istanbul ignore else */
+  if (err.stack) {
+    const stack = err.stack.split('\n').filter(line => {
+      line = line.replace(/\\\\?/g, '/');
+      // console.log('[%s] %s', !internal.some(rule => match(line, rule)) ? '  ' : '--', line);
+      return !internal.some(rule => match(line, rule));
+    });
+    err.stack = stack.join('\n');
+  }
+  return originFn.call(this, test, err);
+};
+
+function match(line, rule) {
+  if (rule instanceof RegExp) return rule.test(line);
+  return line.includes(rule);
+}

package.json

@@ -12,14 +12,14 @@
     "co-mocha": "^1.2.0",
     "common-bin": "^2.4.0",
     "debug": "^2.6.8",
-    "detect-port": "^1.1.3",
+    "detect-port": "^1.2.0",
     "egg-utils": "^2.2.0",
     "globby": "^6.1.0",
     "intelli-espower-loader": "^1.0.1",
     "istanbul": "^1.1.0-alpha.1",
     "mocha": "^3.4.2",
     "mz-modules": "^1.0.0",
-    "power-assert": "^1.4.2",
+    "power-assert": "^1.4.3",
     "ypkgfiles": "^1.4.0"
   },
   "devDependencies": {

test/fixtures/test-files-stack/package.json

@@ -0,0 +1,3 @@
+{
+  "name": "test-files-stack"
+}

test/fixtures/test-files-stack/test/assert.test.js

@@ -0,0 +1,11 @@
+'use strict';
+
+const assert = require('assert');
+
+describe('assert.test.js', () => {
+  it('should fail with simplify stack', () => {
+    [ 1 ].forEach(() => {
+      assert(1 === 0);
+    });
+  });
+});

test/fixtures/test-files-stack/test/promise.test.js

@@ -0,0 +1,13 @@
+'use strict';
+
+const co = require('co');
+
+describe('promise.test.js', () => {
+  it('should fail with simplify stack', function* () {
+    yield co(function* () {
+      return yield new Promise((resolve, reject) => {
+        reject(new Error('this is an error'));
+      });
+    });
+  });
+});

test/fixtures/test-files-stack/test/sleep.test.js

@@ -0,0 +1,11 @@
+'use strict';
+
+const sleep = require('my-sleep');
+
+describe('sleep.test.js', () => {
+  it('should fail with simplify stack', done => {
+    sleep(() => {
+      done(new Error('this is an error'));
+    });
+  });
+});

test/lib/cmd/cov.test.js

@@ -136,7 +136,7 @@ describe('test/lib/cmd/cov.test.js', () => {
         mm(process.env, 'TESTS', 'noexist.js');
         const cwd = path.join(__dirname, '../../fixtures/prerequire');
         yield coffee.fork(eggBin, [ 'cov' ], { cwd })
-          .debug()
+          // .debug()
           .coverage(false)
           .expect('code', 0)
           .end();

test/lib/cmd/test.test.js

@@ -3,6 +3,7 @@
 const path = require('path');
 const coffee = require('coffee');
 const mm = require('mm');
+const assert = require('assert');
 
 describe('test/lib/cmd/test.test.js', () => {
   const eggBin = require.resolve('../../../bin/egg-bin.js');
@@ -107,4 +108,63 @@ describe('test/lib/cmd/test.test.js', () => {
     .expect('code', 0)
     .end();
   });
+
+  describe('simplify mocha error stack', () => {
+    const cwd = path.join(__dirname, '../../fixtures/test-files-stack');
+
+    it('should clean assert error stack', done => {
+      mm(process.env, 'TESTS', 'test/assert.test.js');
+      coffee.fork(eggBin, [ 'test' ], { cwd })
+        // .debug()
+        .end((err, { stdout, code }) => {
+          assert(stdout.match(/AssertionError/));
+          assert(stdout.match(/at forEach .*assert.test.js:\d+:\d+/));
+          assert(stdout.match(/at Context.it .*assert.test.js:\d+:\d+/));
+          assert(stdout.match(/\bat\s+/g).length === 3);
+          assert(code === 1);
+          done(err);
+        });
+    });
+
+    it('should should show full stack trace', done => {
+      mm(process.env, 'TESTS', 'test/assert.test.js');
+      coffee.fork(eggBin, [ 'test', '--full-trace' ], { cwd })
+        // .debug()
+        .end((err, { stdout, code }) => {
+          assert(stdout.match(/AssertionError/));
+          assert(stdout.match(/at .*node_modules.*mocha/));
+          assert(stdout.match(/\bat\s+/g).length > 10);
+          assert(code === 1);
+          done(err);
+        });
+    });
+
+    it('should clean co error stack', done => {
+      mm(process.env, 'TESTS', 'test/promise.test.js');
+      coffee.fork(eggBin, [ 'test' ], { cwd })
+        // .debug()
+        .end((err, { stdout, code }) => {
+          assert(stdout.match(/Error: this is an error/));
+          assert(stdout.match(/at Promise .*promise.test.js:\d+:\d+/));
+          assert(stdout.match(/at Context\.<anonymous> .*promise.test.js:\d+:\d+/));
+          assert(stdout.match(/\bat\s+/g).length === 3);
+          assert(code === 1);
+          done(err);
+        });
+    });
+
+    it('should clean callback error stack', done => {
+      mm(process.env, 'TESTS', 'test/sleep.test.js');
+      coffee.fork(eggBin, [ 'test' ], { cwd })
+        // .debug()
+        .end((err, { stdout, code }) => {
+          assert(stdout.match(/Error: this is an error/));
+          assert(stdout.match(/at sleep .*sleep.test.js:\d+:\d+/));
+          assert(stdout.match(/at Timeout.setTimeout .*node_modules.*my-sleep.*index.js:\d+:\d+/));
+          assert(stdout.match(/\bat\s+/g).length === 2);
+          assert(code === 1);
+          done(err);
+        });
+    });
+  });
 });