summaryrefslogtreecommitdiff
path: root/internal/lsp/provider_native_success_test.go
diff options
context:
space:
mode:
Diffstat (limited to 'internal/lsp/provider_native_success_test.go')
-rw-r--r--internal/lsp/provider_native_success_test.go66
1 files changed, 37 insertions, 29 deletions
diff --git a/internal/lsp/provider_native_success_test.go b/internal/lsp/provider_native_success_test.go
index fd7afad..dd1abcd 100644
--- a/internal/lsp/provider_native_success_test.go
+++ b/internal/lsp/provider_native_success_test.go
@@ -1,54 +1,62 @@
package lsp
import (
- "context"
- "testing"
+ "context"
+ "testing"
- "codeberg.org/snonux/hexai/internal/llm"
+ "codeberg.org/snonux/hexai/internal/llm"
)
type fakeCompleterOk struct{}
-func (fakeCompleterOk) Chat(context.Context, []llm.Message, ...llm.RequestOption) (string, error) { return "", nil }
+func (fakeCompleterOk) Chat(context.Context, []llm.Message, ...llm.RequestOption) (string, error) {
+ return "", nil
+}
func (fakeCompleterOk) Name() string { return "prov" }
func (fakeCompleterOk) DefaultModel() string { return "m" }
func (fakeCompleterOk) CodeCompletion(context.Context, string, string, int, string, float64) ([]string, error) {
- return []string{"SUGG"}, nil
+ return []string{"SUGG"}, nil
}
func TestProviderNativeCompletion_Success(t *testing.T) {
- s := newTestServer()
- s.llmClient = fakeCompleterOk{}
- // current line with dot trigger; position after dot
- current := "fmt."
- p := CompletionParams{TextDocument: TextDocumentIdentifier{URI: "file:///x.go"}, Position: Position{Line: 0, Character: len(current)}}
- items, ok := s.tryProviderNativeCompletion(current, p, "", "", "func f(){}", "doc", false, "", false)
- if !ok || len(items) == 0 {
- t.Fatalf("expected provider-native items")
- }
- if items[0].Label == "" || items[0].TextEdit == nil {
- t.Fatalf("unexpected completion item: %+v", items[0])
- }
+ s := newTestServer()
+ s.llmClient = fakeCompleterOk{}
+ // current line with dot trigger; position after dot
+ current := "fmt."
+ p := CompletionParams{TextDocument: TextDocumentIdentifier{URI: "file:///x.go"}, Position: Position{Line: 0, Character: len(current)}}
+ items, ok := s.tryProviderNativeCompletion(current, p, "", "", "func f(){}", "doc", false, "", false)
+ if !ok || len(items) == 0 {
+ t.Fatalf("expected provider-native items")
+ }
+ if items[0].Label == "" || items[0].TextEdit == nil {
+ t.Fatalf("unexpected completion item: %+v", items[0])
+ }
}
type fakeCompleterIndent struct{}
-func (fakeCompleterIndent) Chat(context.Context, []llm.Message, ...llm.RequestOption) (string, error) { return "", nil }
+func (fakeCompleterIndent) Chat(context.Context, []llm.Message, ...llm.RequestOption) (string, error) {
+ return "", nil
+}
func (fakeCompleterIndent) Name() string { return "prov" }
func (fakeCompleterIndent) DefaultModel() string { return "m" }
func (fakeCompleterIndent) CodeCompletion(context.Context, string, string, int, string, float64) ([]string, error) {
- return []string{"a\nb"}, nil
+ return []string{"a\nb"}, nil
}
func TestProviderNativeCompletion_IndentWithDoubleOpen(t *testing.T) {
- s := newTestServer()
- s.llmClient = fakeCompleterIndent{}
- current := " >>do>" // leading indent + double-open marker
- p := CompletionParams{TextDocument: TextDocumentIdentifier{URI: "file:///x.go"}, Position: Position{Line: 0, Character: len(current)}}
- items, ok := s.tryProviderNativeCompletion(current, p, "", "", "func f(){}", "doc", false, "", false)
- if !ok || len(items) == 0 { t.Fatalf("expected provider-native items") }
- if items[0].TextEdit == nil { t.Fatalf("expected text edit") }
- if got := items[0].TextEdit.NewText; len(got) < 2 || got[:2] != " " {
- t.Fatalf("expected indentation applied, got %q", got)
- }
+ s := newTestServer()
+ s.llmClient = fakeCompleterIndent{}
+ current := " >>do>" // leading indent + double-open marker
+ p := CompletionParams{TextDocument: TextDocumentIdentifier{URI: "file:///x.go"}, Position: Position{Line: 0, Character: len(current)}}
+ items, ok := s.tryProviderNativeCompletion(current, p, "", "", "func f(){}", "doc", false, "", false)
+ if !ok || len(items) == 0 {
+ t.Fatalf("expected provider-native items")
+ }
+ if items[0].TextEdit == nil {
+ t.Fatalf("expected text edit")
+ }
+ if got := items[0].TextEdit.NewText; len(got) < 2 || got[:2] != " " {
+ t.Fatalf("expected indentation applied, got %q", got)
+ }
}