args_test.go RawHistoryFolder

1package agent
2
3import "testing"
4
5func TestParseArgsTemperatureAllowsZero(t *testing.T) {
6	args, err := ParseArgs([]string{"--temperature", "0"})
7	if err != nil {
8		t.Fatalf("ParseArgs returned %v", err)
9	}
10	if args.Temperature == nil || *args.Temperature != 0 {
11		t.Fatalf("Temperature = %v; want 0", args.Temperature)
12	}
13}
14
15func TestParseArgsTemperatureRejectsOutOfRange(t *testing.T) {
16	if _, err := ParseArgs([]string{"--temperature", "2.1"}); err == nil {
17		t.Fatal("ParseArgs accepted out-of-range temperature")
18	}
19}
20
21func TestParseArgsYes(t *testing.T) {
22	for _, flag := range []string{"-y", "--yes"} {
23		args, err := ParseArgs([]string{flag, "--print", "hi"})
24		if err != nil {
25			t.Fatalf("ParseArgs(%q): %v", flag, err)
26		}
27		if !args.Yes {
28			t.Fatalf("ParseArgs(%q): Yes = false", flag)
29		}
30		if args.Mode != ModePrint || args.Prompt != "hi" {
31			t.Fatalf("ParseArgs(%q): Mode=%q Prompt=%q", flag, args.Mode, args.Prompt)
32		}
33	}
34}
35
36func TestParseArgsStatsRequiresPrintMode(t *testing.T) {
37	args, err := ParseArgs([]string{"-p", "--stats", "stats.json", "hi"})
38	if err != nil {
39		t.Fatal(err)
40	}
41	if args.StatsPath != "stats.json" || args.Mode != ModePrint {
42		t.Fatalf("StatsPath=%q Mode=%q", args.StatsPath, args.Mode)
43	}
44
45	if _, err := ParseArgs([]string{"--stats", "stats.json", "hi"}); err == nil {
46		t.Fatal("ParseArgs accepted --stats without print mode")
47	}
48}
49
50func TestParseArgsNoContextFiles(t *testing.T) {
51	for _, flag := range []string{"--no-context-files", "-nc"} {
52		args, err := ParseArgs([]string{flag})
53		if err != nil {
54			t.Fatalf("ParseArgs(%q): %v", flag, err)
55		}
56		if !args.NoContextFiles {
57			t.Fatalf("ParseArgs(%q): NoContextFiles = false", flag)
58		}
59	}
60}
61
62func TestParseArgsStream(t *testing.T) {
63	args, err := ParseArgs([]string{"--stream", "hi"})
64	if err != nil {
65		t.Fatal(err)
66	}
67	if args.Mode != ModeStream || args.Prompt != "hi" {
68		t.Fatalf("Mode=%q Prompt=%q", args.Mode, args.Prompt)
69	}
70}