internal/lsp/chat_prompt_test.go


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

package lsp

import (
    "bytes"
    "testing"
    "time"
)

func TestDetectAndHandleChat_UsesConfiguredSystemPrompt(t *testing.T) {
    s := newTestServer()
    cap := &captureLLM{}
    s.llmClient = cap
    s.promptChatSystem = "CHAT-SYS"
    uri := "file:///chat.txt"
    // Avoid nil writer in applyChatEdits
    var out bytes.Buffer
    s.out = &out
    // Line that should trigger chat: ends with '>' and previous char in prefixes
    s.setDocument(uri, "help?>\n")
    s.detectAndHandleChat(uri)
    // Wait briefly for async goroutine to call Chat
    for i := 0; i < 20 && len(cap.msgs) == 0; i++ {
        time.Sleep(10 * time.Millisecond)
    }
    if len(cap.msgs) == 0 {
        t.Fatalf("expected Chat to be called")
    }
    if cap.msgs[0].Role != "system" || cap.msgs[0].Content != "CHAT-SYS" {
        t.Fatalf("unexpected system msg: %+v", cap.msgs[0])
    }
    // Last should be user with prompt without trailing '>'
    last := cap.msgs[len(cap.msgs)-1]
    if last.Role != "user" || last.Content != "help?" {
        t.Fatalf("unexpected last user msg: %+v", last)
    }
}