subagent_policy_test.go RawHistoryFolder

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