diff options
Diffstat (limited to 'internal/lsp/gotest_append_test.go')
| -rw-r--r-- | internal/lsp/gotest_append_test.go | 51 |
1 files changed, 30 insertions, 21 deletions
diff --git a/internal/lsp/gotest_append_test.go b/internal/lsp/gotest_append_test.go index 4fff684..7ceb9e6 100644 --- a/internal/lsp/gotest_append_test.go +++ b/internal/lsp/gotest_append_test.go @@ -1,28 +1,37 @@ package lsp import ( - "os" - "path/filepath" - "strings" - "testing" + "os" + "path/filepath" + "strings" + "testing" ) func TestResolveGoTest_AppendsToExisting(t *testing.T) { - s := newTestServer() - dir := t.TempDir() - src := filepath.Join(dir, "m.go") - uri := "file://" + src - s.setDocument(uri, "package m\n\nfunc F(){}\n") - // Create existing test file - testPath := filepath.Join(dir, "m_test.go") - if err := os.WriteFile(testPath, []byte("package m\n\nimport \"testing\"\n\n"), 0o644); err != nil { t.Fatal(err) } - // LLM path to increase generateGoTestFunction coverage - s.llmClient = fakeLLM{resp: "func TestF(t *testing.T) {}"} - we, testURI, jump, ok := s.resolveGoTest(uri, Position{Line:2}) - if !ok || len(we.Changes) == 0 { t.Fatalf("expected append edit") } - if !strings.HasSuffix(testURI, "_test.go") { t.Fatalf("unexpected uri: %s", testURI) } - edits := we.Changes[testURI] - if len(edits) != 1 || !strings.Contains(edits[0].NewText, "TestF") { t.Fatalf("expected append with TestF") } - if jump.Start.Line < 0 { t.Fatalf("expected non-negative jump line") } + s := newTestServer() + dir := t.TempDir() + src := filepath.Join(dir, "m.go") + uri := "file://" + src + s.setDocument(uri, "package m\n\nfunc F(){}\n") + // Create existing test file + testPath := filepath.Join(dir, "m_test.go") + if err := os.WriteFile(testPath, []byte("package m\n\nimport \"testing\"\n\n"), 0o644); err != nil { + t.Fatal(err) + } + // LLM path to increase generateGoTestFunction coverage + s.llmClient = fakeLLM{resp: "func TestF(t *testing.T) {}"} + we, testURI, jump, ok := s.resolveGoTest(uri, Position{Line: 2}) + if !ok || len(we.Changes) == 0 { + t.Fatalf("expected append edit") + } + if !strings.HasSuffix(testURI, "_test.go") { + t.Fatalf("unexpected uri: %s", testURI) + } + edits := we.Changes[testURI] + if len(edits) != 1 || !strings.Contains(edits[0].NewText, "TestF") { + t.Fatalf("expected append with TestF") + } + if jump.Start.Line < 0 { + t.Fatalf("expected non-negative jump line") + } } - |
