Function main

scripts/windows-testing/parallel-executor.mjs:15–85

()

const initialStatusRegex = /Running (\d+) tests/;

async function main() {
  const [runfilesDir, targetName, ...testArgs] = process.argv.slice(2);
  const testEntrypoint = path.resolve(runfilesDir, '../', targetName);
  const testWorkingDir = path.resolve(runfilesDir, '_main');
  const tasks = [];
  const progress = {};

  tasks.push(
    spawnTest(
      'bash',
      [testEntrypoint, ...testArgs],
      {
        cwd: testWorkingDir,
        env: {
          // Try to construct a pretty hermetic environment, as within Bazel.
          PATH: process.env.PATH,
          E2E_SHARD_TOTAL: process.env.E2E_SHARD_TOTAL,
          E2E_SHARD_INDEX: process.env.E2E_SHARD_INDEX,
          FORCE_COLOR: '0',
          // Needed by `rules_js`
          BAZEL_BINDIR: '.',
          // Needed to run the E2E in a different temp path.
          E2E_TEMP: process.env.E2E_TEMP,
          // Using the `--glob` causes a bunch of issues due to path expansion in nested bash scripts.
          TESTBRIDGE_TEST_ONLY: process.env.TESTBRIDGE_TEST_ONLY,
        },
      },
      (s) => (progress[0] = s),
    ),
  );

  const printUpdate = () => {
    console.error(`----`);
    for (const [taskId, status] of Object.entries(progress)) {
      const durationInMin = (Date.now() - status.startTime) / 1000 / 60;
      console.error(
        `Shard #${taskId}: stage ${status.state} | ` +
          `${status.current}/${status.max} tests completed (${durationInMin.toFixed(2)}min)`,
      );
    }
  };

  const progressInterval = setInterval(printUpdate, 4000);

  try {
    const outputs = await Promise.all(tasks);
    printUpdate();

    for (const [idx, text] of outputs.entries()) {
      console.log(`---------- ${idx} -----------`);
      console.log(text);
    }

    console.error('');
    console.error('Done! Passing');
  } catch (e) {
    if (e instanceof TestSpawnError) {
      console.error(e.output);