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
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
|
package lsp
import (
"encoding/json"
"testing"
)
func TestResolveCodeAction_UsesRewritePrompts(t *testing.T) {
s := newTestServer()
cap := &captureLLM{}
s.llmClient = cap
cfg := s.cfg
cfg.PromptCodeActionRewriteSystem = "RSYS"
cfg.PromptCodeActionRewriteUser = "RUSER {{instruction}} {{selection}}"
s.cfg = cfg
uri := "file:///x.go"
s.setDocument(uri, "package p\nvar a=1\n")
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: Range{Start: Position{Line: 1}, End: Position{Line: 1, Character: 5}}, Instruction: "do it", Selection: "var a"}
raw, _ := json.Marshal(payload)
ca := CodeAction{Title: "Hexai: rewrite selection", Data: raw}
_, _ = s.resolveCodeAction(ca)
if len(cap.msgs) < 2 {
t.Fatalf("expected chat messages")
}
if cap.msgs[0].Content != "RSYS" || cap.msgs[1].Role != "user" || cap.msgs[1].Content != "RUSER do it var a" {
t.Fatalf("unexpected rewrite prompts: %#v", cap.msgs)
}
}
func TestResolveCodeAction_UsesDiagnosticsPrompts(t *testing.T) {
s := newTestServer()
cap := &captureLLM{}
s.llmClient = cap
cfg := s.cfg
cfg.PromptCodeActionDiagnosticsSystem = "DSYS"
cfg.PromptCodeActionDiagnosticsUser = "DUSER {{diagnostics}} {{selection}}"
s.cfg = cfg
uri := "file:///x.go"
s.setDocument(uri, "package p\nvar a=1\n")
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: Range{Start: Position{Line: 1}}, Selection: "var a", Diagnostics: []Diagnostic{{Message: "oops1"}, {Message: "oops2"}}}
raw, _ := json.Marshal(payload)
ca := CodeAction{Title: "Hexai: resolve diagnostics", Data: raw}
_, _ = s.resolveCodeAction(ca)
if len(cap.msgs) < 2 {
t.Fatalf("expected chat messages")
}
if cap.msgs[0].Content != "DSYS" || cap.msgs[1].Role != "user" {
t.Fatalf("unexpected diagnostics prompts: %#v", cap.msgs)
}
if got := cap.msgs[1].Content; !(contains(got, "oops1") && contains(got, "oops2") && contains(got, "var a")) {
t.Fatalf("diagnostics/user content mismatch: %q", got)
}
}
func TestResolveCodeAction_UsesDocumentPrompts(t *testing.T) {
s := newTestServer()
cap := &captureLLM{}
s.llmClient = cap
cfg := s.cfg
cfg.PromptCodeActionDocumentSystem = "DOCSYS"
cfg.PromptCodeActionDocumentUser = "DOCUSER {{selection}}"
s.cfg = cfg
uri := "file:///x.go"
s.setDocument(uri, "package p\nvar a=1\n")
payload := struct {
Type string `json:"type"`
URI string `json:"uri"`
Range Range `json:"range"`
Selection string `json:"selection"`
}{Type: "document", URI: uri, Range: Range{Start: Position{Line: 1}}, Selection: "var a"}
raw, _ := json.Marshal(payload)
ca := CodeAction{Title: "Hexai: document selection", Data: raw}
_, _ = s.resolveCodeAction(ca)
if len(cap.msgs) < 2 {
t.Fatalf("expected chat messages")
}
if cap.msgs[0].Content != "DOCSYS" || cap.msgs[1].Content != "DOCUSER var a" {
t.Fatalf("unexpected document prompts: %#v", cap.msgs)
}
}
func TestGenerateGoTest_UsesPrompts(t *testing.T) {
s := newTestServer()
cap := &captureLLM{}
s.llmClient = cap
cfg := s.cfg
cfg.PromptCodeActionGoTestSystem = "GTSYS"
cfg.PromptCodeActionGoTestUser = "GTUSER {{function}}"
s.cfg = cfg
_ = s.generateGoTestFunction("func Add(a,b int) int {return a+b}")
if len(cap.msgs) < 2 {
t.Fatalf("expected chat messages")
}
if cap.msgs[0].Content != "GTSYS" || !contains(cap.msgs[1].Content, "func Add") {
t.Fatalf("unexpected gotest prompts: %#v", cap.msgs)
}
}
|
