Fixes #1005, trailing trivia preserved when replacing closures (#1006)

Author: seanwoodward

Sources/InlineSnapshotTesting/AssertInlineSnapshot.swift

@@ -554,13 +554,25 @@ public struct InlineSnapshotSyntaxDescriptor: Hashable {
         case 0:
           if snapshot.wasRecording || functionCallExpr.trailingClosure == nil {
             functionCallExpr.rightParen?.trailingTrivia = .space
+            let trailingClosureTrivia = functionCallExpr.trailingClosure?.trailingTrivia
             if let snapshotClosure {
-              functionCallExpr.trailingClosure = snapshotClosure  // FIXME: ?? multipleTrailingClosures.removeFirst()
+              // FIXME: ?? multipleTrailingClosures.removeFirst()
+              functionCallExpr.trailingClosure =
+                if let trailingClosureTrivia, trailingClosureTrivia.count > 0 {
+                  snapshotClosure.with(\.trailingTrivia, snapshotClosure.trailingTrivia + trailingClosureTrivia)
+                } else {
+                  snapshotClosure
+                }
             } else if !functionCallExpr.additionalTrailingClosures.isEmpty {
               let additionalTrailingClosure = functionCallExpr.additionalTrailingClosures.remove(
                 at: functionCallExpr.additionalTrailingClosures.startIndex
               )
-              functionCallExpr.trailingClosure = additionalTrailingClosure.closure
+              functionCallExpr.trailingClosure =
+                if let trailingClosureTrivia, trailingClosureTrivia.count > 0 {
+                  additionalTrailingClosure.closure.with(\.trailingTrivia, additionalTrailingClosure.closure.trailingTrivia + trailingClosureTrivia)
+                } else {
+                  additionalTrailingClosure.closure
+                }
             } else {
               functionCallExpr.rightParen?.trailingTrivia = ""
               functionCallExpr.trailingClosure = nil
@@ -599,7 +611,13 @@ public struct InlineSnapshotSyntaxDescriptor: Hashable {
               if snapshot.wasRecording {
                 if let snapshotClosure {
                   functionCallExpr.additionalTrailingClosures[index].label = snapshotLabel
-                  functionCallExpr.additionalTrailingClosures[index].closure = snapshotClosure
+                  let trailingTrivia = functionCallExpr.additionalTrailingClosures[index].closure.trailingTrivia
+                  functionCallExpr.additionalTrailingClosures[index].closure =
+                    if trailingTrivia.count > 0 {
+                      snapshotClosure.with(\.trailingTrivia, snapshotClosure.trailingTrivia + trailingTrivia)
+                    } else {
+                      snapshotClosure
+                    }
                 } else {
                   functionCallExpr.additionalTrailingClosures.remove(at: index)
                 }