summaryrefslogtreecommitdiff
path: root/internal/hexaiaction/run_seam_test.go
blob: 9aa92bfa309d3ab04ac24d6fef32f665cdd2e434 (plain)
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
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())
	// 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
	var errBuf bytes.Buffer
	in := bytes.NewBufferString("some code")
	if err := 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
	chooseActionFn = func() (ActionKind, error) { return ActionRewrite, nil }
	out.Reset()
	in = bytes.NewBufferString(";upper;\nhello")
	if err := Run(context.Background(), in, &out, &errBuf); err != nil {
		t.Fatalf("Run rewrite: %v", err)
	}
	if out.String() == "" {
		t.Fatalf("expected non-empty rewrite output")
	}
}