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
|
package hexaiaction
import (
"bytes"
"context"
"os"
"testing"
"codeberg.org/snonux/hexai/internal/appconfig"
"codeberg.org/snonux/hexai/internal/editor"
"codeberg.org/snonux/hexai/internal/llm"
)
type llmFake2 struct{}
func (llmFake2) Chat(_ context.Context, _ []llm.Message, _ ...llm.RequestOption) (string, error) {
return "DONE", nil
}
func (llmFake2) Name() string { return "fake" }
func (llmFake2) DefaultModel() string { return "m" }
func TestActionCustom_UsesEditorPrompt(t *testing.T) {
// Seam: choose custom, fake client, and fake editor
oldChoose := chooseActionFn
oldNew := newClientFromApp
chooseActionFn = func() (ActionKind, error) { return ActionCustom, nil }
newClientFromApp = func(_ appconfig.App) (llm.Client, error) { return llmFake2{}, nil }
t.Cleanup(func() { chooseActionFn = oldChoose; newClientFromApp = oldNew })
oldRunEd := editor.RunEditor
editor.RunEditor = func(_ string, path string) error {
return os.WriteFile(path, []byte("make it done"), 0o600)
}
t.Cleanup(func() { editor.RunEditor = oldRunEd })
t.Setenv("HEXAI_EDITOR", "dummy")
in := bytes.NewBufferString("some code")
var out bytes.Buffer
var errb bytes.Buffer
if err := Run(context.Background(), in, &out, &errb); err != nil {
t.Fatalf("Run: %v", err)
}
if out.String() == "" {
t.Fatalf("expected output")
}
}
|
