diff options
Diffstat (limited to 'internal/lsp/build_prompts_table_test.go')
| -rw-r--r-- | internal/lsp/build_prompts_table_test.go | 36 |
1 files changed, 18 insertions, 18 deletions
diff --git a/internal/lsp/build_prompts_table_test.go b/internal/lsp/build_prompts_table_test.go index 06a3743..bc4f031 100644 --- a/internal/lsp/build_prompts_table_test.go +++ b/internal/lsp/build_prompts_table_test.go @@ -3,22 +3,22 @@ package lsp import "testing" func TestBuildPrompts_Table(t *testing.T) { - p := CompletionParams{TextDocument: TextDocumentIdentifier{URI: "file:///x.go"}, Position: Position{Line: 5, Character: 7}} - cases := []struct { - name string - inParams bool - }{ - {"generic", false}, - {"in_params", true}, - } - for _, c := range cases { - s := newTestServer() - msgs := s.buildCompletionMessages(false, false, "", c.inParams, p, "above", "current", "below", "func ctx") - if len(msgs) < 2 || msgs[0].Role != "system" || msgs[1].Role != "user" { - t.Fatalf("%s: unexpected messages", c.name) - } - if msgs[0].Content == "" || msgs[1].Content == "" { - t.Fatalf("%s: prompts empty", c.name) - } - } + p := CompletionParams{TextDocument: TextDocumentIdentifier{URI: "file:///x.go"}, Position: Position{Line: 5, Character: 7}} + cases := []struct { + name string + inParams bool + }{ + {"generic", false}, + {"in_params", true}, + } + for _, c := range cases { + s := newTestServer() + msgs := s.buildCompletionMessages(false, false, "", c.inParams, p, "above", "current", "below", "func ctx") + if len(msgs) < 2 || msgs[0].Role != "system" || msgs[1].Role != "user" { + t.Fatalf("%s: unexpected messages", c.name) + } + if msgs[0].Content == "" || msgs[1].Content == "" { + t.Fatalf("%s: prompts empty", c.name) + } + } } |
