123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175 |
- 'use strict';
- Object.defineProperty(exports, '__esModule', {
- value: true
- });
- exports.install = install;
- var _isGeneratorFn = require('is-generator-fn');
- var _isGeneratorFn2 = _interopRequireDefault(_isGeneratorFn);
- var _co = require('co');
- var _co2 = _interopRequireDefault(_co);
- var _is_error = require('./is_error');
- var _is_error2 = _interopRequireDefault(_is_error);
- function _interopRequireDefault(obj) {
- return obj && obj.__esModule ? obj : {default: obj};
- }
- /**
- * Copyright (c) 2014-present, Facebook, Inc. All rights reserved.
- *
- * This source code is licensed under the MIT license found in the
- * LICENSE file in the root directory of this source tree.
- *
- *
- */
- /**
- * This module adds ability to test async promise code with jasmine by
- * returning a promise from `it/test` and `before/afterEach/All` blocks.
- */
- function isPromise(obj) {
- return obj && typeof obj.then === 'function';
- }
- function promisifyLifeCycleFunction(originalFn, env) {
- return function(fn, timeout) {
- if (!fn) {
- return originalFn.call(env);
- }
- const hasDoneCallback = fn.length > 0;
- if (hasDoneCallback) {
- // Jasmine will handle it
- return originalFn.call(env, fn, timeout);
- }
- const extraError = new Error();
- // We make *all* functions async and run `done` right away if they
- // didn't return a promise.
- const asyncJestLifecycle = function(done) {
- const wrappedFn = (0, _isGeneratorFn2.default)(fn)
- ? _co2.default.wrap(fn)
- : fn;
- const returnValue = wrappedFn.call({});
- if (isPromise(returnValue)) {
- returnValue.then(done.bind(null, null), error => {
- var _checkIsError = (0, _is_error2.default)(error);
- const isError = _checkIsError.isError,
- message = _checkIsError.message;
- if (message) {
- extraError.message = message;
- }
- done.fail(isError ? error : extraError);
- });
- } else {
- done();
- }
- };
- return originalFn.call(env, asyncJestLifecycle, timeout);
- };
- }
- // Similar to promisifyLifeCycleFunction but throws an error
- // when the return value is neither a Promise nor `undefined`
- function promisifyIt(originalFn, env) {
- return function(specName, fn, timeout) {
- if (!fn) {
- const spec = originalFn.call(env, specName);
- spec.pend('not implemented');
- return spec;
- }
- const hasDoneCallback = fn.length > 0;
- if (hasDoneCallback) {
- return originalFn.call(env, specName, fn, timeout);
- }
- const extraError = new Error();
- const asyncJestTest = function(done) {
- const wrappedFn = (0, _isGeneratorFn2.default)(fn)
- ? _co2.default.wrap(fn)
- : fn;
- const returnValue = wrappedFn.call({});
- if (isPromise(returnValue)) {
- returnValue.then(done.bind(null, null), error => {
- var _checkIsError2 = (0, _is_error2.default)(error);
- const isError = _checkIsError2.isError,
- message = _checkIsError2.message;
- if (message) {
- extraError.message = message;
- }
- done.fail(isError ? error : extraError);
- });
- } else if (returnValue === undefined) {
- done();
- } else {
- done.fail(
- new Error(
- 'Jest: `it` and `test` must return either a Promise or undefined.'
- )
- );
- }
- };
- return originalFn.call(env, specName, asyncJestTest, timeout);
- };
- }
- function makeConcurrent(originalFn, env) {
- return function(specName, fn, timeout) {
- if (env != null && !env.specFilter({getFullName: () => specName || ''})) {
- return originalFn.call(env, specName, () => Promise.resolve(), timeout);
- }
- let promise;
- try {
- promise = fn();
- if (!isPromise(promise)) {
- throw new Error(
- `Jest: concurrent test "${specName}" must return a Promise.`
- );
- }
- } catch (error) {
- return originalFn.call(env, specName, () => Promise.reject(error));
- }
- return originalFn.call(env, specName, () => promise, timeout);
- };
- }
- function install(global) {
- const jasmine = global.jasmine;
- const env = jasmine.getEnv();
- env.it = promisifyIt(env.it, env);
- env.fit = promisifyIt(env.fit, env);
- global.it.concurrent = makeConcurrent(env.it, env);
- global.it.concurrent.only = makeConcurrent(env.fit, env);
- global.it.concurrent.skip = makeConcurrent(env.xit, env);
- global.fit.concurrent = makeConcurrent(env.fit);
- env.afterAll = promisifyLifeCycleFunction(env.afterAll, env);
- env.afterEach = promisifyLifeCycleFunction(env.afterEach, env);
- env.beforeAll = promisifyLifeCycleFunction(env.beforeAll, env);
- env.beforeEach = promisifyLifeCycleFunction(env.beforeEach, env);
- }
|