leave comment blocks before imports

incu6us · May 25, 2020 · d9dd0f9 · d9dd0f9
1 parent 946ab5a
commit d9dd0f9
Show file tree

Hide file tree

Showing 2 changed files with 39 additions and 1 deletion.
diff --git a/reviser/reviser.go b/reviser/reviser.go
@@ -281,7 +281,7 @@ type importPosition struct {
 }
 
 func (p *importPosition) IsInRange(comment *ast.CommentGroup) bool {
-	if p.Start >= comment.Pos() || comment.Pos() <= p.End {
+	if p.Start <= comment.Pos() && comment.Pos() <= p.End {
 		return true
 	}
 

diff --git a/reviser/reviser_test.go b/reviser/reviser_test.go
@@ -421,6 +421,44 @@ import (
 	_ "github.com/incu6us/goimports-reviser/testdata/innderpkg" // custom package
 )
 
+// nolint:gomnd
+func main() {
+	_ = fmt.Println("test")
+}
+`,
+			wantChange: true,
+			wantErr:    false,
+		},
+		{
+			name: "success with comments before imports",
+			args: args{
+				projectName: "github.com/incu6us/goimports-reviser",
+				filePath:    "./testdata/example.go",
+				fileContent: `// Some comments are here
+package testdata
+
+// test
+import (
+	"fmt" //fmt package
+	_ "github.com/incu6us/goimports-reviser/testdata/innderpkg" //custom package
+)
+
+// nolint:gomnd
+func main(){
+  _ = fmt.Println("test")
+}
+`,
+			},
+			want: `// Some comments are here
+package testdata
+
+// test
+import (
+	"fmt" // fmt package
+
+	_ "github.com/incu6us/goimports-reviser/testdata/innderpkg" // custom package
+)
+
 // nolint:gomnd
 func main() {
 	_ = fmt.Println("test")