mirror of
https://github.com/grafana/grafana.git
synced 2025-07-30 12:42:17 +08:00
34 lines
875 B
TypeScript
34 lines
875 B
TypeScript
const { layout } = jest.requireActual('../../app/plugins/panel/nodeGraph/layout.worker.js');
|
|
|
|
class LayoutMockWorker {
|
|
timeout: number | undefined;
|
|
constructor() {}
|
|
postMessage(data: any) {
|
|
const { nodes, edges, config } = data;
|
|
this.timeout = window.setTimeout(() => {
|
|
this.timeout = undefined;
|
|
layout(nodes, edges, config);
|
|
// @ts-ignore
|
|
this.onmessage({ data: { nodes, edges } });
|
|
}, 1);
|
|
}
|
|
terminate() {
|
|
if (this.timeout) {
|
|
clearTimeout(this.timeout);
|
|
}
|
|
}
|
|
}
|
|
|
|
jest.mock('../../app/plugins/panel/nodeGraph/createLayoutWorker', () => ({
|
|
createWorker: () => new LayoutMockWorker(),
|
|
}));
|
|
|
|
class BasicMockWorker {
|
|
postMessage() {}
|
|
}
|
|
const mockCreateWorker = {
|
|
createWorker: () => new BasicMockWorker(),
|
|
};
|
|
|
|
jest.mock('../../app/features/live/centrifuge/createCentrifugeServiceWorker', () => mockCreateWorker);
|