multica/server/internal/daemon/execenv/openclaw_config_timeout_test.go

package execenv

import (
	"context"
	"errors"
	"os"
	"os/exec"
	"path/filepath"
	"runtime"
	"strings"
	"testing"
	"time"
)

// slowestMeasuredOpenclawCall is the worst `openclaw config file` timing
// reported from the field (#7112): 10.86s on a 2020 Intel MacBook Pro, with
// 8.41s and 10.35s on the same host. It is a floor for what the default
// deadline must tolerate, not a target.
const slowestMeasuredOpenclawCall = 10860 * time.Millisecond

func TestResolveOpenclawCLITimeout(t *testing.T) {
	cases := []struct {
		name     string
		explicit time.Duration
		env      string
		want     time.Duration
	}{
		{name: "unset uses the default", want: openclawCLITimeout},
		{name: "explicit wins over env", explicit: 3 * time.Second, env: "90s", want: 3 * time.Second},
		{name: "duration string", env: "45s", want: 45 * time.Second},
		{name: "bare seconds", env: "45", want: 45 * time.Second},
		{name: "minutes past the ceiling are clamped", env: "1m30s", want: openclawCLIMaxTimeout},
		{name: "large bare seconds are clamped", env: "3600", want: openclawCLIMaxTimeout},
		// Past time.Duration's range there is nothing to clamp: parsing fails
		// and the default applies. Worth pinning because the old Atoi path
		// silently produced a negative duration here instead of failing.
		{name: "bare seconds beyond time.Duration fall back", env: "99999999999999999999", want: openclawCLITimeout},
		// A typo must degrade to the default rather than disabling the
		// deadline or failing task preparation outright.
		{name: "unparseable falls back", env: "soon", want: openclawCLITimeout},
		{name: "empty falls back", env: "   ", want: openclawCLITimeout},
		{name: "zero falls back", env: "0", want: openclawCLITimeout},
		{name: "negative falls back", env: "-5s", want: openclawCLITimeout},
		// Clamped, not honoured: too small breaks every host, too large eats
		// the outer preparation budget.
		{name: "below the floor is clamped", env: "10ms", want: openclawCLIMinTimeout},
		{name: "above the ceiling is clamped", env: "30m", want: openclawCLIMaxTimeout},
	}
	for _, tc := range cases {
		t.Run(tc.name, func(t *testing.T) {
			t.Setenv(OpenclawCLITimeoutEnv, tc.env)
			if got := resolveOpenclawCLITimeout(tc.explicit, nil); got != tc.want {
				t.Errorf("resolveOpenclawCLITimeout(%v, env=%q) = %v, want %v", tc.explicit, tc.env, got, tc.want)
			}
		})
	}
}

// TestExecOpenclawCLITimeoutIsTypedSentinel is what lets the daemon classify a
// local CLI stall structurally. Before this, the only signal was the words
// "deadline exceeded" in the message, which the shared classifier maps to
// agent_error.provider_network — "check your network" copy plus an auto-retry,
// for a failure that is local and repeats identically.
//
// The pre-existing context contract must survive alongside it: callers that
// check errors.Is(err, context.DeadlineExceeded) keep working.
func TestExecOpenclawCLITimeoutIsTypedSentinel(t *testing.T) {
	if runtime.GOOS == "windows" {
		t.Skip("shell shim shape is covered by the windows-tagged tests")
	}
	sleepBin, err := exec.LookPath("sleep")
	if err != nil {
		t.Skipf("no sleep binary available to build a slow shim: %v", err)
	}
	shim := writeShim(t, t.TempDir(), "#!/bin/sh\n"+sleepBin+" 1\n", "")

	ctx, cancel := context.WithTimeout(context.Background(), 100*time.Millisecond)
	defer cancel()
	_, err = execOpenclawCLI(ctx, shim, "config", "file")
	if err == nil {
		t.Fatal("expected the timed-out invocation to fail")
	}
	if !errors.Is(err, ErrOpenclawCLITimeout) {
		t.Errorf("errors.Is(err, ErrOpenclawCLITimeout) must hold\ngot: %s", err)
	}
	if !errors.Is(err, context.DeadlineExceeded) {
		t.Errorf("errors.Is(err, context.DeadlineExceeded) must keep holding\ngot: %s", err)
	}
	if !errors.Is(err, ErrOpenclawCLITimeout) || errors.Is(errors.New("openclaw config file: boom"), ErrOpenclawCLITimeout) {
		t.Errorf("the sentinel must not match unrelated CLI failures")
	}
}

