types_test.go RawHistoryFolder

1package sdk
2
3import (
4	"encoding/json"
5	"testing"
6	"time"
7
8	"github.com/bnema/zut/packages/core"
9)
10
11func TestToEventPreservesActivityAndToolStreamFacts(t *testing.T) {
12	retry := toEvent(core.EvRetryScheduled{
13		Scope: core.RetryScopeProvider, Attempt: 2, MaxAttempts: 3, Delay: 250 * time.Millisecond,
14	})
15	if retry.Type != "retry_scheduled" || retry.Scope != "provider" || retry.Attempt != 2 || retry.MaxAttempts != 3 || retry.DelayMS == nil || *retry.DelayMS != 250 {
16		t.Fatalf("retry event = %#v", retry)
17	}
18
19	zeroDelay := toEvent(core.EvRetryScheduled{Scope: core.RetryScopeProvider, Attempt: 2, MaxAttempts: 3})
20	encoded, err := json.Marshal(zeroDelay)
21	if err != nil {
22		t.Fatalf("marshal zero-delay retry: %v", err)
23	}
24	var wire map[string]any
25	if err := json.Unmarshal(encoded, &wire); err != nil {
26		t.Fatalf("decode zero-delay retry: %v", err)
27	}
28	if got, ok := wire["delay_ms"]; !ok || got != float64(0) {
29		t.Fatalf("zero-delay retry JSON delay_ms = %#v, want 0", got)
30	}
31
32	toolStart := toEvent(core.EvToolUseStart{ID: "call-1", Name: "read"})
33	if toolStart.Type != "tool_use_start" || toolStart.ID != "call-1" || toolStart.Name != "read" {
34		t.Fatalf("tool start = %#v", toolStart)
35	}
36	toolArgs := toEvent(core.EvToolUseArgs{ID: "call-1", Delta: `{"path":"a"`})
37	if toolArgs.Type != "tool_use_args" || toolArgs.ID != "call-1" || toolArgs.Delta == "" {
38		t.Fatalf("tool args = %#v", toolArgs)
39	}
40	toolEnd := toEvent(core.EvToolExecutionStarted{ID: "call-1", Name: "read"})
41	if toolEnd.Type != "tool_execution_started" || toolEnd.ID != "call-1" || toolEnd.Name != "read" {
42		t.Fatalf("tool execution = %#v", toolEnd)
43	}
44}