diff --git a/src/cli/program/register.setup.test.ts b/src/cli/program/register.setup.test.ts index 53c49b4238bd..0d078ab3fa04 100644 --- a/src/cli/program/register.setup.test.ts +++ b/src/cli/program/register.setup.test.ts @@ -522,6 +522,30 @@ describe("registerSetupCommand", () => { expect(setupCommandMock).not.toHaveBeenCalled(); }); + it.each([ + ["guided", ["--wizard"]], + ["classic", ["--classic"]], + ["non-interactive", ["--non-interactive", "--accept-risk"]], + ])("forwards --agent-name through %s setup", async (_mode, modeArgs) => { + await runCli([ + "setup", + ...modeArgs, + "--agent-name", + "robby", + "--workspace", + "/tmp/robby", + "--skip-bootstrap", + ]); + + expect(lastWizardOptions()).toMatchObject({ + agentName: "robby", + workspace: "/tmp/robby", + skipBootstrap: true, + }); + expect(setupCommandMock).not.toHaveBeenCalled(); + expect(runSystemAgentMock).not.toHaveBeenCalled(); + }); + it("runs setup wizard command for migration import flags", async () => { await runCli([ "setup", diff --git a/src/cli/program/register.setup.ts b/src/cli/program/register.setup.ts index 68d55e2a481c..415e98d7bd40 100644 --- a/src/cli/program/register.setup.ts +++ b/src/cli/program/register.setup.ts @@ -139,6 +139,7 @@ async function runOnboardingEntry( await setupWizardCommand( { workspace: readStringValue(options.workspace), + agentName: readStringValue(options.agentName), nonInteractive: Boolean(options.nonInteractive), acceptRisk: Boolean(options.acceptRisk), classic: Boolean(options.classic), @@ -206,6 +207,7 @@ export function registerSetupCommand(program: Command): void { "--workspace