Function TestPrepareUserMessage_CommandResolution

pkg/cli/runner_test.go:462–483

TestPrepareUserMessage_CommandResolution tests that commands are resolved correctly before agent switching.

(t *testing.T)

// TestPrepareUserMessage_CommandResolution tests that commands are resolved
// correctly before agent switching.
func TestPrepareUserMessage_CommandResolution(t *testing.T) {
	t.Parallel()

	rt := &mockRuntimeWithOverrides{
		mockRuntime: &mockRuntime{},
	}
	rt.currentAgentInfoFn = func(context.Context) runtime.CurrentAgentInfo {
		return runtime.CurrentAgentInfo{
			Name: "test",
			Commands: map[string]types.Command{
				"fix": {
					Instruction: "Fix the file ${args[0]}",
				},
			},
		}
	}

	msg, _, err := PrepareUserMessage(t.Context(), rt, "/fix main.go", "")
	assert.NilError(t, err)

	assert.Equal(t, "Fix the file main.go", msg.Message.Content, "Command should be resolved with args")
}