wehub-resource-sync
63 lines
2.0 KiB
Go
63 lines
2.0 KiB
Go
package builtin
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
// TestDeleteSymbolRemovesDocComment probes whether deleting a documented symbol
|
|
// also removes its doc comment. AST node Pos() excludes the Doc, so a naive
|
|
// offset delete would orphan the comment above the gone function.
|
|
func TestDeleteSymbolRemovesDocComment(t *testing.T) {
|
|
dir := t.TempDir()
|
|
path := filepath.Join(dir, "f.go")
|
|
src := "package p\n\n// Foo does a thing.\n// Second line.\nfunc Foo() int { return 1 }\n\nfunc Bar() {}\n"
|
|
if err := os.WriteFile(path, []byte(src), 0o644); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
args, _ := json.Marshal(map[string]any{"path": path, "name": "Foo"})
|
|
if _, err := (deleteSymbol{}).Execute(context.Background(), args); err != nil {
|
|
t.Fatalf("execute: %v", err)
|
|
}
|
|
got, _ := os.ReadFile(path)
|
|
if strings.Contains(string(got), "Foo does a thing") {
|
|
t.Fatalf("doc comment orphaned after deleting Foo:\n%s", string(got))
|
|
}
|
|
}
|
|
|
|
// TestDeleteSymbolGroupedSpecKeepsGroupDoc proves deleting one spec from a
|
|
// parenthesized block removes that spec's own doc but never the block's group
|
|
// doc or its siblings.
|
|
func TestDeleteSymbolGroupedSpecKeepsGroupDoc(t *testing.T) {
|
|
dir := t.TempDir()
|
|
path := filepath.Join(dir, "g.go")
|
|
src := "package p\n\n// Group doc.\nconst (\n\t// ADoc.\n\tA = 1\n\tB = 2\n)\n"
|
|
if err := os.WriteFile(path, []byte(src), 0o644); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
args, _ := json.Marshal(map[string]any{"path": path, "name": "A"})
|
|
if _, err := (deleteSymbol{}).Execute(context.Background(), args); err != nil {
|
|
t.Fatalf("execute: %v", err)
|
|
}
|
|
got := string(mustRead(t, path))
|
|
if strings.Contains(got, "ADoc") || strings.Contains(got, "A = 1") {
|
|
t.Fatalf("A and its own doc should be gone:\n%s", got)
|
|
}
|
|
if !strings.Contains(got, "Group doc.") || !strings.Contains(got, "B = 2") {
|
|
t.Fatalf("group doc and sibling B must remain:\n%s", got)
|
|
}
|
|
}
|
|
|
|
func mustRead(t *testing.T, path string) []byte {
|
|
t.Helper()
|
|
b, err := os.ReadFile(path)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
return b
|
|
}
|