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
|
package lsp
import (
"encoding/json"
"testing"
)
// Ensure a visible busy item is returned when a prior LLM request is in flight.
func TestLLMBusy_YieldsBusyCompletionItem(t *testing.T) {
s := &Server{ maxTokens: 32, triggerChars: []string{"."}, compCache: make(map[string]string) }
s.llmClient = &countingLLM{}
// Mark busy
s.setLLMBusy(true)
t.Cleanup(func(){ s.setLLMBusy(false) })
line := "obj."
p := CompletionParams{ Position: Position{ Line: 0, Character: len(line) }, TextDocument: TextDocumentIdentifier{URI: "file://busy.go"} }
// Simulate manual invoke to bypass min-prefix
p.Context = json.RawMessage([]byte(`{"triggerKind":1}`))
items, ok := s.tryLLMCompletion(p, "", line, "", "", "", false, "")
if !ok { t.Fatalf("expected ok=true") }
if len(items) != 1 { t.Fatalf("expected one busy item, got %d", len(items)) }
if items[0].InsertText != "" { t.Fatalf("busy item should not insert text") }
if items[0].Label == "" { t.Fatalf("busy item should have a label") }
}
|