58 lines
2.5 KiB
TypeScript
58 lines
2.5 KiB
TypeScript
import * as fs from "node:fs";
|
|
import * as os from "node:os";
|
|
import * as path from "node:path";
|
|
import { afterEach, describe, expect, it } from "vitest";
|
|
import { cleanupAnalysisArtifacts, getAnalysisArtifactPaths } from "./analysisArtifacts";
|
|
|
|
const tempDirs: string[] = [];
|
|
|
|
async function makeDataDir() {
|
|
const dir = await fs.promises.mkdtemp(path.join(os.tmpdir(), "analysis-artifacts-"));
|
|
tempDirs.push(dir);
|
|
return dir;
|
|
}
|
|
|
|
afterEach(async () => {
|
|
await Promise.all(tempDirs.splice(0).map((dir) => fs.promises.rm(dir, { recursive: true, force: true })));
|
|
});
|
|
|
|
describe("cleanupAnalysisArtifacts", () => {
|
|
it("deletes the plugin directory and all legacy files for only the requested result", async () => {
|
|
const dataDir = await makeDataDir();
|
|
const analysisDir = path.join(dataDir, "tasks", "7", "analysis");
|
|
await fs.promises.mkdir(path.join(analysisDir, "57"), { recursive: true });
|
|
await fs.promises.mkdir(path.join(analysisDir, "58"), { recursive: true });
|
|
const files = [
|
|
"heatmap_57.bin",
|
|
"deviation_57.bin",
|
|
"assembly_vertex_57_car-body.bin",
|
|
"assembly_vertex_57_bogie.bin",
|
|
"heatmap_58.bin",
|
|
"assembly_vertex_570_other.bin",
|
|
];
|
|
await Promise.all(files.map((name) => fs.promises.writeFile(path.join(analysisDir, name), name)));
|
|
|
|
await cleanupAnalysisArtifacts(7, 57, dataDir);
|
|
|
|
await expect(fs.promises.stat(path.join(analysisDir, "57"))).rejects.toMatchObject({ code: "ENOENT" });
|
|
for (const name of files.slice(0, 4)) {
|
|
await expect(fs.promises.stat(path.join(analysisDir, name))).rejects.toMatchObject({ code: "ENOENT" });
|
|
}
|
|
await expect(fs.promises.stat(path.join(analysisDir, "58"))).resolves.toBeDefined();
|
|
await expect(fs.promises.stat(path.join(analysisDir, "heatmap_58.bin"))).resolves.toBeDefined();
|
|
await expect(fs.promises.stat(path.join(analysisDir, "assembly_vertex_570_other.bin"))).resolves.toBeDefined();
|
|
});
|
|
|
|
it("rejects invalid IDs before constructing a filesystem target", () => {
|
|
expect(() => getAnalysisArtifactPaths(7, Number.NaN, "/data")).toThrow("resultId");
|
|
expect(() => getAnalysisArtifactPaths(7, -1, "/data")).toThrow("resultId");
|
|
expect(() => getAnalysisArtifactPaths(1.5, 2, "/data")).toThrow("taskId");
|
|
});
|
|
|
|
it("is idempotent when artifacts are already absent", async () => {
|
|
const dataDir = await makeDataDir();
|
|
await expect(cleanupAnalysisArtifacts(7, 57, dataDir)).resolves.toBeUndefined();
|
|
await expect(cleanupAnalysisArtifacts(7, 57, dataDir)).resolves.toBeUndefined();
|
|
});
|
|
});
|