add Test.prototype._checkPlanCount()

Author: ariporad

lib/test.js

@@ -169,23 +169,23 @@ Test.prototype._end = function (err) {
 	this.exit();
 };
 
-Test.prototype.exit = function () {
-	var self = this;
-
-	function checkAssertLength() {
-		if (self.assertError === undefined && self.planCount !== null && self.planCount !== self.assertions.length) {
-			self._setAssertError(new assert.AssertionError({
-				actual: self.assertions.length,
-				expected: self.planCount,
-				message: 'Assertion count does not match planned',
-				operator: 'plan'
-			}));
+Test.prototype._checkPlanCount = function () {
+	if (this.assertError === undefined && this.planCount !== null && this.planCount !== this.assertions.length) {
+		this._setAssertError(new assert.AssertionError({
+			actual: this.assertions.length,
+			expected: this.planCount,
+			message: 'Assertion count does not match planned',
+			operator: 'plan'
+		}));
 
-			self.assertError.stack = self.planStack;
-		}
+		this.assertError.stack = this.planStack;
 	}
+};
+
+Test.prototype.exit = function () {
+	var self = this;
 
-	checkAssertLength();
+	this._checkPlanCount();
 
 	Promise.all(this.assertions)
 		.catch(function (err) {
@@ -198,7 +198,7 @@ Test.prototype.exit = function () {
 			// stop infinite timer
 			globals.clearTimeout(self._timeout);
 
-			checkAssertLength();
+			self._checkPlanCount();
 
 			if (!self.ended) {
 				self.ended = true;