print_stats_test.go RawHistoryFolder

1package agent
2
3import (
4	"encoding/json"
5	"os"
6	"path/filepath"
7	"testing"
8	"time"
9
10	"github.com/bnema/zut/packages/provider"
11)
12
13func TestWritePrintStats(t *testing.T) {
14	path := filepath.Join(t.TempDir(), "stats.json")
15	usage := provider.Usage{
16		InputTokens:          100,
17		OutputTokens:         40,
18		ReasoningTokens:      15,
19		ReasoningTokensKnown: true,
20		CacheReadTokens:      20,
21		CacheWriteTokens:     5,
22	}
23	if err := writePrintStats(path, "openai", "gpt-test", usage, 1500*time.Millisecond); err != nil {
24		t.Fatal(err)
25	}
26
27	data, err := os.ReadFile(path)
28	if err != nil {
29		t.Fatal(err)
30	}
31	var got printStats
32	if err := json.Unmarshal(data, &got); err != nil {
33		t.Fatal(err)
34	}
35	if got.Provider != "openai" || got.Model != "gpt-test" {
36		t.Fatalf("provider/model = %q/%q", got.Provider, got.Model)
37	}
38	if got.PromptTokens != 125 || got.ReasoningTokens == nil || *got.ReasoningTokens != 15 {
39		t.Fatalf("token stats = %+v", got)
40	}
41	if got.GeneratedOutputTokens != 25 || got.ElapsedMS != 1500 {
42		t.Fatalf("output/time stats = %+v", got)
43	}
44}
45
46func TestWritePrintStatsUsesNullForUnreportedReasoning(t *testing.T) {
47	path := filepath.Join(t.TempDir(), "stats.json")
48	usage := provider.Usage{InputTokens: 10, OutputTokens: 7}
49	if err := writePrintStats(path, "anthropic", "claude-test", usage, time.Second); err != nil {
50		t.Fatal(err)
51	}
52
53	data, err := os.ReadFile(path)
54	if err != nil {
55		t.Fatal(err)
56	}
57	var got map[string]any
58	if err := json.Unmarshal(data, &got); err != nil {
59		t.Fatal(err)
60	}
61	if got["reasoning_tokens"] != nil {
62		t.Fatalf("reasoning_tokens = %#v, want null", got["reasoning_tokens"])
63	}
64	if got["generated_output_tokens"] != float64(7) {
65		t.Fatalf("generated_output_tokens = %#v", got["generated_output_tokens"])
66	}
67}