interactive_extension_policy_test.go RawHistoryFolder

1package agent
2
3import (
4	"context"
5	"encoding/json"
6	"strings"
7	"testing"
8
9	"github.com/bnema/zut/packages/agent/extensions"
10	"github.com/bnema/zut/packages/agent/modes"
11	"github.com/bnema/zut/packages/agent/tools"
12	"github.com/bnema/zut/packages/core"
13)
14
15func TestExtensionAdapterPreservesInteractivePolicy(t *testing.T) {
16	manager := extensions.New(t.TempDir(), t.TempDir(), "test", "provider", "model", nonInteractiveExtHooks{})
17	adapter := &extToolAdapter{mgr: manager}
18	tool := adapter.NewExtensionTool(ExtensionToolInfo{
19		Name: "ask_user", Interactive: true, Schema: json.RawMessage(`{"type":"object"}`),
20	})
21	if !extensions.IsInteractiveTool(tool) {
22		t.Fatal("adapter lost interactive registration")
23	}
24	result, err := tool.Execute(context.Background(), json.RawMessage(`{}`), nil)
25	if err != nil || !result.IsError {
26		t.Fatalf("headless execution result=%+v err=%v", result, err)
27	}
28}
29
30func TestInteractiveExtensionSupportRequiresAttachedHost(t *testing.T) {
31	hooks := &interactiveExtHooks{}
32	if hooks.SupportsInteractiveTools() {
33		t.Fatal("unattached host permits interactive tools")
34	}
35	hooks.attachInteractive(&modes.Interactive{})
36	if !hooks.SupportsInteractiveTools() {
37		t.Fatal("attached TUI rejects interactive tools")
38	}
39}
40
41func TestScheduledSessionRegistryExcludesInteractiveExtensionTools(t *testing.T) {
42	interactive := extensions.NewTool(nil, extensions.ToolInfo{Name: "ask_user", Interactive: true, Schema: json.RawMessage(`{"type":"object"}`)})
43	ordinary := extensions.NewTool(nil, extensions.ToolInfo{Name: "lookup", Schema: json.RawMessage(`{"type":"object"}`)})
44	registry := scheduledSessionRegistry(core.Registry{"ask_user": interactive, "lookup": ordinary})
45	if _, ok := registry["ask_user"]; ok {
46		t.Fatal("scheduled background session retained an interactive tool")
47	}
48	if registry["lookup"] != ordinary {
49		t.Fatal("scheduled background session lost an ordinary extension tool")
50	}
51}
52
53func TestResidentRegistryRejectsInteractiveExtensionTools(t *testing.T) {
54	tool := extensions.NewTool(nil, extensions.ToolInfo{Name: "ask_user", Interactive: true, Schema: json.RawMessage(`{"type":"object"}`)})
55	if _, err := residentChildRegistry(core.Registry{"ask_user": tool}, []string{"ask_user"}); err == nil || !strings.Contains(err.Error(), "interactive") {
56		t.Fatalf("resident registry accepted interactive tool: %v", err)
57	}
58}
59
60func TestNativeQuestionsExcludedFromBackgroundRegistries(t *testing.T) {
61	tool := &tools.AskUserQuestionTool{}
62	if !extensions.IsInteractiveTool(tool) {
63		t.Fatal("native question tool is not interactive")
64	}
65	reg := core.Registry{tool.Name(): tool}
66	if _, ok := scheduledSessionRegistry(core.Registry{tool.Name(): tool})[tool.Name()]; ok {
67		t.Fatal("scheduled registry retained questions")
68	}
69	if _, err := residentChildRegistry(reg, []string{tool.Name()}); err == nil || !strings.Contains(err.Error(), "interactive") {
70		t.Fatalf("resident registry: %v", err)
71	}
72}