-
-
Notifications
You must be signed in to change notification settings - Fork 6.4k
/
state.ts
78 lines (67 loc) · 2.26 KB
/
state.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
76
77
78
/**
* Copyright (c) Meta Platforms, Inc. and affiliates.
*
* This source code is licensed under the MIT license found in the
* LICENSE file in the root directory of this source tree.
*/
import type {Circus, Global} from '@jest/types';
import eventHandler from './eventHandler';
import formatNodeAssertErrors from './formatNodeAssertErrors';
import {EVENT_HANDLERS, STATE_SYM} from './types';
import {makeDescribe} from './utils';
/* eslint-disable no-restricted-globals */
const handlers: Array<Circus.EventHandler> = ((global as Global.Global)[
EVENT_HANDLERS
] = ((global as Global.Global)[
EVENT_HANDLERS
] as Array<Circus.EventHandler>) || [eventHandler, formatNodeAssertErrors]);
/* eslint-enable */
export const ROOT_DESCRIBE_BLOCK_NAME = 'ROOT_DESCRIBE_BLOCK';
const createState = (): Circus.State => {
const ROOT_DESCRIBE_BLOCK = makeDescribe(ROOT_DESCRIBE_BLOCK_NAME);
return {
currentDescribeBlock: ROOT_DESCRIBE_BLOCK,
currentlyRunningTest: null,
expand: undefined,
hasFocusedTests: false,
hasStarted: false,
includeTestLocationInResult: false,
maxConcurrency: 5,
parentProcess: null,
rootDescribeBlock: ROOT_DESCRIBE_BLOCK,
seed: 0,
testNamePattern: null,
testTimeout: 5000,
unhandledErrors: [],
unhandledRejectionErrorByPromise: new Map(),
};
};
/* eslint-disable no-restricted-globals */
export const resetState = (): void => {
(global as Global.Global)[STATE_SYM] = createState();
};
resetState();
export const getState = (): Circus.State =>
(global as Global.Global)[STATE_SYM] as Circus.State;
export const setState = (state: Circus.State): Circus.State =>
((global as Global.Global)[STATE_SYM] = state);
/* eslint-enable */
export const dispatch = async (event: Circus.AsyncEvent): Promise<void> => {
for (const handler of handlers) {
await handler(event, getState());
}
};
export const dispatchSync = (event: Circus.SyncEvent): void => {
for (const handler of handlers) {
handler(event, getState());
}
};
export const addEventHandler = (handler: Circus.EventHandler): void => {
handlers.push(handler);
};
export const removeEventHandler = (handler: Circus.EventHandler): void => {
const index = handlers.lastIndexOf(handler);
if (index !== -1) {
handlers.splice(index, 1);
}
};