mirror of https://github.com/openclaw/openclaw.git
106 lines
3.6 KiB
TypeScript
106 lines
3.6 KiB
TypeScript
import { Command } from "commander";
|
|
import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
|
|
|
|
const { acpAction, registerAcpCli } = vi.hoisted(() => {
|
|
const action = vi.fn();
|
|
const register = vi.fn((program: Command) => {
|
|
program.command("acp").action(action);
|
|
});
|
|
return { acpAction: action, registerAcpCli: register };
|
|
});
|
|
|
|
const { nodesAction, registerNodesCli } = vi.hoisted(() => {
|
|
const action = vi.fn();
|
|
const register = vi.fn((program: Command) => {
|
|
const nodes = program.command("nodes");
|
|
nodes.command("list").action(action);
|
|
});
|
|
return { nodesAction: action, registerNodesCli: register };
|
|
});
|
|
|
|
vi.mock("../acp-cli.js", () => ({ registerAcpCli }));
|
|
vi.mock("../nodes-cli.js", () => ({ registerNodesCli }));
|
|
|
|
const { registerSubCliByName, registerSubCliCommands } = await import("./register.subclis.js");
|
|
|
|
describe("registerSubCliCommands", () => {
|
|
const originalArgv = process.argv;
|
|
const originalDisableLazySubcommands = process.env.OPENCLAW_DISABLE_LAZY_SUBCOMMANDS;
|
|
|
|
const createRegisteredProgram = (argv: string[], name?: string) => {
|
|
process.argv = argv;
|
|
const program = new Command();
|
|
if (name) {
|
|
program.name(name);
|
|
}
|
|
registerSubCliCommands(program, process.argv);
|
|
return program;
|
|
};
|
|
|
|
beforeEach(() => {
|
|
if (originalDisableLazySubcommands === undefined) {
|
|
delete process.env.OPENCLAW_DISABLE_LAZY_SUBCOMMANDS;
|
|
} else {
|
|
process.env.OPENCLAW_DISABLE_LAZY_SUBCOMMANDS = originalDisableLazySubcommands;
|
|
}
|
|
registerAcpCli.mockClear();
|
|
acpAction.mockClear();
|
|
registerNodesCli.mockClear();
|
|
nodesAction.mockClear();
|
|
});
|
|
|
|
afterEach(() => {
|
|
process.argv = originalArgv;
|
|
if (originalDisableLazySubcommands === undefined) {
|
|
delete process.env.OPENCLAW_DISABLE_LAZY_SUBCOMMANDS;
|
|
} else {
|
|
process.env.OPENCLAW_DISABLE_LAZY_SUBCOMMANDS = originalDisableLazySubcommands;
|
|
}
|
|
});
|
|
|
|
it("registers only the primary placeholder and dispatches", async () => {
|
|
const program = createRegisteredProgram(["node", "openclaw", "acp"]);
|
|
|
|
expect(program.commands.map((cmd) => cmd.name())).toEqual(["acp"]);
|
|
|
|
await program.parseAsync(["acp"], { from: "user" });
|
|
|
|
expect(registerAcpCli).toHaveBeenCalledTimes(1);
|
|
expect(acpAction).toHaveBeenCalledTimes(1);
|
|
});
|
|
|
|
it("registers placeholders for all subcommands when no primary", () => {
|
|
const program = createRegisteredProgram(["node", "openclaw"]);
|
|
|
|
const names = program.commands.map((cmd) => cmd.name());
|
|
expect(names).toContain("acp");
|
|
expect(names).toContain("gateway");
|
|
expect(names).toContain("clawbot");
|
|
expect(registerAcpCli).not.toHaveBeenCalled();
|
|
});
|
|
|
|
it("re-parses argv for lazy subcommands", async () => {
|
|
const program = createRegisteredProgram(["node", "openclaw", "nodes", "list"], "openclaw");
|
|
|
|
expect(program.commands.map((cmd) => cmd.name())).toEqual(["nodes"]);
|
|
|
|
await program.parseAsync(["nodes", "list"], { from: "user" });
|
|
|
|
expect(registerNodesCli).toHaveBeenCalledTimes(1);
|
|
expect(nodesAction).toHaveBeenCalledTimes(1);
|
|
});
|
|
|
|
it("replaces placeholder when registering a subcommand by name", async () => {
|
|
const program = createRegisteredProgram(["node", "openclaw", "acp", "--help"], "openclaw");
|
|
|
|
await registerSubCliByName(program, "acp");
|
|
|
|
const names = program.commands.map((cmd) => cmd.name());
|
|
expect(names.filter((name) => name === "acp")).toHaveLength(1);
|
|
|
|
await program.parseAsync(["acp"], { from: "user" });
|
|
expect(registerAcpCli).toHaveBeenCalledTimes(1);
|
|
expect(acpAction).toHaveBeenCalledTimes(1);
|
|
});
|
|
});
|