Disable markTypes in fragments

Author: calda

Sources/Rules/MarkTypes.swift

@@ -18,9 +18,10 @@ public extension FormatRule {
     ) { formatter in
         var declarations = formatter.parseDeclarations()
 
-        // Do nothing if there is only one top-level declaration in the file (excluding imports)
+        // Do nothing if there is only one top-level declaration in the file (excluding imports),
+        // or if this is a fragment.
         let declarationsWithoutImports = declarations.filter { $0.keyword != "import" }
-        guard declarationsWithoutImports.count > 1 else {
+        guard declarationsWithoutImports.count > 1, !formatter.options.fragment else {
             return
         }
 

Tests/Rules/MarkTypesTests.swift

@@ -110,21 +110,14 @@ class MarkTypesTests: XCTestCase {
         testFormatting(for: input, output, rule: .markTypes, exclude: [.emptyExtensions])
     }
 
-    func testCustomTypeMark() {
+    func testFragment() {
         let input = """
         struct Foo {}
         extension Foo {}
         """
 
-        let output = """
-        // TYPE DEFINITION: Foo
-
-        struct Foo {}
-        extension Foo {}
-        """
-
         testFormatting(
-            for: input, output, rule: .markTypes,
+            for: input, rule: .markTypes,
             options: FormatOptions(typeMarkComment: "TYPE DEFINITION: %t", fragment: true),
             exclude: [.emptyExtensions]
         )