btw_lifecycle_test.go RawHistoryFolder

1package modes
2
3import (
4	"context"
5	"encoding/json"
6	"testing"
7
8	"github.com/patriceckhart/zot/packages/core"
9	"github.com/patriceckhart/zot/packages/provider"
10)
11
12func TestBtwPreservesContextAwarePolicy(t *testing.T) {
13	main := core.NewAgent(nil, "test", "", nil)
14	ctx, cancel := context.WithCancel(context.Background())
15	defer cancel()
16	called := false
17	main.BeforeToolExecuteContext = func(got context.Context, call provider.ToolCallBlock) (bool, string, json.RawMessage) {
18		called = true
19		if got != ctx || call.ID != "call" {
20			t.Fatal("policy lost call context")
21		}
22		return false, "blocked", nil
23	}
24	side := newBtwAgent(main, "", "test")
25	if side.BeforeToolExecuteContext == nil {
26		t.Fatal("side chat bypasses policy")
27	}
28	allowed, reason, _ := side.BeforeToolExecuteContext(ctx, provider.ToolCallBlock{ID: "call"})
29	if allowed || reason != "blocked" || !called {
30		t.Fatal("policy decision changed")
31	}
32}