-
-
Notifications
You must be signed in to change notification settings - Fork 6.4k
/
runGlobalHook.js
88 lines (73 loc) · 2.25 KB
/
runGlobalHook.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
/**
* 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.
*
* @flow
*/
import type {GlobalConfig} from 'types/Config';
import type {Test} from 'types/TestRunner';
import {extname, resolve} from 'path';
import pEachSeries from 'p-each-series';
import {addHook} from 'pirates';
import {ScriptTransformer} from 'jest-runtime';
const inJestsSource = __dirname.includes('packages/jest-cli');
let packagesRoot;
if (inJestsSource) {
packagesRoot = resolve(__dirname, '../../');
}
export default ({
allTests,
globalConfig,
moduleName,
}: {
allTests: Array<Test>,
globalConfig: GlobalConfig,
moduleName: 'globalSetup' | 'globalTeardown',
}): Promise<void> => {
const globalModulePaths = new Set(
allTests.map(test => test.context.config[moduleName]),
);
if (globalConfig[moduleName]) {
globalModulePaths.add(globalConfig[moduleName]);
}
if (globalModulePaths.size > 0) {
return pEachSeries(Array.from(globalModulePaths), async modulePath => {
if (!modulePath) {
return;
}
const projectConfig =
allTests
.map(t => t.context.config)
.find(c => c[moduleName] === modulePath) ||
// Fallback to first one
allTests[0].context.config;
const transformer = new ScriptTransformer(projectConfig);
const revertHook = addHook(
(code, filename) =>
transformer.transformSource(filename, code, false).code || code,
{
exts: [extname(modulePath)],
matcher(filename) {
// `babel-jest` etc would normally be caught by `node_modules`, but not in Jest's own repo
if (inJestsSource && filename.includes(packagesRoot)) {
return false;
}
return transformer._shouldTransform(filename);
},
},
);
// $FlowFixMe
const globalModule = require(modulePath);
if (typeof globalModule !== 'function') {
throw new TypeError(
`${moduleName} file must export a function at ${modulePath}`,
);
}
await globalModule(globalConfig);
revertHook();
});
}
return Promise.resolve();
};