plan errors

tape-testing · Nov 25, 2012 · 40aafce · 40aafce
1 parent 1ea9057
commit 40aafce
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 0 deletions.
diff --git a/example/timing.js b/example/timing.js
@@ -0,0 +1,12 @@
+var test = require('../');
+
+test('a simple test', function (t) {
+    t.plan(2);
+
+    t.equal(typeof Date.now, 'function');
+    var start = Date.now();
+
+    setTimeout(function () {
+        t.equal(Date.now() - start, 100);
+    }, 100);
+});
diff --git a/lib/test.js b/lib/test.js
@@ -39,6 +39,13 @@ Test.prototype.plan = function (n) {
 
 Test.prototype.end = function () {
     if (!this.ended) this.emit('end');
+    if (!this._planError && this.assertCount !== this._plan) {
+        this._planError = true;
+        this.fail('plan != count', {
+            expected : this._plan,
+            actual : this.assertCount
+        });
+    }
     this.ended = true;
 };
 
@@ -68,6 +75,14 @@ Test.prototype._assert = function assert (ok, opts) {
             if (!self.ended) self.end();
         });
     }
+
+    if (!self._planError && self.assertCount > self._plan) {
+        self._planError = true;
+        self.fail('plan != count', {
+            expected : self._plan,
+            actual : self.assertCount
+        });
+    }
 };
 
 Test.prototype.fail = function (msg, extra) {