extension_call_tool_test.go RawHistoryFolder

1package agent
2
3import (
4	"context"
5	"encoding/json"
6	"strings"
7	"testing"
8
9	"github.com/patriceckhart/zot/packages/agent/extensions"
10	"github.com/patriceckhart/zot/packages/core"
11	"github.com/patriceckhart/zot/packages/provider"
12)
13
14func TestExtensionToolCallerHonorsGuardAndTranscript(t *testing.T) {
15	tool := &lifecycleTool{}
16	ag := core.NewAgent(nil, "test", "", core.Registry{"probe": tool})
17	mgr := extensions.New(t.TempDir(), t.TempDir(), "test", "", "", nil)
18	var callID string
19	ag.BeforeToolExecuteContext = func(_ context.Context, call provider.ToolCallBlock) (bool, string, json.RawMessage) {
20		callID = call.ID
21		if got := mgr.ToolCallOrigin(call.ID); got != "caller" {
22			t.Errorf("call origin = %q", got)
23		}
24		return false, "denied by policy", nil
25	}
26	call := extensionToolCaller(mgr, func() *core.Agent { return ag })
27	out := call(context.Background(), "caller", "probe", json.RawMessage(`{}`))
28	if !out.IsError || len(out.Content) != 1 || !strings.Contains(out.Content[0].Text, "denied by policy") {
29		t.Fatalf("policy result: %+v", out)
30	}
31	if mgr.ToolCallOrigin(callID) != "" {
32		t.Fatal("call origin was not cleared")
33	}
34	if tool.args != nil || len(ag.Messages()) != 0 {
35		t.Fatalf("tool ran or transcript changed: args=%s messages=%v", tool.args, ag.Messages())
36	}
37	out = call(context.Background(), "caller", "missing", json.RawMessage(`{}`))
38	if !out.IsError || !strings.Contains(out.Content[0].Text, "unknown tool") {
39		t.Fatalf("unknown tool: %+v", out)
40	}
41}