libp2p-test-plans/transport-interop/src/compose-runner.ts

78 lines
2.6 KiB
TypeScript

import { tmpdir } from 'os'
import { promises as fs } from 'fs';
import path from 'path';
import { exec as execStd } from 'child_process';
import util from 'util';
import { ComposeSpecification, PropertiesServices } from "../compose-spec/compose-spec";
import { stringify } from 'yaml';
import { dialerStdout, dialerTimings } from './compose-stdout-helper';
const exec = util.promisify(execStd);
const timeoutSecs = 3 * 60
export type RunOpts = {
up: {
exitCodeFrom: string
renewAnonVolumes?: boolean
}
}
export type RunFailure = any
export async function run(namespace: string, compose: ComposeSpecification, opts: RunOpts): Promise<RunFailure | null> {
// sanitize namespace
const sanitizedNamespace = namespace.replace(/[^a-zA-Z0-9]/g, "-")
const dir = path.join(tmpdir(), "compose-runner", sanitizedNamespace)
// Check if directory exists
try {
await fs.access(dir)
await fs.rm(dir, { recursive: true, force: true })
} catch (e) {
}
await fs.mkdir(dir, { recursive: true })
// Create compose.yaml file
// Some docker compose environments don't like the name field to have special characters
const sanitizedComposeName = compose?.name.replace(/[^a-zA-Z0-9_-]/g, "_")
await fs.writeFile(path.join(dir, "compose.yaml"), stringify({ ...compose, name: sanitizedComposeName }))
const upFlags: Array<string> = []
if (opts.up.exitCodeFrom) {
upFlags.push(`--exit-code-from=${opts.up.exitCodeFrom}`)
}
if (opts.up.renewAnonVolumes) {
upFlags.push("--renew-anon-volumes")
}
try {
const { stdout, stderr } = await exec(`docker compose -f ${path.join(dir, "compose.yaml")} up ${upFlags.join(" ")}`, {
signal: AbortSignal.timeout(1000 * timeoutSecs)
})
try {
const testResultsParsed = dialerTimings(dialerStdout(stdout))
console.log("Finished:", namespace, testResultsParsed)
} catch (e) {
console.log("Failed to parse test results.")
console.log("stdout:")
console.log(stdout)
console.log("")
console.log("stderr:")
console.log(stderr)
console.log("")
throw e
}
} catch (e: any) {
console.log("Failure", e)
return e
} finally {
try {
const { stdout, stderr } = await exec(`docker compose -f ${path.join(dir, "compose.yaml")} down`);
} catch (e) {
console.log("Failed to compose down", e)
}
await fs.rm(dir, { recursive: true, force: true })
}
}