// TestPrepareOpenclawConfigPropagatesTimeoutSentinel proves the sentinel
// survives the wrapping prepareOpenclawConfig does on the way out, which is
// what the daemon actually classifies.
func TestPrepareOpenclawConfigPropagatesTimeoutSentinel(t *testing.T) {
	stub := installOpenclawStub(t, map[string]openclawResponse{
		"config file": {err: &openclawCLITimeoutError{
			msg:   "openclaw config file: context deadline exceeded (process: signal: killed)",
			cause: context.DeadlineExceeded,
		}},
	})

	envRoot := t.TempDir()
	_, err := prepareOpenclawConfig(envRoot, envRoot, OpenclawConfigPrep{OpenclawBin: stub.bin})
	if err == nil {
		t.Fatal("expected preparation to fail on a CLI timeout")
	}
	if !errors.Is(err, ErrOpenclawCLITimeout) {
		t.Errorf("wrapped preparation error must keep the sentinel\ngot: %s", err)
	}
}

// TestExecOpenclawCLICancellationIsNotATimeout keeps the sentinel honest. A
// daemon shutdown or a cancelled task also lands in the cancellation branch,
// but it is not a slow CLI — labelling it as one would tell the user to raise
// a deadline that was never involved.
func TestExecOpenclawCLICancellationIsNotATimeout(t *testing.T) {
	if runtime.GOOS == "windows" {
		t.Skip("shell shim shape is covered by the windows-tagged tests")
	}
	sleepBin, err := exec.LookPath("sleep")
	if err != nil {
		t.Skipf("no sleep binary available to build a slow shim: %v", err)
	}
	shim := writeShim(t, t.TempDir(), "#!/bin/sh\n"+sleepBin+" 5\n", "")

	ctx, cancel := context.WithCancel(context.Background())
	go func() {
		time.Sleep(50 * time.Millisecond)
		cancel()
	}()
	defer cancel()
	_, err = execOpenclawCLI(ctx, shim, "config", "file")
	if err == nil {
		t.Fatal("expected the cancelled invocation to fail")
	}
	if errors.Is(err, ErrOpenclawCLITimeout) {
		t.Errorf("cancellation must not be reported as a CLI timeout\ngot: %s", err)
	}
	if !errors.Is(err, context.Canceled) {
		t.Errorf("cancellation must keep the wrapped context cause\ngot: %s", err)
	}
}

// TestOpenclawActiveConfigPathSpendsOneBudgetAcrossBothAttempts is the
// consequence of sharing that deadline, asserted rather than left implicit.
//
// A `config validate --json` that burns the whole budget leaves the `config file`
// fallback none, and the call fails on the shared deadline instead of starting a
// second full one. That is the intended trade: a CLI that cannot say where its
// config is inside the entire deadline will not answer the same question on a
// fresh one, and the failure still carries ErrOpenclawCLITimeout — the specific,
// non-retryable reason — rather than doubling the time spent inside the outer
// preparation budget to reach an identical conclusion.
func TestOpenclawActiveConfigPathSpendsOneBudgetAcrossBothAttempts(t *testing.T) {
	if runtime.GOOS == "windows" {
		t.Skip("shell shim shape is covered by the windows-tagged tests")
	}
	sleepBin, err := exec.LookPath("sleep")
	if err != nil {
		t.Skipf("no sleep binary available to build a slow shim: %v", err)
	}
	// `config validate --json` hangs; `config file` would answer instantly if it
	// were ever given the chance.
	shim := writeShim(t, t.TempDir(),
		"#!/bin/sh\n"+
			"case \"$*\" in\n"+
			"  'config validate --json') "+sleepBin+" 30 ;;\n"+
			"  'config file') printf '/tmp/openclaw.json\\n' ;;\n"+
			"esac\n", "")

	const budget = 1500 * time.Millisecond
	start := time.Now()
	_, _, err = openclawActiveConfigPath(shim, budget)
	elapsed := time.Since(start)

	if err == nil {
		t.Fatal("expected path resolution to fail once its shared deadline expired")
	}
	if !errors.Is(err, ErrOpenclawCLITimeout) {
		t.Errorf("errors.Is(err, ErrOpenclawCLITimeout) must hold so the daemon still "+
			"reports the specific, non-retryable reason\ngot: %s", err)
	}
	// The point of the assertion: two budgets would take at least 2x the deadline.
	// The margin above one budget covers the collector's reap and settle windows.
	if elapsed >= 2*budget {
		t.Errorf("path resolution took %v, i.e. at least two %v budgets — the "+
			"fallback must share the deadline, not start a fresh one", elapsed, budget)
	}
}

