1package tools
2
3import (
4 "context"
5 "encoding/json"
6 "testing"
7 "time"
8
9 "github.com/bnema/zut/packages/agent/scheduler"
10 "github.com/bnema/zut/packages/core"
11 "github.com/bnema/zut/packages/provider"
12)
13
14func TestScheduleToolScopesTasksToCurrentSession(t *testing.T) {
15 now := time.Date(2026, time.August, 15, 12, 0, 0, 0, time.UTC)
16 engine := scheduler.NewEngine(func() time.Time { return now })
17 tool := &ScheduleTool{
18 Engine: engine,
19 SessionID: func() string { return "session-a" },
20 Location: func() *time.Location { return time.UTC },
21 }
22
23 ctx := context.Background()
24 result, err := tool.Execute(ctx, json.RawMessage(`{"action":"add","cron":"@hourly","message":"check build"}`), nil)
25 if err != nil || result.IsError {
26 t.Fatalf("Execute(add) = %#v, %v", result, err)
27 }
28 if got := resultText(result); got == "" {
29 t.Fatal("Execute(add) returned no confirmation")
30 }
31
32 otherSession, err := engine.Add(scheduler.NewTaskInput{SessionID: "session-b", Cron: "@hourly", Message: "hidden", Location: time.UTC})
33 if err != nil {
34 t.Fatalf("Engine.Add() error = %v", err)
35 }
36 result, err = tool.Execute(ctx, json.RawMessage(`{"action":"cancel","id":"`+otherSession.ID+`"}`), nil)
37 if err != nil || !result.IsError {
38 t.Fatalf("Execute(cancel other session) = %#v, %v; want scoped error", result, err)
39 }
40}
41
42func resultText(result core.ToolResult) string {
43 for _, content := range result.Content {
44 if text, ok := content.(provider.TextBlock); ok {
45 return text.Text
46 }
47 }
48 return ""
49}