perf(test): speed up suites and reduce fs churn

This commit is contained in:
Peter Steinberger
2026-02-15 19:18:49 +00:00
parent 8fdde0429e
commit 92f8c0fac3
32 changed files with 1793 additions and 1398 deletions

View File

@@ -1,4 +1,4 @@
import { afterEach, describe, expect, it, vi } from "vitest";
import { afterEach, beforeAll, describe, expect, it, vi } from "vitest";
const noop = () => {};
let lifecycleHandler:
@@ -39,16 +39,24 @@ vi.mock("./subagent-registry.store.js", () => ({
}));
describe("subagent registry steer restarts", () => {
let mod: typeof import("./subagent-registry.js");
beforeAll(async () => {
mod = await import("./subagent-registry.js");
});
const flushAnnounce = async () => {
await new Promise<void>((resolve) => setImmediate(resolve));
};
afterEach(async () => {
announceSpy.mockClear();
announceSpy.mockReset();
announceSpy.mockResolvedValue(true);
lifecycleHandler = undefined;
const mod = await import("./subagent-registry.js");
mod.resetSubagentRegistryForTests({ persist: false });
});
it("suppresses announce for interrupted runs and only announces the replacement run", async () => {
const mod = await import("./subagent-registry.js");
mod.registerSubagentRun({
runId: "run-old",
childSessionKey: "agent:main:subagent:steer",
@@ -70,7 +78,7 @@ describe("subagent registry steer restarts", () => {
data: { phase: "end" },
});
await new Promise((resolve) => setTimeout(resolve, 0));
await flushAnnounce();
expect(announceSpy).not.toHaveBeenCalled();
const replaced = mod.replaceSubagentRunAfterSteer({
@@ -90,7 +98,7 @@ describe("subagent registry steer restarts", () => {
data: { phase: "end" },
});
await new Promise((resolve) => setTimeout(resolve, 0));
await flushAnnounce();
expect(announceSpy).toHaveBeenCalledTimes(1);
const announce = announceSpy.mock.calls[0]?.[0] as { childRunId?: string };
@@ -98,8 +106,6 @@ describe("subagent registry steer restarts", () => {
});
it("restores announce for a finished run when steer replacement dispatch fails", async () => {
const mod = await import("./subagent-registry.js");
mod.registerSubagentRun({
runId: "run-failed-restart",
childSessionKey: "agent:main:subagent:failed-restart",
@@ -117,11 +123,11 @@ describe("subagent registry steer restarts", () => {
data: { phase: "end" },
});
await new Promise((resolve) => setTimeout(resolve, 0));
await flushAnnounce();
expect(announceSpy).not.toHaveBeenCalled();
expect(mod.clearSubagentRunSteerRestart("run-failed-restart")).toBe(true);
await new Promise((resolve) => setTimeout(resolve, 0));
await flushAnnounce();
expect(announceSpy).toHaveBeenCalledTimes(1);
const announce = announceSpy.mock.calls[0]?.[0] as { childRunId?: string };
@@ -129,7 +135,6 @@ describe("subagent registry steer restarts", () => {
});
it("marks killed runs terminated and inactive", async () => {
const mod = await import("./subagent-registry.js");
const childSessionKey = "agent:main:subagent:killed";
mod.registerSubagentRun({
@@ -156,7 +161,6 @@ describe("subagent registry steer restarts", () => {
});
it("retries deferred parent cleanup after a descendant announces", async () => {
const mod = await import("./subagent-registry.js");
let parentAttempts = 0;
announceSpy.mockImplementation(async (params: unknown) => {
const typed = params as { childRunId?: string };
@@ -189,14 +193,14 @@ describe("subagent registry steer restarts", () => {
runId: "run-parent",
data: { phase: "end" },
});
await new Promise((resolve) => setTimeout(resolve, 0));
await flushAnnounce();
lifecycleHandler?.({
stream: "lifecycle",
runId: "run-child",
data: { phase: "end" },
});
await new Promise((resolve) => setTimeout(resolve, 0));
await flushAnnounce();
const childRunIds = announceSpy.mock.calls.map(
(call) => (call[0] as { childRunId?: string }).childRunId,

View File

@@ -1,12 +1,31 @@
import fs from "node:fs/promises";
import os from "node:os";
import path from "node:path";
import { describe, expect, it } from "vitest";
import { makeTempWorkspace } from "../test-helpers/workspace.js";
import { afterAll, beforeAll, describe, expect, it } from "vitest";
import { loadExtraBootstrapFiles } from "./workspace.js";
describe("loadExtraBootstrapFiles", () => {
let fixtureRoot = "";
let fixtureCount = 0;
const createWorkspaceDir = async (prefix: string) => {
const dir = path.join(fixtureRoot, `${prefix}-${fixtureCount++}`);
await fs.mkdir(dir, { recursive: true });
return dir;
};
beforeAll(async () => {
fixtureRoot = await fs.mkdtemp(path.join(os.tmpdir(), "openclaw-extra-bootstrap-"));
});
afterAll(async () => {
if (fixtureRoot) {
await fs.rm(fixtureRoot, { recursive: true, force: true });
}
});
it("loads recognized bootstrap files from glob patterns", async () => {
const workspaceDir = await makeTempWorkspace("openclaw-extra-bootstrap-glob-");
const workspaceDir = await createWorkspaceDir("glob");
const packageDir = path.join(workspaceDir, "packages", "core");
await fs.mkdir(packageDir, { recursive: true });
await fs.writeFile(path.join(packageDir, "TOOLS.md"), "tools", "utf-8");
@@ -20,7 +39,7 @@ describe("loadExtraBootstrapFiles", () => {
});
it("keeps path-traversal attempts outside workspace excluded", async () => {
const rootDir = await makeTempWorkspace("openclaw-extra-bootstrap-root-");
const rootDir = await createWorkspaceDir("root");
const workspaceDir = path.join(rootDir, "workspace");
const outsideDir = path.join(rootDir, "outside");
await fs.mkdir(workspaceDir, { recursive: true });
@@ -37,7 +56,7 @@ describe("loadExtraBootstrapFiles", () => {
return;
}
const rootDir = await makeTempWorkspace("openclaw-extra-bootstrap-symlink-");
const rootDir = await createWorkspaceDir("symlink");
const realWorkspace = path.join(rootDir, "real-workspace");
const linkedWorkspace = path.join(rootDir, "linked-workspace");
await fs.mkdir(realWorkspace, { recursive: true });