Files
orca/config/scripts/ssh-docker-ci-sharding.test.mjs
T

118 lines
4.3 KiB
JavaScript

import { readFileSync } from 'node:fs'
import { createRequire } from 'node:module'
import { dirname, join } from 'node:path'
import { expect, it } from 'vitest'
import { parse } from 'yaml'
import { runProcess } from '../../src/shared/child-process/run-process'
const require = createRequire(import.meta.url)
const root = join(import.meta.dirname, '../..')
const workflow = parse(readFileSync(join(root, '.github/workflows/e2e.yml'), 'utf8'))
const job = workflow.jobs['ssh-docker-watcher-isolation']
const readRunner = (name) => readFileSync(join(root, `config/scripts/${name}.mjs`), 'utf8')
const runnerSpecs = (name) =>
[...readRunner(name).matchAll(/^ '(tests\/e2e\/[^']+\.spec\.ts)',/gm)].map((match) => match[1])
const runners = [
'run-ssh-docker-e2e',
'run-ssh-docker-watcher-isolation-e2e',
'run-ssh-docker-terminal-parking-e2e'
]
it('gives every removed SSH spec a dedicated owner even for test-only edits', () => {
const changedRun = workflow.jobs['changed-e2e'].steps.find(
(step) => step.name === 'Run changed E2E specs'
).run
const excluded = [...changedRun.matchAll(/\. != "([^"]+)"/g)].map((match) => match[1])
const owned = runners.flatMap(runnerSpecs)
expect(owned.length).toBeGreaterThan(25)
expect(new Set(owned).size).toBe(owned.length)
for (const spec of owned) {
expect(excluded, spec).toContain(spec)
expect(job.if, spec).toContain(`contains(inputs.test_files, '${spec}')`)
}
expect(excluded.filter((spec) => !owned.includes(spec)).sort()).toEqual([
'tests/e2e/ssh-browser-network-execution-route.docker.unit.test.ts',
'tests/e2e/ssh-localhost.spec.ts',
'tests/e2e/terminal-ibus-hangul-native.spec.ts'
])
})
it('runs all four isolated shards and the special SSH suites exactly once', () => {
expect(job.strategy.matrix.shard).toEqual([1, 2, 3, 4])
expect(job.strategy['fail-fast']).toBe(false)
const remaining = job.steps.find((step) => step.name === 'Run remaining Docker SSH E2E')
expect(remaining.run).toContain('pnpm run test:e2e:ssh-docker --shard=${{ matrix.shard }}/4')
expect(remaining.if).toBe('always()')
expect(remaining['continue-on-error']).toBeUndefined()
expect(readRunner(runners[0])).toContain("'--workers=1'")
expect(job.steps.find((step) => step.name === 'Run Docker SSH watcher isolation E2E').if).toBe(
'matrix.shard == 1'
)
expect(
job.steps.find(
(step) => step.name === 'Run Docker SSH terminal parking + startup readiness E2E'
).if
).toBe('always() && matrix.shard == 1')
const upload = job.steps.find((step) => step.uses === 'actions/upload-artifact@v7')
expect(upload.with.name).toContain('${{ matrix.shard }}')
})
it('native Playwright shards preserve every SSH test and project exactly once', async () => {
const cli = join(dirname(require.resolve('playwright/package.json')), 'cli.js')
const env = {
...process.env,
ORCA_BACKGROUND_LAUNCH: '1',
ORCA_E2E_SSH_DOCKER: '1',
ORCA_E2E_LOCAL_SSH_BROWSER: '1',
ORCA_E2E_SSH_CLIENT_HOSTED_BROWSER: '1',
ORCA_E2E_WEB_CLIENT: '1'
}
async function discover(extra = []) {
const result = await runProcess({
program: process.execPath,
cwd: root,
args: [
cli,
'test',
...runnerSpecs(runners[0]),
'--config',
'tests/playwright.config.ts',
'--project=electron-headless',
'--project=electron-headful',
'--workers=1',
'--list',
'--reporter=json',
...extra
],
env,
timeoutMs: 30000
})
expect(result.code, result.stderr).toBe(0)
const report = JSON.parse(result.stdout)
expect(report.errors).toEqual([])
const ids = []
function visit(suite) {
for (const spec of suite.specs ?? []) {
for (const test of spec.tests) {
ids.push(`${spec.id}:${test.projectName}`)
}
}
for (const child of suite.suites ?? []) {
visit(child)
}
}
visit(report)
return ids
}
const full = await discover()
const sharded = []
for (const index of job.strategy.matrix.shard) {
const selected = await discover([`--shard=${index}/4`])
expect(selected.length).toBeGreaterThan(0)
sharded.push(...selected)
}
expect(full.length).toBeGreaterThanOrEqual(40)
expect(new Set(sharded).size).toBe(sharded.length)
expect(sharded.sort()).toEqual(full.sort())
}, 90000)