diff options
Diffstat (limited to 'internal/lsp/provider_native_success_test.go')
| -rw-r--r-- | internal/lsp/provider_native_success_test.go | 66 |
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) + } } |
