/
JestHooks.ts
75 lines (66 loc) · 1.73 KB
/
JestHooks.ts
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
/**
* Copyright (c) Facebook, Inc. and its affiliates. 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.
*/
import {
JestHookSubscriber,
JestHookEmitter,
FileChange,
ShouldRunTestSuite,
TestRunComplete,
} from './types';
type AvailableHooks =
| 'onFileChange'
| 'onTestRunComplete'
| 'shouldRunTestSuite';
class JestHooks {
_listeners: {
onFileChange: Array<FileChange>;
onTestRunComplete: Array<TestRunComplete>;
shouldRunTestSuite: Array<ShouldRunTestSuite>;
};
constructor() {
this._listeners = {
onFileChange: [],
onTestRunComplete: [],
shouldRunTestSuite: [],
};
}
isUsed(hook: AvailableHooks) {
return this._listeners[hook] && this._listeners[hook].length;
}
getSubscriber(): JestHookSubscriber {
return {
onFileChange: fn => {
this._listeners.onFileChange.push(fn);
},
onTestRunComplete: fn => {
this._listeners.onTestRunComplete.push(fn);
},
shouldRunTestSuite: fn => {
this._listeners.shouldRunTestSuite.push(fn);
},
};
}
getEmitter(): JestHookEmitter {
return {
onFileChange: fs =>
this._listeners.onFileChange.forEach(listener => listener(fs)),
onTestRunComplete: results =>
this._listeners.onTestRunComplete.forEach(listener =>
listener(results),
),
shouldRunTestSuite: async testSuiteInfo => {
const result = await Promise.all(
this._listeners.shouldRunTestSuite.map(listener =>
listener(testSuiteInfo),
),
);
return result.every(Boolean);
},
};
}
}
export default JestHooks;