summaryrefslogtreecommitdiff
path: root/internal/hexaiaction/custom_action_test.go
diff options
context:
space:
mode:
Diffstat (limited to 'internal/hexaiaction/custom_action_test.go')
-rw-r--r--internal/hexaiaction/custom_action_test.go59
1 files changed, 33 insertions, 26 deletions
diff --git a/internal/hexaiaction/custom_action_test.go b/internal/hexaiaction/custom_action_test.go
index 451a313..72cfbc4 100644
--- a/internal/hexaiaction/custom_action_test.go
+++ b/internal/hexaiaction/custom_action_test.go
@@ -1,39 +1,46 @@
package hexaiaction
import (
- "bytes"
- "context"
- "testing"
+ "bytes"
+ "context"
+ "os"
+ "testing"
- "codeberg.org/snonux/hexai/internal/appconfig"
- "codeberg.org/snonux/hexai/internal/editor"
- "codeberg.org/snonux/hexai/internal/llm"
- "os"
+ "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) 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 })
+ // 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")
+ 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") }
+ 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")
+ }
}