namuol · namuol · Jun 13, 2016 · Mar 10, 2016 · Mar 16, 2016 · Jun 13, 2016
diff --git a/Makefile b/Makefile
@@ -1,4 +1,4 @@
-.PHONY: pass fail test
+.PHONY: pass fail mix test
 
 TAP = ./node_modules/tape/bin/tape
 MIN = ./bin/tap-difflet
@@ -9,4 +9,10 @@ pass:
 fail:
 	@$(TAP) test/fail.js | $(MIN)
 
+mix:
+	@$(TAP) test/mix.js | $(MIN)
+
+mixp:
+	@$(TAP) test/mix.js | $(MIN) -p 
+
 test: pass fail
diff --git a/README.md b/README.md
@@ -12,7 +12,12 @@ npm install tap-difflet --save-dev
 ## Usage
 
 ~~~ text
-tape test/*.js | tap-difflet
+tape test/*.js | tap-difflet [options]
+
+Options:
+  -p --pessimistic  Only output failed tests.
+  -v --version      Print the version of tap-difflet.
+  -h --help         Show this.
 ~~~
 
 ## Output

diff --git a/bin/tap-difflet b/bin/tap-difflet
@@ -15,11 +15,82 @@ var objdiff   = require('difflet')({
 
 var ansidiff  = require('ansidiff');
 
+var docopt = require('docopt').docopt;
+
+var usage = [
+  'Process TAP output on STDIN and print diffs of failed comparisons.',
+  '',
+  'Usage: tap-difflet [options]',
+  '',
+  'Options:',
+  '  -p --pessimistic  Only output failed tests.',
+  '  -v --version      Print the version of tap-difflet.',
+  '  -h --help         Show this.',
+].join('\n');
+
+var pkg = require('../package.json');
+var options = docopt(usage, {
+  version: pkg.version,
+});
+
 var tap = parser();
 var out = through();
 var dup = duplexer(tap, out);
 
+if (options['--pessimistic']) {
+  var anAssertFailed = true;
+  var originalOut = out;
+  var buffer = null;
+  tap.on('comment', function(comment) {
+    if (buffer && anAssertFailed) {
+      for (var bi=0; bi<buffer.length; ++bi) {
+        originalOut.push(buffer[bi]);
+      }
+    }
+
+    buffer = null;
+    anAssertFailed = false;
+    out = originalOut;
+
+    if (getCommentType(comment) === CommentType.OTHER) {
+      buffer = [];
+      out = buffer;
+    }
+  });
+
+  tap.on('assert', function(res) {
+    if (!res.ok) {
+      anAssertFailed = true;
+    }
+  });
+}
+
 /* Helpers */
+var CommentType = {
+  OTHER: 1,
+  TESTS: 2,
+  PASS: 3,
+  FAIL: 4,
+  OK: 5
+};
+
+function getCommentType(comment) {
+  if (/^tests\s+[0-9]+$/gi.test(comment)) {
+    return CommentType.TESTS;
+  }
+  else if (/^pass\s+[0-9]+$/gi.test(comment)) {
+    return CommentType.PASS;
+  }
+  else if (/^fail\s+[0-9]+$/gi.test(comment)) {
+    return CommentType.FAIL;
+  }
+  else if (/^ok$/gi.test(comment)) {
+    return CommentType.OK;
+  }
+  else {
+    return CommentType.OTHER;
+  }
+};
 
 function output(str) {
   out.push(str);
@@ -31,25 +102,26 @@ var timer   = hirestime();
 var errors  = [];
 
 tap.on('comment', function(comment) {
-    if (/^tests\s+[0-9]+$/gi.test(comment)) {
-        output('\n');
-        comment = chalk.white.bold(comment);
-    }
-    else if (/^pass\s+[0-9]+$/gi.test(comment)) {
-        comment = chalk.green.bold(comment);
-    }
-    else if (/^fail\s+[0-9]+$/gi.test(comment)) {
-        comment = chalk.red.bold(comment);
-    }
-    else if (/^ok$/gi.test(comment)) {
-        return;
-    }
-    else {
-        output('\n');
-        comment = chalk.white.bold(comment);
-    }
+  var commentType = getCommentType(comment);
+  if (commentType === CommentType.TESTS) {
+    output('\n');
+    comment = chalk.white.bold(comment);
+  }
+  else if (commentType === CommentType.PASS) {
+    comment = chalk.green.bold(comment);
+  }
+  else if (commentType === CommentType.FAIL) {
+    comment = chalk.red.bold(comment);
+  }
+  else if (commentType === CommentType.OK) {
+    return;
+  }
+  else {
+    output('\n');
+    comment = chalk.white.bold(comment);
+  }
 
-    output('  ' + comment + '\n');
+  output('  ' + comment + '\n');
 });
 
 tap.on('assert', function(res) {
@@ -90,10 +162,10 @@ tap.on('results', function(res) {
 
 tap.on('diag', function (diag) {
   var expected, actual, at,
-      gotExpected = true,
-      gotActual = true,
-      gotAt = true,
-      str = '';
+    gotExpected = true,
+    gotActual = true,
+    gotAt = true,
+    str = '';
 
   if (diag.hasOwnProperty('expected')) {
     expected = diag.expected;
@@ -120,15 +192,15 @@ tap.on('diag', function (diag) {
 
   if (gotActual && gotExpected) {
     if (typeof expected !== typeof actual ||
-        typeof expected === "object" && (!actual || !expected)) {
+        typeof expected === 'object' && (!actual || !expected)) {
       str = 'Expected ' + typeof expected + ' but got ' + typeof actual;
-    } else if (typeof expected === "string") {
+    } else if (typeof expected === 'string') {
       if (str.indexOf('\n') >= 0) {
         str = ansidiff.lines(expected, actual);
       } else {
         str = ansidiff.chars(expected, actual);
       }
-    } else if (typeof expected === "object") {
+    } else if (typeof expected === 'object') {
       str = objdiff(expected, actual);
     } else {
       str = chalk.white('Expected ') + chalk.bold(''+expected) + chalk.white(' but got ') + chalk.bold(''+actual);

diff --git a/package.json b/package.json
@@ -37,6 +37,7 @@
     "ansidiff": "^1.0.0",
     "chalk": "^0.5.1",
     "difflet": "^0.2.6",
+    "docopt": "^0.6.2",
     "duplexer": "^0.1.1",
     "hirestime": "^0.2.4",
     "pretty-ms": "^1.0.0",

diff --git a/test/mix.js b/test/mix.js
@@ -0,0 +1,2 @@
+require ('./pass');
+require ('./fail');