8 "golang.org/x/tools/internal/lsp/diff"
9 "golang.org/x/tools/internal/lsp/diff/myers"
12 func TestImportPrefix(t *testing.T) {
13 for i, tt := range []struct {
16 {"package foo", "package foo"},
17 {"package foo\n", "package foo\n"},
18 {"package foo\n\nfunc f(){}\n", "package foo\n"},
19 {"package foo\n\nimport \"fmt\"\n", "package foo\n\nimport \"fmt\""},
20 {"package foo\nimport (\n\"fmt\"\n)\n", "package foo\nimport (\n\"fmt\"\n)"},
21 {"\n\n\npackage foo\n", "\n\n\npackage foo\n"},
22 {"// hi \n\npackage foo //xx\nfunc _(){}\n", "// hi \n\npackage foo //xx\n"},
23 {"package foo //hi\n", "package foo //hi\n"},
24 {"//hi\npackage foo\n//a\n\n//b\n", "//hi\npackage foo\n//a\n\n//b\n"},
26 "package a\n\nimport (\n \"fmt\"\n)\n//hi\n",
27 "package a\n\nimport (\n \"fmt\"\n)\n//hi\n",
29 {`package a /*hi*/`, `package a /*hi*/`},
30 {"package main\r\n\r\nimport \"go/types\"\r\n\r\n/*\r\n\r\n */\r\n", "package main\r\n\r\nimport \"go/types\"\r\n\r\n/*\r\n\r\n */\r\n"},
31 {"package x; import \"os\"; func f() {}\n\n", "package x; import \"os\""},
32 {"package x; func f() {fmt.Println()}\n\n", "package x"},
34 got := importPrefix([]byte(tt.input))
36 t.Errorf("%d: failed for %q:\n%s", i, tt.input, diffStr(tt.want, got))
41 func TestCRLFFile(t *testing.T) {
42 for i, tt := range []struct {
61 got := importPrefix([]byte(strings.ReplaceAll(tt.input, "\n", "\r\n")))
62 want := strings.ReplaceAll(tt.want, "\n", "\r\n")
64 t.Errorf("%d: failed for %q:\n%s", i, tt.input, diffStr(want, got))
69 func diffStr(want, got string) string {
73 // Add newlines to avoid newline messages in diff.
76 d := myers.ComputeEdits("", want, got)
77 return fmt.Sprintf("%q", diff.ToUnified("want", "got", want, d))