subagent_policy_test.go RawHistoryFolder

1package agent
2
3import (
4	"context"
5	"os"
6	"testing"
7
8	"github.com/bnema/zut/packages/agent/subagents"
9)
10
11func TestSubagentPolicyIgnoresLegacyMaxTotalSpawned(t *testing.T) {
12	t.Setenv("ZUT_HOME", t.TempDir())
13	if err := os.WriteFile(ConfigPath(), []byte(`{"subagents":{"max_concurrent":1,"max_total_spawned":1}}`), 0o600); err != nil {
14		t.Fatal(err)
15	}
16	cfg, err := LoadConfig()
17	if err != nil {
18		t.Fatal(err)
19	}
20	if got := cfg.Subagents.MaxTotalSpawned; got != 1 {
21		t.Fatalf("legacy max_total_spawned = %d, want 1", got)
22	}
23
24	supervisor := subagents.New(subagents.Config{
25		Root:     t.TempDir(),
26		RepoRoot: t.TempDir(),
27		Policy:   subagentPolicyFromConfig(cfg.Subagents),
28		NewRunner: func(*subagents.Agent) subagents.Runner {
29			return subagents.RunnerFunc(func(context.Context, subagents.Sink) error { return nil })
30		},
31	})
32	t.Cleanup(supervisor.StopAll)
33
34	for _, task := range []string{"first", "second"} {
35		agent, err := supervisor.Spawn(context.Background(), task)
36		if err != nil {
37			t.Fatalf("Spawn(%q): %v", task, err)
38		}
39		agent.Wait()
40		if got := agent.Status(); got != subagents.StatusDone {
41			t.Fatalf("%s status = %s, want %s", task, got, subagents.StatusDone)
42		}
43	}
44}