mirror of
https://github.com/stablyai/orca.git
synced 2026-10-08 16:02:37 +00:00
Neil
d8c871a1f0
* Schedule unit tests across runtimes by measured duration * Keep new SQLite fixture suites on Node after updating main * Inject scheduling timings instead of mocking the module * Keep agent-session database lifecycle contracts on Node
32 lines
1.3 KiB
JavaScript
32 lines
1.3 KiB
JavaScript
import { relative } from 'node:path'
|
|
import { BaseSequencer } from 'vitest/node'
|
|
import { rankFilesByDuration, readTimingBaseline } from './ci-shard-assignment.mjs'
|
|
|
|
export default class RuntimeSequencer extends BaseSequencer {
|
|
constructor(ctx, timingBaseline = undefined) {
|
|
super(ctx)
|
|
this.timingBaseline = timingBaseline
|
|
}
|
|
|
|
async sort(specifications) {
|
|
const inherited = await super.sort(specifications)
|
|
if (new Set(inherited.map((spec) => spec.project.name)).size < 2) {
|
|
return inherited
|
|
}
|
|
// Start long contracts across both runtimes instead of leaving Node until the end.
|
|
const baseline = this.timingBaseline ?? readTimingBaseline('unit')
|
|
const key = (spec) => relative(this.ctx.config.root, spec.moduleId).replaceAll('\\', '/')
|
|
const { weighted } = rankFilesByDuration(
|
|
[...new Set(inherited.map(key))],
|
|
baseline.timings,
|
|
baseline.overheadMs
|
|
)
|
|
const ranks = new Map(weighted.map(({ file }, index) => [file, index]))
|
|
return inherited.sort((a, b) => {
|
|
const groupOrder = a.project.config.sequence.groupOrder - b.project.config.sequence.groupOrder
|
|
const isolationOrder = Number(b.project.config.isolate) - Number(a.project.config.isolate)
|
|
return groupOrder || isolationOrder || ranks.get(key(a)) - ranks.get(key(b))
|
|
})
|
|
}
|
|
}
|