handle comment lines smarter

tapjs · Apr 5, 2011 · db5b034 · db5b034
1 parent f59cff0
commit db5b034
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/lib/tap.js b/lib/tap.js
@@ -4,8 +4,12 @@ module.exports = tap
 
 function tap (res, count, diag) {
   if (typeof res === "string") {
-    return ("\n# "+res.trim().split(/\r?\n/).join("\n# ")+"\n")
-         .split("\n# \n").join("\n\n")
+    res = res.split(/\r?\n/).map(function (l) {
+      if (!l.trim()) return l.trim()
+      return "# " + l
+    }).join("\n")
+    if (res.substr(-1) !== "\n") res += "\n"
+    return res
   }
 
   if (!!process.env.TAP_NODIAG) diag = false