confirmation_events_test.go RawHistoryFolder

1package agent
2
3import (
4	"reflect"
5	"testing"
6
7	"github.com/bnema/zut/packages/agent/extproto"
8	"github.com/bnema/zut/packages/core"
9)
10
11type rememberingDetailedConfirmer struct {
12	calls []core.ToolCallConfirmation
13}
14
15func (c *rememberingDetailedConfirmer) Confirm(string, string) core.ConfirmDecision {
16	panic("Confirm called instead of ConfirmToolCall")
17}
18
19func (c *rememberingDetailedConfirmer) ConfirmToolCall(call core.ToolCallConfirmation) core.ConfirmDecision {
20	c.calls = append(c.calls, call)
21	return core.ConfirmDecision{Allow: true, RememberTool: true}
22}
23
24func TestConfirmationEventConfirmerEmitsOnlyWhenPrompting(t *testing.T) {
25	inner := &rememberingDetailedConfirmer{}
26	var events []extproto.EventFromHost
27	notifier := &confirmationEventConfirmer{
28		inner: inner,
29		emit: func(event extproto.EventFromHost) {
30			events = append(events, event)
31		},
32	}
33	gate := core.NewConfirmGate(notifier)
34	call := core.ToolCallConfirmation{
35		ID:      "call-1",
36		Name:    "edit",
37		Summary: "main.go",
38		Content: "-old\n+new\n",
39	}
40
41	if allowed, _, _ := gate.CheckToolCall(call); !allowed {
42		t.Fatal("first call should be allowed")
43	}
44	if allowed, _, _ := gate.CheckToolCall(core.ToolCallConfirmation{Name: "edit", Summary: "other.go"}); !allowed {
45		t.Fatal("remembered tool should be allowed")
46	}
47
48	if len(events) != 1 {
49		t.Fatalf("events = %d, want 1", len(events))
50	}
51	want := extproto.EventFromHost{
52		Event:       "tool_confirmation_requested",
53		ToolID:      "call-1",
54		ToolName:    "edit",
55		ToolPreview: "main.go",
56	}
57	if !reflect.DeepEqual(events[0], want) {
58		t.Fatalf("event = %+v, want %+v", events[0], want)
59	}
60	if len(inner.calls) != 1 || inner.calls[0] != call {
61		t.Fatalf("inner calls = %+v, want [%+v]", inner.calls, call)
62	}
63}