blob: 66f70a39076c9cb2227adb7b84caf8d45971d5e5 [file] [log] [blame]
var assert = require('assert');
var mustCallChecks = [];
function runCallChecks() {
var failed_count = 0;
for (var i=0 ; i< mustCallChecks.length; ++i) {
var context = mustCallChecks[i];
if (context.actual === context.expected) {
continue;
}
failed_count++;
console.log('Mismatched %s function calls. Expected %d, actual %d.',
context.name,
context.expected,
context.actual);
console.log(context.stack.split('\n').slice(2).join('\n'));
}
assert(failed_count === 0);
}
after(runCallChecks);
exports.mustCall = function(fn, expected) {
if (typeof expected !== 'number') expected = 1;
var context = {
expected: expected,
actual: 0,
stack: (new Error).stack,
name: fn.name || '<anonymous>'
};
mustCallChecks.push(context);
return function() {
context.actual++;
return fn.apply(this, arguments);
};
};