1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
|
// Summary: Test helpers for Hexai CLI tests (stdin swapping and fake LLM clients/streamers).
package hexaicli
import (
"context"
"os"
"path/filepath"
"testing"
"codeberg.org/snonux/hexai/internal/llm"
)
// setStdin sets os.Stdin from a string and returns a restore func and reader.
func setStdin(t *testing.T, content string) (func(), *os.File) {
t.Helper()
tmpDir := t.TempDir()
fpath := filepath.Join(tmpDir, "stdin.txt")
if err := os.WriteFile(fpath, []byte(content), 0o600); err != nil {
t.Fatalf("write temp stdin: %v", err)
}
f, err := os.Open(fpath)
if err != nil {
t.Fatalf("open temp stdin: %v", err)
}
old := os.Stdin
os.Stdin = f
restore := func() {
f.Close()
os.Stdin = old
}
return restore, f
}
// fakeClient implements llm.Client for tests.
type fakeClient struct {
name string
model string
resp string
gotMsgs []llm.Message
}
func (f *fakeClient) Chat(ctx context.Context, messages []llm.Message, opts ...llm.RequestOption) (string, error) {
f.gotMsgs = append([]llm.Message{}, messages...)
return f.resp, nil
}
func (f fakeClient) Name() string { return f.name }
func (f fakeClient) DefaultModel() string { return f.model }
// fakeStreamer implements llm.Streamer over fakeClient.
type fakeStreamer struct {
fakeClient
chunks []string
sMsgs []llm.Message
}
func (s *fakeStreamer) ChatStream(ctx context.Context, messages []llm.Message, onDelta func(string), opts ...llm.RequestOption) error {
s.sMsgs = append([]llm.Message{}, messages...)
for _, c := range s.chunks {
onDelta(c)
}
return nil
}
// small TOML writer for tests (string values only)
func writeTOML(t *testing.T, path string, m map[string]string) {
t.Helper()
if err := os.MkdirAll(filepath.Dir(path), 0o755); err != nil {
t.Fatalf("mkdir: %v", err)
}
f, err := os.Create(path)
if err != nil {
t.Fatalf("create: %v", err)
}
defer f.Close()
for k, v := range m {
if _, err := f.WriteString(k + " = \"" + v + "\"\n"); err != nil {
t.Fatalf("write: %v", err)
}
}
}
func writeConfigString(t *testing.T, path string, contents string) {
t.Helper()
if err := os.MkdirAll(filepath.Dir(path), 0o755); err != nil {
t.Fatalf("mkdir: %v", err)
}
if err := os.WriteFile(path, []byte(contents), 0o644); err != nil {
t.Fatalf("write: %v", err)
}
}
func testingTempDir(t *testing.T) string { t.Helper(); return t.TempDir() }
|
