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
|
// Summary: Test helpers for Hexai CLI tests (stdin swapping and fake LLM clients/streamers).
// Not yet reviewed by a human
package hexaicli
import (
"context"
"os"
"path/filepath"
"testing"
"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
}
|
