LLM/moltbot
1
0
Fork 0
mirror of https://github.com/moltbot/moltbot.git synced 2026-04-26 16:06:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor(test): share sessions_spawn e2e mocks

Browse Source
This commit is contained in:
Peter Steinberger
2026-02-14 23:07:46 +00:00
parent cf26c409c6
commit 615f6e1e40
7 changed files with 92 additions and 190 deletions
Download Patch File Download Diff File Expand all files Collapse all files

46
src/agents/openclaw-tools.subagents.sessions-spawn-allows-cross-agent-spawning-configured.e2e.test.ts
View File

@@ -1,44 +1,22 @@
import { beforeEach, describe, expect, it, vi } from "vitest";
const callGatewayMock = vi.fn();
vi.mock("../gateway/call.js", () => ({
callGateway: (opts: unknown) => callGatewayMock(opts),
}));
let configOverride: ReturnType<(typeof import("../config/config.js"))["loadConfig"]> = {
session: {
mainKey: "main",
scope: "per-sender",
},
};
vi.mock("../config/config.js", async (importOriginal) => {
const actual = await importOriginal<typeof import("../config/config.js")>();
return {
...actual,
loadConfig: () => configOverride,
resolveGatewayPort: () => 18789,
};
});
import "./test-helpers/fast-core-tools.js";
import { beforeEach, describe, expect, it } from "vitest";
import { createOpenClawTools } from "./openclaw-tools.js";
import "./test-helpers/fast-core-tools.js";
import {
callGatewayMock,
resetConfigOverride,
setConfigOverride,
} from "./openclaw-tools.subagents.sessions-spawn.mocks.js";
import { resetSubagentRegistryForTests } from "./subagent-registry.js";
describe("openclaw-tools: subagents", () => {
beforeEach(() => {
configOverride = {
session: {
mainKey: "main",
scope: "per-sender",
},
};
resetConfigOverride();
});
it("sessions_spawn allows cross-agent spawning when configured", async () => {
resetSubagentRegistryForTests();
callGatewayMock.mockReset();
configOverride = {
setConfigOverride({
session: {
mainKey: "main",
scope: "per-sender",
@@ -53,7 +31,7 @@ describe("openclaw-tools: subagents", () => {
},
],
},
};
});
let childSessionKey: string | undefined;
callGatewayMock.mockImplementation(async (opts: unknown) => {
@@ -91,7 +69,7 @@ describe("openclaw-tools: subagents", () => {
it("sessions_spawn allows any agent when allowlist is *", async () => {
resetSubagentRegistryForTests();
callGatewayMock.mockReset();
configOverride = {
setConfigOverride({
session: {
mainKey: "main",
scope: "per-sender",
@@ -106,7 +84,7 @@ describe("openclaw-tools: subagents", () => {
},
],
},
};
});
let childSessionKey: string | undefined;
callGatewayMock.mockImplementation(async (opts: unknown) => {

35
src/agents/openclaw-tools.subagents.sessions-spawn-announces-agent-wait-lifecycle-events.e2e.test.ts
View File

@@ -1,39 +1,16 @@
import { beforeEach, describe, expect, it, vi } from "vitest";
const callGatewayMock = vi.fn();
vi.mock("../gateway/call.js", () => ({
callGateway: (opts: unknown) => callGatewayMock(opts),
}));
let configOverride: ReturnType<(typeof import("../config/config.js"))["loadConfig"]> = {
session: {
mainKey: "main",
scope: "per-sender",
},
};
vi.mock("../config/config.js", async (importOriginal) => {
const actual = await importOriginal<typeof import("../config/config.js")>();
return {
...actual,
loadConfig: () => configOverride,
resolveGatewayPort: () => 18789,
};
});
import "./test-helpers/fast-core-tools.js";
import { sleep } from "../utils.js";
import "./test-helpers/fast-core-tools.js";
import { createOpenClawTools } from "./openclaw-tools.js";
import {
callGatewayMock,
resetConfigOverride,
} from "./openclaw-tools.subagents.sessions-spawn.mocks.js";
import { resetSubagentRegistryForTests } from "./subagent-registry.js";
describe("openclaw-tools: subagents", () => {
beforeEach(() => {
configOverride = {
session: {
mainKey: "main",
scope: "per-sender",
},
};
resetConfigOverride();
});
it("sessions_spawn deletes session when cleanup=delete via agent.wait", async () => {

42
src/agents/openclaw-tools.subagents.sessions-spawn-applies-model-child-session.e2e.test.ts
View File

@@ -1,38 +1,16 @@
import { beforeEach, describe, expect, it, vi } from "vitest";
const callGatewayMock = vi.fn();
vi.mock("../gateway/call.js", () => ({
callGateway: (opts: unknown) => callGatewayMock(opts),
}));
let configOverride: ReturnType<(typeof import("../config/config.js"))["loadConfig"]> = {
session: {
mainKey: "main",
scope: "per-sender",
},
};
vi.mock("../config/config.js", async (importOriginal) => {
const actual = await importOriginal<typeof import("../config/config.js")>();
return {
...actual,
loadConfig: () => configOverride,
resolveGatewayPort: () => 18789,
};
});
import "./test-helpers/fast-core-tools.js";
import { beforeEach, describe, expect, it } from "vitest";
import { createOpenClawTools } from "./openclaw-tools.js";
import "./test-helpers/fast-core-tools.js";
import {
callGatewayMock,
resetConfigOverride,
setConfigOverride,
} from "./openclaw-tools.subagents.sessions-spawn.mocks.js";
import { resetSubagentRegistryForTests } from "./subagent-registry.js";
describe("openclaw-tools: subagents", () => {
beforeEach(() => {
configOverride = {
session: {
mainKey: "main",
scope: "per-sender",
},
};
resetConfigOverride();
});
it("sessions_spawn applies a model to the child session", async () => {
@@ -164,10 +142,10 @@ describe("openclaw-tools: subagents", () => {
it("sessions_spawn applies default subagent model from defaults config", async () => {
resetSubagentRegistryForTests();
callGatewayMock.mockReset();
configOverride = {
setConfigOverride({
session: { mainKey: "main", scope: "per-sender" },
agents: { defaults: { subagents: { model: "minimax/MiniMax-M2.1" } } },
};
});
const calls: Array<{ method?: string; params?: unknown }> = [];
callGatewayMock.mockImplementation(async (opts: unknown) => {

44
src/agents/openclaw-tools.subagents.sessions-spawn-normalizes-allowlisted-agent-ids.e2e.test.ts
View File

@@ -1,45 +1,23 @@
import { beforeEach, describe, expect, it, vi } from "vitest";
const callGatewayMock = vi.fn();
vi.mock("../gateway/call.js", () => ({
callGateway: (opts: unknown) => callGatewayMock(opts),
}));
let configOverride: ReturnType<(typeof import("../config/config.js"))["loadConfig"]> = {
session: {
mainKey: "main",
scope: "per-sender",
},
};
vi.mock("../config/config.js", async (importOriginal) => {
const actual = await importOriginal<typeof import("../config/config.js")>();
return {
...actual,
loadConfig: () => configOverride,
resolveGatewayPort: () => 18789,
};
});
import { emitAgentEvent } from "../infra/agent-events.js";
import "./test-helpers/fast-core-tools.js";
import { createOpenClawTools } from "./openclaw-tools.js";
import "./test-helpers/fast-core-tools.js";
import {
callGatewayMock,
resetConfigOverride,
setConfigOverride,
} from "./openclaw-tools.subagents.sessions-spawn.mocks.js";
import { resetSubagentRegistryForTests } from "./subagent-registry.js";
describe("openclaw-tools: subagents", () => {
beforeEach(() => {
configOverride = {
session: {
mainKey: "main",
scope: "per-sender",
},
};
resetConfigOverride();
});
it("sessions_spawn normalizes allowlisted agent ids", async () => {
resetSubagentRegistryForTests();
callGatewayMock.mockReset();
configOverride = {
setConfigOverride({
session: {
mainKey: "main",
scope: "per-sender",
@@ -54,7 +32,7 @@ describe("openclaw-tools: subagents", () => {
},
],
},
};
});
let childSessionKey: string | undefined;
callGatewayMock.mockImplementation(async (opts: unknown) => {
@@ -92,7 +70,7 @@ describe("openclaw-tools: subagents", () => {
it("sessions_spawn forbids cross-agent spawning when not allowed", async () => {
resetSubagentRegistryForTests();
callGatewayMock.mockReset();
configOverride = {
setConfigOverride({
session: {
mainKey: "main",
scope: "per-sender",
@@ -107,7 +85,7 @@ describe("openclaw-tools: subagents", () => {
},
],
},
};
});
const tool = createOpenClawTools({
agentSessionKey: "main",

42
src/agents/openclaw-tools.subagents.sessions-spawn-prefers-per-agent-subagent-model.e2e.test.ts
View File

@@ -1,50 +1,28 @@
import { beforeEach, describe, expect, it, vi } from "vitest";
const callGatewayMock = vi.fn();
vi.mock("../gateway/call.js", () => ({
callGateway: (opts: unknown) => callGatewayMock(opts),
}));
let configOverride: ReturnType<(typeof import("../config/config.js"))["loadConfig"]> = {
session: {
mainKey: "main",
scope: "per-sender",
},
};
vi.mock("../config/config.js", async (importOriginal) => {
const actual = await importOriginal<typeof import("../config/config.js")>();
return {
...actual,
loadConfig: () => configOverride,
resolveGatewayPort: () => 18789,
};
});
import "./test-helpers/fast-core-tools.js";
import { beforeEach, describe, expect, it } from "vitest";
import { createOpenClawTools } from "./openclaw-tools.js";
import "./test-helpers/fast-core-tools.js";
import {
callGatewayMock,
resetConfigOverride,
setConfigOverride,
} from "./openclaw-tools.subagents.sessions-spawn.mocks.js";
import { resetSubagentRegistryForTests } from "./subagent-registry.js";
describe("openclaw-tools: subagents", () => {
beforeEach(() => {
configOverride = {
session: {
mainKey: "main",
scope: "per-sender",
},
};
resetConfigOverride();
});
it("sessions_spawn prefers per-agent subagent model over defaults", async () => {
resetSubagentRegistryForTests();
callGatewayMock.mockReset();
configOverride = {
setConfigOverride({
session: { mainKey: "main", scope: "per-sender" },
agents: {
defaults: { subagents: { model: "minimax/MiniMax-M2.1" } },
list: [{ id: "research", subagents: { model: "opencode/claude" } }],
},
};
});
const calls: Array<{ method?: string; params?: unknown }> = [];
callGatewayMock.mockImplementation(async (opts: unknown) => {

37
src/agents/openclaw-tools.subagents.sessions-spawn-resolves-main-announce-target-from.e2e.test.ts
View File

@@ -1,40 +1,17 @@
import { beforeEach, describe, expect, it, vi } from "vitest";
import { sleep } from "../utils.ts";
const callGatewayMock = vi.fn();
vi.mock("../gateway/call.js", () => ({
callGateway: (opts: unknown) => callGatewayMock(opts),
}));
let configOverride: ReturnType<(typeof import("../config/config.js"))["loadConfig"]> = {
session: {
mainKey: "main",
scope: "per-sender",
},
};
vi.mock("../config/config.js", async (importOriginal) => {
const actual = await importOriginal<typeof import("../config/config.js")>();
return {
...actual,
loadConfig: () => configOverride,
resolveGatewayPort: () => 18789,
};
});
import { emitAgentEvent } from "../infra/agent-events.js";
import "./test-helpers/fast-core-tools.js";
import { sleep } from "../utils.ts";
import { createOpenClawTools } from "./openclaw-tools.js";
import "./test-helpers/fast-core-tools.js";
import {
callGatewayMock,
resetConfigOverride,
} from "./openclaw-tools.subagents.sessions-spawn.mocks.js";
import { resetSubagentRegistryForTests } from "./subagent-registry.js";
describe("openclaw-tools: subagents", () => {
beforeEach(() => {
configOverride = {
session: {
mainKey: "main",
scope: "per-sender",
},
};
resetConfigOverride();
});
it("sessions_spawn runs cleanup flow after subagent completion", async () => {

36
src/agents/openclaw-tools.subagents.sessions-spawn.mocks.ts Normal file
View File

@@ -0,0 +1,36 @@
import { vi } from "vitest";
export const callGatewayMock = vi.fn();
vi.mock("../gateway/call.js", () => ({
callGateway: (opts: unknown) => callGatewayMock(opts),
}));
export type SessionsSpawnTestConfig = ReturnType<
(typeof import("../config/config.js"))["loadConfig"]
>;
const defaultConfigOverride: SessionsSpawnTestConfig = {
session: {
mainKey: "main",
scope: "per-sender",
},
};
let configOverride: SessionsSpawnTestConfig = defaultConfigOverride;
export function resetConfigOverride() {
configOverride = defaultConfigOverride;
}
export function setConfigOverride(next: SessionsSpawnTestConfig) {
configOverride = next;
}
vi.mock("../config/config.js", async (importOriginal) => {
const actual = await importOriginal<typeof import("../config/config.js")>();
return {
...actual,
loadConfig: () => configOverride,
resolveGatewayPort: () => 18789,
};
});