summaryrefslogtreecommitdiff
path: root/internal/lsp/llm_busy_test.go
blob: 95123d29a54fbd914bbfbc5b5c1faaf64fe43817 (plain)
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") }
}