summaryrefslogtreecommitdiff
path: root/internal/hexaicli/editor_integration_test.go
diff options
context:
space:
mode:
Diffstat (limited to 'internal/hexaicli/editor_integration_test.go')
-rw-r--r--internal/hexaicli/editor_integration_test.go83
1 files changed, 47 insertions, 36 deletions
diff --git a/internal/hexaicli/editor_integration_test.go b/internal/hexaicli/editor_integration_test.go
index 1bebf75..7d53d1f 100644
--- a/internal/hexaicli/editor_integration_test.go
+++ b/internal/hexaicli/editor_integration_test.go
@@ -1,51 +1,62 @@
package hexaicli
import (
- "bytes"
- "context"
- "os"
- "testing"
+ "bytes"
+ "context"
+ "os"
+ "testing"
- "codeberg.org/snonux/hexai/internal/appconfig"
- "codeberg.org/snonux/hexai/internal/editor"
- "codeberg.org/snonux/hexai/internal/llm"
+ "codeberg.org/snonux/hexai/internal/appconfig"
+ "codeberg.org/snonux/hexai/internal/editor"
+ "codeberg.org/snonux/hexai/internal/llm"
)
type cliFake struct{}
-func (cliFake) Chat(_ context.Context, _ []llm.Message, _ ...llm.RequestOption) (string, error) { return "OUT", nil }
-func (cliFake) Name() string { return "fake" }
+
+func (cliFake) Chat(_ context.Context, _ []llm.Message, _ ...llm.RequestOption) (string, error) {
+ return "OUT", nil
+}
+func (cliFake) Name() string { return "fake" }
func (cliFake) DefaultModel() string { return "m" }
-func (cliFake) CodeCompletion(context.Context, string, string, int, string, float64) ([]string, error) { return nil, nil }
+func (cliFake) CodeCompletion(context.Context, string, string, int, string, float64) ([]string, error) {
+ return nil, nil
+}
func TestRun_NoArgs_OpensEditor(t *testing.T) {
- // Seam: fake client and editor
- oldNew := newClientFromApp
- newClientFromApp = func(_ appconfig.App) (llm.Client, error) { return cliFake{}, nil }
- t.Cleanup(func(){ newClientFromApp = oldNew })
- oldRun := editor.RunEditor
- editor.RunEditor = func(_ string, path string) error { return os.WriteFile(path, []byte("PROMPT"), 0o600) }
- t.Cleanup(func(){ editor.RunEditor = oldRun })
- t.Setenv("HEXAI_EDITOR", "dummy")
+ // Seam: fake client and editor
+ oldNew := newClientFromApp
+ newClientFromApp = func(_ appconfig.App) (llm.Client, error) { return cliFake{}, nil }
+ t.Cleanup(func() { newClientFromApp = oldNew })
+ oldRun := editor.RunEditor
+ editor.RunEditor = func(_ string, path string) error { return os.WriteFile(path, []byte("PROMPT"), 0o600) }
+ t.Cleanup(func() { editor.RunEditor = oldRun })
+ t.Setenv("HEXAI_EDITOR", "dummy")
- // Provide stdin selection
- var stdout, stderr bytes.Buffer
- if err := Run(context.Background(), nil, bytes.NewBufferString("SELECTION"), &stdout, &stderr); err != nil {
- t.Fatalf("Run: %v", err)
- }
- if stdout.String() == "" { t.Fatalf("expected some output") }
+ // Provide stdin selection
+ var stdout, stderr bytes.Buffer
+ if err := Run(context.Background(), nil, bytes.NewBufferString("SELECTION"), &stdout, &stderr); err != nil {
+ t.Fatalf("Run: %v", err)
+ }
+ if stdout.String() == "" {
+ t.Fatalf("expected some output")
+ }
}
func TestRun_WithArgs_DoesNotOpenEditor(t *testing.T) {
- // Provide args; still use fake client
- oldNew := newClientFromApp
- newClientFromApp = func(_ appconfig.App) (llm.Client, error) { return cliFake{}, nil }
- t.Cleanup(func(){ newClientFromApp = oldNew })
- // Stub editor and detect if called (should not be)
- called := false
- oldRun := editor.RunEditor
- editor.RunEditor = func(_ string, _ string) error { called = true; return nil }
- t.Cleanup(func(){ editor.RunEditor = oldRun })
- var stdout, stderr bytes.Buffer
- if err := Run(context.Background(), []string{"ARG"}, bytes.NewBufferString("SEL"), &stdout, &stderr); err != nil { t.Fatalf("Run: %v", err) }
- if called { t.Fatalf("editor should not be invoked when args provided") }
+ // Provide args; still use fake client
+ oldNew := newClientFromApp
+ newClientFromApp = func(_ appconfig.App) (llm.Client, error) { return cliFake{}, nil }
+ t.Cleanup(func() { newClientFromApp = oldNew })
+ // Stub editor and detect if called (should not be)
+ called := false
+ oldRun := editor.RunEditor
+ editor.RunEditor = func(_ string, _ string) error { called = true; return nil }
+ t.Cleanup(func() { editor.RunEditor = oldRun })
+ var stdout, stderr bytes.Buffer
+ if err := Run(context.Background(), []string{"ARG"}, bytes.NewBufferString("SEL"), &stdout, &stderr); err != nil {
+ t.Fatalf("Run: %v", err)
+ }
+ if called {
+ t.Fatalf("editor should not be invoked when args provided")
+ }
}