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
47
48
49
50
|
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) {
// Isolate from user config to avoid environment-dependent behavior/logging.
t.Setenv("XDG_CONFIG_HOME", t.TempDir())
runner := NewRunner()
runner.newClient = func(_ appconfig.App) (actionClient, error) { return llmFake{}, nil }
// 1) Skip -> echoes selection
runner.chooseAction = func(_ appconfig.App) (actionChoice, error) {
return actionChoice{kind: ActionSkip}, nil
}
var out bytes.Buffer
var errBuf bytes.Buffer
in := bytes.NewBufferString("some code")
if err := runner.Run(context.Background(), in, &out, &errBuf); 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
runner.chooseAction = func(_ appconfig.App) (actionChoice, error) {
return actionChoice{kind: ActionRewrite}, nil
}
out.Reset()
in = bytes.NewBufferString(";upper;\nhello")
if err := runner.Run(context.Background(), in, &out, &errBuf); err != nil {
t.Fatalf("Run rewrite: %v", err)
}
if out.String() == "" {
t.Fatalf("expected non-empty rewrite output")
}
}
|
