Per-worker setup/teardown

e2e/__tests__/globalSetupAndPerWorker.test.ts

@@ -0,0 +1,115 @@
+/**
+ * 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 {tmpdir} from 'os';
+import * as path from 'path';
+import * as fs from 'graceful-fs';
+import {cleanup, runYarnInstall} from '../Utils';
+import {json as runWithJson} from '../runJest';
+
+const DIR = path.join(tmpdir(), 'jest-global-setup-and-per-worker');
+const project1DIR = path.join(
+  tmpdir(),
+  'jest-global-setup-and-per-worker-project-1',
+);
+const project2DIR = path.join(
+  tmpdir(),
+  'jest-global-setup-and-per-worker-project-2',
+);
+const e2eDir = path.resolve(__dirname, '../global-setup-and-per-worker');
+
+beforeAll(() => {
+  runYarnInstall(e2eDir);
+});
+
+beforeEach(() => {
+  cleanup(DIR);
+  cleanup(project1DIR);
+  cleanup(project2DIR);
+});
+
+afterAll(() => {
+  cleanup(DIR);
+  cleanup(project1DIR);
+  cleanup(project2DIR);
+});
+
+test('globalSetup triggered once + globalSetupPerWorker triggered once per worker', () => {
+  const setupPath = path.join(e2eDir, 'setup.js');
+  const setupPerWorkerPath = path.join(e2eDir, 'setup-per-worker.js');
+  const result = runWithJson(e2eDir, [
+    '--maxWorkers=2',
+    '--workerIdleMemoryLimit=100MB',
+    `--globalSetup=${setupPath}`,
+    `--globalSetupPerWorker=${setupPerWorkerPath}`,
+    '--testPathPatterns=__tests__',
+  ]);
+
+  expect(result.exitCode).toBe(0);
+  const files = fs.readdirSync(DIR);
+  expect(files).toHaveLength(3);
+  const content = files.map(file => {
+    const data = fs.readFileSync(path.join(DIR, file), 'utf8');
+    return data.split('\n');
+  });
+  for (const [firstLine, secondLine] of content) {
+    if (secondLine) {
+      expect(firstLine).toBe('setup-per-worker');
+    } else {
+      expect(firstLine).toBe('setup');
+    }
+  }
+  const secondLines = content.map(([, secondLine]) => secondLine);
+  secondLines.sort();
+  expect(secondLines).toEqual(['1', '2', undefined]);
+});
+test('globalSetup + globalSetupPerWorker with worker threads', () => {
+  const setupPath = path.join(e2eDir, 'setup.js');
+  const setupPerWorkerPath = path.join(e2eDir, 'setup-per-worker.js');
+  const result = runWithJson(e2eDir, [
+    '--maxWorkers=2',
+    '--workerIdleMemoryLimit=100MB',
+    `--globalSetup=${setupPath}`,
+    `--globalSetupPerWorker=${setupPerWorkerPath}`,
+    '--testPathPatterns=__tests__',
+    '--workerThreads',
+  ]);
+
+  expect(result.exitCode).toBe(0);
+  const files = fs.readdirSync(DIR);
+  expect(files).toHaveLength(3);
+  const content = files.map(file => {
+    const data = fs.readFileSync(path.join(DIR, file), 'utf8');
+    return data.split('\n');
+  });
+  for (const [firstLine, secondLine] of content) {
+    if (secondLine) {
+      expect(firstLine).toBe('setup-per-worker');
+    } else {
+      expect(firstLine).toBe('setup');
+    }
+  }
+  const secondLines = content.map(([, secondLine]) => secondLine);
+  secondLines.sort();
+  expect(secondLines).toEqual(['1', '2', undefined]);
+});
+
+test('multiple projects', () => {
+  const configPath = path.resolve(e2eDir, 'projects.jest.config.js');
+
+  const result = runWithJson(e2eDir, [
+    '--maxWorkers=2',
+    '--workerIdleMemoryLimit=100MB',
+    `--config=${configPath}`,
+  ]);
+
+  expect(result.exitCode).toBe(0);
+
+  expect(fs.existsSync(DIR)).toBe(true);
+  expect(fs.existsSync(project1DIR)).toBe(true);
+  expect(fs.existsSync(project2DIR)).toBe(true);
+});