Function TestPrepareUserMessage_AgentSwitching

pkg/cli/runner_test.go:322–435

TestPrepareUserMessage_AgentSwitching tests that PrepareUserMessage correctly handles agent-switching commands and returns empty messages on switch failures.

(t *testing.T)

// TestPrepareUserMessage_AgentSwitching tests that PrepareUserMessage correctly
// handles agent-switching commands and returns empty messages on switch failures.
func TestPrepareUserMessage_AgentSwitching(t *testing.T) {
	t.Parallel()

	tests := []struct {
		name              string
		userInput         string
		commandAgent      string
		setAgentErr       error
		expectedContent   string
		expectedAttach    string
		expectAgentSwitch bool
		expectNilMessage  bool
		expectError       bool
	}{
		{
			name:              "agent switch succeeds with trailing args",
			userInput:         "/plan design a login flow",
			commandAgent:      "planner",
			setAgentErr:       nil,
			expectedContent:   "design a login flow",
			expectedAttach:    "",
			expectAgentSwitch: true,
		},
		{
			name:              "agent switch succeeds without trailing args",
			userInput:         "/plan",
			commandAgent:      "planner",
			setAgentErr:       nil,
			expectedContent:   "",
			expectedAttach:    "",
			expectAgentSwitch: true,
			expectNilMessage:  true,
		},
		{
			name:              "agent switch fails - returns error",
			userInput:         "/plan design a login flow",
			commandAgent:      "planner",
			setAgentErr:       errors.New("agent not found"),
			expectedContent:   "",
			expectedAttach:    "",
			expectAgentSwitch: true,
			expectError:       true,
		},
		{
			name:              "non-agent command - no switch",
			userInput:         "/test regular command",
			commandAgent:      "",
			setAgentErr:       nil,
			expectedContent:   "This is the test instruction regular command",
			expectedAttach:    "",
			expectAgentSwitch: false,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			t.Parallel()