149 lines
4.2 KiB
JavaScript
149 lines
4.2 KiB
JavaScript
'use strict'; /**
|
|
* Copyright (c) 2014-present, Facebook, Inc. All rights reserved.
|
|
*
|
|
* This source code is licensed under the BSD-style license found in the
|
|
* LICENSE file in the root directory of this source tree. An additional grant
|
|
* of patent rights can be found in the PATENTS file in the same directory.
|
|
*
|
|
*/
|
|
// This file is a heavily modified fork of Jasmine. Original license:
|
|
/*
|
|
Copyright (c) 2008-2016 Pivotal Labs
|
|
|
|
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.
|
|
*/
|
|
|
|
/* eslint-disable sort-keys */
|
|
|
|
const CallTracker = require('./CallTracker');
|
|
const createSpy = require('./createSpy');
|
|
const SpyStrategy = require('./SpyStrategy');
|
|
|
|
const formatErrorMsg = function () {
|
|
function generateErrorMsg(domain, usage) {
|
|
const usageDefinition = usage ? '\nUsage: ' + usage : '';
|
|
|
|
return function errorMsg(msg) {
|
|
return domain + ' : ' + msg + usageDefinition;
|
|
};
|
|
}
|
|
|
|
return generateErrorMsg;
|
|
};
|
|
|
|
function isSpy(putativeSpy) {
|
|
if (!putativeSpy) {
|
|
return false;
|
|
}
|
|
return (
|
|
putativeSpy.and instanceof SpyStrategy &&
|
|
putativeSpy.calls instanceof CallTracker);
|
|
|
|
}
|
|
|
|
const getErrorMsg = formatErrorMsg('<spyOn>', 'spyOn(<object>, <methodName>)');
|
|
|
|
function SpyRegistry(options) {
|
|
options = options || {};
|
|
const currentSpies =
|
|
options.currentSpies ||
|
|
function () {
|
|
return [];
|
|
};
|
|
|
|
this.allowRespy = function (allow) {
|
|
this.respy = allow;
|
|
};
|
|
|
|
this.spyOn = function (obj, methodName) {
|
|
if (obj === void 0) {
|
|
throw new Error(
|
|
getErrorMsg(
|
|
'could not find an object to spy upon for ' + methodName + '()'));
|
|
|
|
|
|
}
|
|
|
|
if (methodName === void 0) {
|
|
throw new Error(getErrorMsg('No method name supplied'));
|
|
}
|
|
|
|
if (obj[methodName] === void 0) {
|
|
throw new Error(getErrorMsg(methodName + '() method does not exist'));
|
|
}
|
|
|
|
if (obj[methodName] && isSpy(obj[methodName])) {
|
|
if (this.respy) {
|
|
return obj[methodName];
|
|
} else {
|
|
throw new Error(
|
|
getErrorMsg(methodName + ' has already been spied upon'));
|
|
|
|
}
|
|
}
|
|
|
|
let descriptor;
|
|
try {
|
|
descriptor = Object.getOwnPropertyDescriptor(obj, methodName);
|
|
} catch (e) {
|
|
// IE 8 doesn't support `definePropery` on non-DOM nodes
|
|
}
|
|
|
|
if (descriptor && !(descriptor.writable || descriptor.set)) {
|
|
throw new Error(
|
|
getErrorMsg(methodName + ' is not declared writable or has no setter'));
|
|
|
|
}
|
|
|
|
const originalMethod = obj[methodName];
|
|
const spiedMethod = createSpy(methodName, originalMethod);
|
|
let restoreStrategy;
|
|
|
|
if (Object.prototype.hasOwnProperty.call(obj, methodName)) {
|
|
restoreStrategy = function () {
|
|
obj[methodName] = originalMethod;
|
|
};
|
|
} else {
|
|
restoreStrategy = function () {
|
|
if (!delete obj[methodName]) {
|
|
obj[methodName] = originalMethod;
|
|
}
|
|
};
|
|
}
|
|
|
|
currentSpies().push({
|
|
restoreObjectToOriginalState: restoreStrategy });
|
|
|
|
|
|
obj[methodName] = spiedMethod;
|
|
|
|
return spiedMethod;
|
|
};
|
|
|
|
this.clearSpies = function () {
|
|
const spies = currentSpies();
|
|
for (let i = spies.length - 1; i >= 0; i--) {
|
|
const spyEntry = spies[i];
|
|
spyEntry.restoreObjectToOriginalState();
|
|
}
|
|
};
|
|
}
|
|
|
|
module.exports = SpyRegistry; |