summaryrefslogtreecommitdiff
path: root/internal/hexaicli/testhelpers_test.go
diff options
context:
space:
mode:
Diffstat (limited to 'internal/hexaicli/testhelpers_test.go')
-rw-r--r--internal/hexaicli/testhelpers_test.go40
1 files changed, 23 insertions, 17 deletions
diff --git a/internal/hexaicli/testhelpers_test.go b/internal/hexaicli/testhelpers_test.go
index 1f75916..512a3ba 100644
--- a/internal/hexaicli/testhelpers_test.go
+++ b/internal/hexaicli/testhelpers_test.go
@@ -2,13 +2,13 @@
package hexaicli
import (
- "context"
- "encoding/json"
- "os"
- "path/filepath"
- "testing"
+ "context"
+ "encoding/json"
+ "os"
+ "path/filepath"
+ "testing"
- "codeberg.org/snonux/hexai/internal/llm"
+ "codeberg.org/snonux/hexai/internal/llm"
)
// setStdin sets os.Stdin from a string and returns a restore func and reader.
@@ -55,21 +55,27 @@ type fakeStreamer struct {
}
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
+ s.sMsgs = append([]llm.Message{}, messages...)
+ for _, c := range s.chunks {
+ onDelta(c)
+ }
+ return nil
}
// small JSON writer for tests
func writeJSON(t *testing.T, path string, v any) {
- 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()
- if err := json.NewEncoder(f).Encode(v); err != nil { t.Fatalf("encode: %v", err) }
+ 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()
+ if err := json.NewEncoder(f).Encode(v); err != nil {
+ t.Fatalf("encode: %v", err)
+ }
}
func testingTempDir(t *testing.T) string { t.Helper(); return t.TempDir() }