// TestPrepareOpenclawConfigWorstCaseCLIBudgets pins the multiplier the budget
// above depends on, and it counts *deadlines* rather than calls.
//
// That distinction is the review finding it exists for. The worst case is not the
// common two calls: `config validate --json` can fail to answer and fall back to
// `config file`, and a 2026.6+ host falls back to the registry subcommand — four
// invocations. The earlier version of this test could not see the first pair,
// because the stub synthesizes a `config validate --json` success from the
// `config file` response, so the two never fired in the same run. With the
// fallback carrying a fresh full deadline, the worst case was five budgets at the
// 60s ceiling, i.e. 5m of CLI time alone, landing exactly on
// daemon.defaultTaskPrepareTimeout — and the specific, non-retryable
// ErrOpenclawCLITimeout collapses back into the generic retryable one.
//
// A managed mcp_config used to add a fifth invocation of its own. It no longer
// reads anything (see TestPrepareOpenclawConfigManagedMcpCostsNoExtraCLICall), so
// this drives the managed path too and still expects four invocations across
// three deadlines: path resolution's two share one.
func TestPrepareOpenclawConfigWorstCaseCLIBudgets(t *testing.T) {
	envRoot := t.TempDir()
	workDir := filepath.Join(envRoot, "workdir")
	if err := os.MkdirAll(workDir, 0o755); err != nil {
		t.Fatalf("mkdir workdir: %v", err)
	}
	userConfigPath := filepath.Join(t.TempDir(), "openclaw.json")
	if err := os.WriteFile(userConfigPath, []byte(`{}`), 0o600); err != nil {
		t.Fatalf("write user cfg: %v", err)
	}

	stub := installOpenclawStub(t, map[string]openclawResponse{
		// 1a. preferred path resolution, on a CLI too old to support it
		"config validate --json": {err: errors.New("error: unknown command 'validate'")},
		// 1b. fallback path resolution, under the same deadline as 1a
		"config file": {stdout: userConfigPath},
		// 2. pre-2026.6 schema read, which this host does not have
		"config get agents.list --json": {err: errors.New("Config path not found: agents.list")},
		// 3. registry fallback
		"agents list --json": {stdout: `[{"id":"scout"}]`},
		// No fourth entry, and the managed mcp_config below is why that is worth
		// stating: managed MCP is prepared from a file this package writes, so it
		// adds no invocation. The stub errors on anything unregistered, so a
		// reintroduced read fails here instead of quietly costing a budget.
	})

	if _, err := prepareOpenclawConfig(envRoot, workDir, OpenclawConfigPrep{
		OpenclawBin: stub.bin,
		McpConfig:   []byte(`{"mcpServers":{"fs":{"command":"fs-server"}}}`),
	}); err != nil {
		t.Fatalf("prepareOpenclawConfig: %v", err)
	}

	var invocations []string
	deadlines := map[time.Time]bool{}
	for _, call := range stub.calls {
		invocation := strings.Join(call.args, " ")
		invocations = append(invocations, invocation)
		if call.deadline.IsZero() {
			t.Errorf("`openclaw %s` ran without a deadline; every CLI step must be bounded", invocation)
			continue
		}
		deadlines[call.deadline] = true
	}

	// The call graph itself, so a new invocation stays visible here even when it
	// costs no extra budget.
	wantInvocations := []string{
		"config validate --json",
		"config file",
		"config get agents.list --json",
		"agents list --json",
	}
	if strings.Join(invocations, " | ") != strings.Join(wantInvocations, " | ") {
		t.Errorf("worst-case invocations:\n got: %v\nwant: %v", invocations, wantInvocations)
	}

	if got := len(deadlines); got != openclawMaxCLIDeadlinesPerPreparation {
		t.Errorf("worst-case preparation consumed %d CLI deadlines (%d invocations), but openclawMaxCLIDeadlinesPerPreparation is %d\ncalls: %v",
			got, len(stub.calls), openclawMaxCLIDeadlinesPerPreparation, invocations)
	}

	// And specifically that the shared budget is path resolution's, since that is
	// the pairing the ceiling now depends on.
	if len(stub.calls) >= 2 && stub.calls[0].deadline != stub.calls[1].deadline {
		t.Errorf("`config validate --json` and its `config file` fallback ran under "+
			"separate deadlines (%v vs %v); they answer the same question and must "+
			"share one budget",
			stub.calls[0].deadline, stub.calls[1].deadline)
	}
}