Fix issue where organizeDeclarations rule would unexpectedly remove n…

…on-mark comment
nicklockwood · Jan 3, 2025 · e2da757 · e2da757
1 parent 6db7324
commit e2da757
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git a/Sources/Rules/OrganizeDeclarations.swift b/Sources/Rules/OrganizeDeclarations.swift
@@ -524,8 +524,7 @@ extension Formatter {
             Array(Set([
                 // The user's specific category separator template
                 $0.markCommentBody(from: options.categoryMarkComment, with: options.organizationMode),
-                // Other common variants that we would want to replace with the correct variant
-                $0.markCommentBody(from: "%c", with: options.organizationMode),
+                // Always look for MARKs even if the user is using a different template
                 $0.markCommentBody(from: "MARK: %c", with: options.organizationMode),
             ]))
         }.compactMap { $0 }

diff --git a/Tests/Rules/OrganizeDeclarationsTests.swift b/Tests/Rules/OrganizeDeclarationsTests.swift
@@ -3593,4 +3593,17 @@ class OrganizeDeclarationsTests: XCTestCase {
 
         testFormatting(for: input, output, rule: .organizeDeclarations, exclude: [.blankLinesAtStartOfScope])
     }
+
+    func testPreservesUnrelatedComments() {
+        let input = """
+        enum Test {
+            /// Test Properties
+            static let foo = "foo"
+            static let bar = "bar"
+            static let baaz = "baaz"
+        }
+        """
+
+        testFormatting(for: input, rule: .organizeDeclarations)
+    }
 }