43 lines
928 B
JavaScript
43 lines
928 B
JavaScript
|
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);
|
||
|
|
};
|
||
|
|
};
|