| // Copyright Joyent, Inc. and other Node contributors. |
| // |
| // Permission is hereby granted, free of charge, to any person obtaining a |
| // copy of this software and associated documentation files (the |
| // "Software"), to deal in the Software without restriction, including |
| // without limitation the rights to use, copy, modify, merge, publish, |
| // distribute, sublicense, and/or sell copies of the Software, and to permit |
| // persons to whom the Software is furnished to do so, subject to the |
| // following conditions: |
| // |
| // The above copyright notice and this permission notice shall be included |
| // in all copies or substantial portions of the Software. |
| // |
| // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS |
| // OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF |
| // MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN |
| // NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, |
| // DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR |
| // OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE |
| // USE OR OTHER DEALINGS IN THE SOFTWARE. |
| |
| var test = require('tape'); |
| var assert = require('assert'); |
| |
| var noop = function() {}; |
| |
| var mustCallChecks = []; |
| |
| function runCallChecks(exitCode) { |
| if (exitCode !== 0) return; |
| |
| var failed = filter(mustCallChecks, function(context) { |
| if ('minimum' in context) { |
| context.messageSegment = 'at least ' + context.minimum; |
| return context.actual < context.minimum; |
| } else { |
| context.messageSegment = 'exactly ' + context.exact; |
| return context.actual !== context.exact; |
| } |
| }); |
| |
| for (var i = 0; i < failed.length; i++) { |
| var context = failed[i]; |
| console.log('Mismatched %s function calls. Expected %s, actual %d.', |
| context.name, |
| context.messageSegment, |
| context.actual); |
| // IE8 has no .stack |
| if (context.stack) console.log(context.stack.split('\n').slice(2).join('\n')); |
| } |
| |
| assert.strictEqual(failed.length, 0); |
| } |
| |
| exports.mustCall = function(fn, exact) { |
| return _mustCallInner(fn, exact, 'exact'); |
| }; |
| |
| function _mustCallInner(fn, criteria, field) { |
| if (typeof criteria == 'undefined') criteria = 1; |
| |
| if (typeof fn === 'number') { |
| criteria = fn; |
| fn = noop; |
| } else if (fn === undefined) { |
| fn = noop; |
| } |
| |
| if (typeof criteria !== 'number') |
| throw new TypeError('Invalid ' + field + ' value: ' + criteria); |
| |
| var context = { |
| actual: 0, |
| stack: (new Error()).stack, |
| name: fn.name || '<anonymous>' |
| }; |
| |
| context[field] = criteria; |
| |
| // add the exit listener only once to avoid listener leak warnings |
| if (mustCallChecks.length === 0) test.onFinish(function() { runCallChecks(0); }); |
| |
| mustCallChecks.push(context); |
| |
| return function() { |
| context.actual++; |
| return fn.apply(this, arguments); |
| }; |
| } |
| |
| exports.mustNotCall = function(msg) { |
| return function mustNotCall() { |
| assert.fail(msg || 'function should not have been called'); |
| }; |
| }; |
| |
| function filter(arr, fn) { |
| if (arr.filter) return arr.filter(fn); |
| var filtered = []; |
| for (var i = 0; i < arr.length; i++) { |
| if (fn(arr[i], i, arr)) filtered.push(arr[i]); |
| } |
| return filtered |
| } |