if a reporter has a .done method, call it before exiting

mochajs · May 17, 2014 · 6acbc7f · 6acbc7f
1 parent a1cb3bb
commit 6acbc7f
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/lib/mocha.js b/lib/mocha.js
@@ -367,5 +367,14 @@ Mocha.prototype.run = function(fn){
   if (options.growl) this._growl(runner, reporter);
   exports.reporters.Base.useColors = options.useColors;
   exports.reporters.Base.inlineDiffs = options.useInlineDiffs;
-  return runner.run(fn);
+
+  function done(failures) {
+      if (reporter.done) {
+          reporter.done(failures, fn);
+      } else {
+          fn(failures);
+      }
+  }
+
+  return runner.run(done);
 };