confirm_lifecycle_test.go RawHistoryFolder

1package core
2
3import "testing"
4
5func TestConfirmationDecisionSources(t *testing.T) {
6	call := ToolCallConfirmation{ID: "call", Name: "bash"}
7	var nilGate *ConfirmGate
8	if d := nilGate.DecideToolCall(call); !d.Allow || d.Source != "yolo" {
9		t.Fatalf("nil gate: %+v", d)
10	}
11	if d := NewConfirmGate(nil).DecideToolCall(call); d.Allow || d.Source != "policy" {
12		t.Fatalf("no confirmer: %+v", d)
13	}
14	for _, rememberAll := range []bool{false, true} {
15		inner := &recordingConfirmer{replies: []ConfirmDecision{{Allow: true, RememberTool: !rememberAll, RememberAll: rememberAll}}}
16		gate := NewConfirmGate(inner)
17		if d := gate.DecideToolCall(call); !d.Allow || d.Source != "user" {
18			t.Fatalf("first: %+v", d)
19		}
20		if d := gate.DecideToolCall(call); !d.Allow || d.Source != "remembered" {
21			t.Fatalf("remembered: %+v", d)
22		}
23		if len(inner.calls) != 1 {
24			t.Fatal("remembered decision prompted again")
25		}
26		gate.AllowAll()
27		if d := gate.DecideToolCall(call); !d.Allow || d.Source != "yolo" {
28			t.Fatalf("explicit yolo: %+v", d)
29		}
30		gate.Reset()
31		if d := gate.DecideToolCall(call); d.Allow || d.Source != "user" {
32			t.Fatalf("reset: %+v", d)
33		}
34	}
35}