mirror of
https://github.com/stablyai/orca.git
synced 2026-09-21 16:02:20 +00:00
Neil
403b62a8d8
* ci: balance existing unit and E2E shards using recorded timings * ci: fix timing refresh units and deferred-menu test traversal * ci: preserve isolated E2E window launch policy * ci: keep diagnostic artifact outages from failing tests
20 lines
874 B
JavaScript
20 lines
874 B
JavaScript
import { relative } from 'node:path'
|
|
import { BaseSequencer } from 'vitest/node'
|
|
import { balanceFiles, readTimingBaseline, writeAssignment } from './ci-shard-assignment.mjs'
|
|
|
|
export default class TimingSequencer extends BaseSequencer {
|
|
async shard(specs) {
|
|
const { index, count } = this.ctx.config.shard
|
|
const key = (spec) => relative(this.ctx.config.root, spec.moduleId).replaceAll('\\', '/')
|
|
const baseline = readTimingBaseline('unit')
|
|
const assignment = balanceFiles(specs.map(key), count, baseline.timings, baseline.overheadMs)
|
|
writeAssignment(process.env.ORCA_SHARD_MANIFEST ?? 'ci-shards/unit-assignment.json', {
|
|
...assignment,
|
|
baselineSha256: baseline.baselineSha256,
|
|
selectedShard: index
|
|
})
|
|
const selected = new Set(assignment.shards[index - 1].files)
|
|
return specs.filter((spec) => selected.has(key(spec)))
|
|
}
|
|
}
|