ci(test): download the Electron binary once before unit specs run (#1739)

This commit is contained in:
4gray authored and GitHub committed 2026-09-29 08:57:07 +02:00
1 parent de7b19aee2
commit 76dd8c099e
6 files changed
+152 -17

No files matched your search

+23 -11
View File
@@ -8,29 +8,41 @@
* bounded worker count, so the runner's total CPU budget stays close to the
* machine's core count instead of multiplying with it.
*/
import { readdirSync } from 'node:fs';
import { readdirSync, readFileSync } from 'node:fs';
import path from 'node:path';
const SPEC_FILE = /\.(spec|test)\.ts$/;
/** Counts spec files under a directory; used to start the big projects first. */
export function countSpecFiles(directory) {
let count = 0;
function listSpecFiles(directory) {
let entries;
try {
entries = readdirSync(directory, { withFileTypes: true });
} catch {
return 0;
return [];
}
for (const entry of entries) {
return entries.flatMap((entry) => {
const fullPath = path.join(directory, entry.name);
if (entry.isDirectory()) {
count += countSpecFiles(fullPath);
} else if (entry.isFile() && SPEC_FILE.test(entry.name)) {
count += 1;
return listSpecFiles(fullPath);
}
}
return count;
return entry.isFile() && SPEC_FILE.test(entry.name) ? [fullPath] : [];
});
}
/** Counts spec files under a directory; used to start the big projects first. */
export function countSpecFiles(directory) {
return listSpecFiles(directory).length;
}
// `createRequire(__filename)('electron')` resolves the binary path, which
// downloads it on first use (see tools/testing/ensure-electron-binary.mjs).
const ELECTRON_BINARY_REQUIRE = /createRequire\([^)]*\)\(\s*['"]electron['"]\s*\)/;
/** Whether any spec under a directory resolves the Electron binary to exec it. */
export function specsResolveElectronBinary(directory) {
return listSpecFiles(directory).some((file) =>
ELECTRON_BINARY_REQUIRE.test(readFileSync(file, 'utf8'))
);
}
/**
+24
View File
@@ -13,6 +13,7 @@ import {
resolveConcurrency,
resolveWorkersPerProject,
runWithConcurrency,
specsResolveElectronBinary,
} from './coverage-run-pool.mjs';
let workDir;
@@ -35,6 +36,29 @@ test('counts spec and test files recursively and ignores sources', async () => {
assert.equal(countSpecFiles(path.join(workDir, 'missing')), 0);
});
test('detects specs that resolve the Electron binary, including split calls', async () => {
const mocked = path.join(workDir, 'mocked');
await mkdir(mocked, { recursive: true });
await writeFile(
path.join(mocked, 'a.spec.ts'),
"jest.mock('electron', () => ({}));\nimport { app } from 'electron';\n"
);
await writeFile(
path.join(mocked, 'b.ts'),
"createRequire(__filename)('electron');\n"
);
assert.equal(specsResolveElectronBinary(mocked), false);
const spawning = path.join(workDir, 'spawning', 'nested');
await mkdir(spawning, { recursive: true });
await writeFile(
path.join(spawning, 'c.spec.ts'),
"const electronPath = createRequire(__filename)(\n 'electron'\n) as string;\n"
);
assert.equal(specsResolveElectronBinary(path.dirname(spawning)), true);
assert.equal(specsResolveElectronBinary(path.join(workDir, 'missing')), false);
});
test('orders longest first and keeps policy order for ties', () => {
const projects = [
{ name: 'small' },
+19 -1
View File
@@ -1,6 +1,6 @@
#!/usr/bin/env node
import { spawn } from 'node:child_process';
import { spawn, spawnSync } from 'node:child_process';
import { existsSync, readFileSync, rmSync } from 'node:fs';
import os from 'node:os';
import path from 'node:path';
@@ -19,6 +19,7 @@ import {
resolveConcurrency,
resolveWorkersPerProject,
runWithConcurrency,
specsResolveElectronBinary,
} from './coverage-run-pool.mjs';
const workspaceRoot = process.cwd();
@@ -253,6 +254,23 @@ const specCounts = new Map(
countSpecFiles(path.join(workspaceRoot, project.sourceRoot)),
])
);
// Concurrent specs would otherwise race to download and extract the binary
// (ETXTBSY), so fetch it once before any project starts.
if (
tierAProjects.some((project) =>
specsResolveElectronBinary(path.join(workspaceRoot, project.sourceRoot))
)
) {
const prefetch = spawnSync(
process.execPath,
[path.join(workspaceRoot, 'tools/testing/ensure-electron-binary.mjs')],
{ stdio: 'inherit' }
);
if (prefetch.status !== 0) {
process.exit(prefetch.status ?? 1);
}
}
const ordered = orderLongestFirst(tierAProjects, (project) =>
specCounts.get(project.name)
);
+62
View File
@@ -0,0 +1,62 @@
#!/usr/bin/env node
// Downloads the Electron binary once and proves it runs, before anything
// spawns it concurrently.
//
// pnpm does not run Electron's postinstall (it is not in
// `onlyBuiltDependencies`), so `pnpm install` leaves no binary behind and
// `require('electron')` downloads and extracts it on first use. Unit specs
// that execute SQLite code under `ELECTRON_RUN_AS_NODE` resolve it that way,
// and Tier A projects and Jest workers run in parallel: one process can exec
// the binary while another is still extracting it (ETXTBSY on Linux, clobbered
// framework symlinks on macOS). Running this first leaves nothing to download.
//
// Usage: node tools/testing/ensure-electron-binary.mjs
import { execFileSync } from 'node:child_process';
import { createRequire } from 'node:module';
import process from 'node:process';
const require = createRequire(import.meta.url);
const { version } = require('electron/package.json');
let electronPath;
try {
electronPath = require('electron');
} catch (error) {
console.error(
`Electron ${version} binary download failed: ${error.message}`
);
process.exit(1);
}
let reported;
try {
reported = execFileSync(
electronPath,
['-e', 'process.stdout.write(process.versions.electron)'],
{
encoding: 'utf8',
// The child's stderr (e.g. a dyld or loader error) goes straight
// to the log instead of being repeated in the message below.
stdio: ['ignore', 'pipe', 'inherit'],
env: { ...process.env, ELECTRON_RUN_AS_NODE: '1' },
timeout: 60_000,
}
).trim();
} catch (error) {
console.error(
`Electron binary at ${electronPath} does not run (${error.code ?? `exit ${error.status}`}). ` +
'If an interrupted download left a partial copy, delete node_modules/electron/dist and path.txt, then rerun.'
);
process.exit(1);
}
if (reported !== version) {
console.error(
`Electron binary at ${electronPath} reports ${reported || 'no version'}, expected ${version}.`
);
process.exit(1);
}
console.log(`Electron ${version} binary ready at ${electronPath}`);