Fixes #69

dave · Dec 9, 2022 · 5fa8d6e · 5fa8d6e
1 parent bc71a76
commit 5fa8d6e
Show file tree

Hide file tree

Showing 5 changed files with 286 additions and 212 deletions.
diff --git a/decorator/decorator_test.go b/decorator/decorator_test.go
@@ -18,6 +18,26 @@ func TestDecorator(t *testing.T) {
 		code       string
 		expect     string
 	}{
+		{
+			name: "package comment bug 3",
+			code: `/*
+1
+*/
+package a`,
+			expect: `File [Start "/*\n1\n*/" "\n"]`,
+		},
+		{
+			name: "package comment bug 2",
+			code: `// 1
+package a`,
+			expect: `File [Start "// 1"]`,
+		},
+		{
+			name: "package comment bug 1",
+			code: `/* 1 */
+package a`,
+			expect: `File [Start "/* 1 */" "\n"]`,
+		},
 		{
 			name: "index list expr",
 			code: `package a