summaryrefslogtreecommitdiff
path: root/internal/hexaiaction/run_seam_test.go
diff options
context:
space:
mode:
Diffstat (limited to 'internal/hexaiaction/run_seam_test.go')
-rw-r--r--internal/hexaiaction/run_seam_test.go36
1 files changed, 36 insertions, 0 deletions
diff --git a/internal/hexaiaction/run_seam_test.go b/internal/hexaiaction/run_seam_test.go
new file mode 100644
index 0000000..0b8761f
--- /dev/null
+++ b/internal/hexaiaction/run_seam_test.go
@@ -0,0 +1,36 @@
+package hexaiaction
+
+import (
+ "bytes"
+ "context"
+ "testing"
+
+ "codeberg.org/snonux/hexai/internal/appconfig"
+ "codeberg.org/snonux/hexai/internal/llm"
+)
+
+type llmFake struct{}
+
+func (llmFake) Chat(_ context.Context, _ []llm.Message, _ ...llm.RequestOption) (string, error) { return "OK", nil }
+func (llmFake) Name() string { return "fake" }
+func (llmFake) DefaultModel() string { return "model" }
+
+func TestRun_WithSeams_SkipAndRewrite(t *testing.T) {
+ // Seam: choose action to Skip first, then Rewrite
+ oldChoose := chooseActionFn
+ oldNew := newClientFromApp
+ t.Cleanup(func(){ chooseActionFn = oldChoose; newClientFromApp = oldNew })
+ // 1) Skip -> echoes selection
+ chooseActionFn = func() (ActionKind, error) { return ActionSkip, nil }
+ newClientFromApp = func(_ appconfig.App) (llm.Client, error) { return llmFake{}, nil }
+ var out bytes.Buffer
+ in := bytes.NewBufferString("some code")
+ if err := Run(context.Background(), in, &out, &out); err != nil { t.Fatalf("Run skip: %v", err) }
+ if out.String() != "some code" { t.Fatalf("skip out: %q", out.String()) }
+ // 2) Rewrite -> requires inline instruction
+ chooseActionFn = func() (ActionKind, error) { return ActionRewrite, nil }
+ out.Reset()
+ in = bytes.NewBufferString(";upper;\nhello")
+ if err := Run(context.Background(), in, &out, &out); err != nil { t.Fatalf("Run rewrite: %v", err) }
+ if out.String() == "" { t.Fatalf("expected non-empty rewrite output") }
+}