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
|
package lsp
import (
"encoding/json"
"testing"
)
func TestResolveRewrite_MultiLine_PreservesRange(t *testing.T) {
s := newTestServer()
s.llmClient = fakeLLM{resp: "line1\nline2"}
uri := "file:///x.go"
s.setDocument(uri, "package p\nvar a=1\n")
r := Range{Start: Position{Line: 1, Character: 0}, End: Position{Line: 1, Character: 5}}
payload := struct {
Type string `json:"type"`
URI string `json:"uri"`
Range Range `json:"range"`
Instruction string `json:"instruction"`
Selection string `json:"selection"`
}{Type: "rewrite", URI: uri, Range: r, Instruction: "expand", Selection: "var a"}
raw, _ := json.Marshal(payload)
ca := CodeAction{Title: "Hexai: rewrite selection", Data: raw}
resolved, ok := s.resolveCodeAction(ca)
if !ok || resolved.Edit == nil {
t.Fatalf("expected resolved rewrite edit")
}
edits := resolved.Edit.Changes[uri]
if len(edits) != 1 {
t.Fatalf("expected 1 edit")
}
if edits[0].Range != r {
t.Fatalf("range mismatch: got %+v want %+v", edits[0].Range, r)
}
if edits[0].NewText == "" || !containsNewline(edits[0].NewText) {
t.Fatalf("expected multi-line replacement text, got %q", edits[0].NewText)
}
}
func TestResolveDiagnostics_MultiLine_PreservesRange(t *testing.T) {
s := newTestServer()
s.llmClient = fakeLLM{resp: "fixed\nvalue"}
uri := "file:///x.go"
s.setDocument(uri, "package p\nvar x = 1\n")
r := Range{Start: Position{Line: 1, Character: 0}, End: Position{Line: 1, Character: 10}}
payload := struct {
Type string `json:"type"`
URI string `json:"uri"`
Range Range `json:"range"`
Selection string `json:"selection"`
Diagnostics []Diagnostic `json:"diagnostics"`
}{Type: "diagnostics", URI: uri, Range: r, Selection: "var x = 1", Diagnostics: []Diagnostic{{Range: Range{Start: Position{Line: 1}, End: Position{Line: 1, Character: 5}}, Message: "msg"}}}
raw, _ := json.Marshal(payload)
ca := CodeAction{Title: "Hexai: resolve diagnostics", Data: raw}
resolved, ok := s.resolveCodeAction(ca)
if !ok || resolved.Edit == nil {
t.Fatalf("expected resolved diagnostics edit")
}
edits := resolved.Edit.Changes[uri]
if len(edits) != 1 {
t.Fatalf("expected 1 edit")
}
if edits[0].Range != r {
t.Fatalf("range mismatch: got %+v want %+v", edits[0].Range, r)
}
if edits[0].NewText == "" || !containsNewline(edits[0].NewText) {
t.Fatalf("expected multi-line replacement text, got %q", edits[0].NewText)
}
}
func containsNewline(s string) bool {
for i := 0; i < len(s); i++ {
if s[i] == '\n' {
return true
}
}
return false
}
|
