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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
|
package hexaiaction
import (
"bytes"
"context"
"log"
"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" }
type recordingActionStatusSink struct {
provider string
model string
}
func (s *recordingActionStatusSink) SetLLMStart(provider, model string) error {
s.provider = provider
s.model = model
return nil
}
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")
}
}
func TestRun_WithInjectedConfigAndStatusSink(t *testing.T) {
t.Setenv("XDG_CONFIG_HOME", t.TempDir())
sink := &recordingActionStatusSink{}
runner := NewRunner()
runner.loadConfig = func(context.Context, *log.Logger) appconfig.App { return appconfig.Load(nil) }
runner.newClient = func(_ appconfig.App) (actionClient, error) { return llmFake{}, nil }
runner.chooseAction = func(_ appconfig.App) (actionChoice, error) {
return actionChoice{kind: ActionSkip}, nil
}
runner.statusSink = sink
var out bytes.Buffer
if err := runner.Run(context.Background(), bytes.NewBufferString("selection"), &out, &bytes.Buffer{}); err != nil {
t.Fatalf("Run: %v", err)
}
if out.String() != "selection" {
t.Fatalf("unexpected output %q", out.String())
}
if sink.provider != "fake" || sink.model == "" {
t.Fatalf("unexpected status sink values: provider=%q model=%q", sink.provider, sink.model)
}
}
|
