plan errors

Author: James Halliday

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);
+});

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) {