Merge pull request swiftlang#4 from allevato/sr-11112

Ensure a space after `case` in a `for case ... in` statement.

Author: allevato

Sources/SwiftFormatPrettyPrint/TokenStreamCreator.swift

@@ -422,6 +422,7 @@ private final class TokenStreamCreator: SyntaxVisitor {
   func visit(_ node: ForInStmtSyntax) -> SyntaxVisitorContinueKind {
     after(node.labelColon, tokens: .space)
     after(node.forKeyword, tokens: .space)
+    after(node.caseKeyword, tokens: .space)
     before(node.inKeyword, tokens: .break)
     after(node.inKeyword, tokens: .space)
 

Tests/SwiftFormatPrettyPrintTests/ForInStmtTests.swift

@@ -163,4 +163,27 @@ public class ForInStmtTests: PrettyPrintTestCase {
 
     assertPrettyPrintEqual(input: input, expected: expected, linelength: 40)
   }
+
+  public func testForCase() {
+    let input =
+      """
+      for case let a as String in [] {
+        let a = 123
+        print(i)
+      }
+      """
+
+    let expected =
+      """
+      for case let a
+        as String in []
+      {
+        let a = 123
+        print(i)
+      }
+
+      """
+
+    assertPrettyPrintEqual(input: input, expected: expected, linelength: 20)
+  }
 }

Tests/SwiftFormatPrettyPrintTests/XCTestManifests.swift

@@ -175,6 +175,7 @@ extension ForInStmtTests {
     // to regenerate.
     static let __allTests__ForInStmtTests = [
         ("testBasicForLoop", testBasicForLoop),
+        ("testForCase", testForCase),
         ("testForLabels", testForLabels),
         ("testForLoopFullWrap", testForLoopFullWrap),
         ("testForWhereLoop", testForWhereLoop),