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