Fix for

realm · Nov 26, 2024 · 1ebcdb7 · 1ebcdb7
1 parent 0a01cf0
commit 1ebcdb7
Show file tree

Hide file tree

Showing 2 changed files with 49 additions and 3 deletions.
diff --git a/Source/SwiftLintBuiltInRules/Rules/Lint/AsyncWithoutAwaitRule.swift b/Source/SwiftLintBuiltInRules/Rules/Lint/AsyncWithoutAwaitRule.swift
@@ -111,10 +111,19 @@ private extension AsyncWithoutAwaitRule {
             pendingAsync = nil
         }
 
+        override func visitPost(_ node: ForStmtSyntax) {
+            if node.awaitKeyword != nil {
+                functionScopes.modifyLast {
+                    $0.containsAwait = true
+                }
+            }
+        }
+
         private func checkViolation() {
-            guard let info = functionScopes.pop(),
-                    let asyncToken = info.asyncToken,
-                    !info.containsAwait
+            guard
+                let info = functionScopes.pop(),
+                let asyncToken = info.asyncToken,
+                !info.containsAwait
             else {
                 return
             }

diff --git a/Source/SwiftLintBuiltInRules/Rules/Lint/AsyncWithoutAwaitRuleExamples.swift b/Source/SwiftLintBuiltInRules/Rules/Lint/AsyncWithoutAwaitRuleExamples.swift
@@ -112,6 +112,28 @@ internal struct AsyncWithoutAwaitRuleExamples {
         typealias Bar = () async -> Void
         let baz = { qux() }
         """),
+        Example("""
+        func test() async {
+            for await foo in bar {}
+        }
+        """),
+        Example("""
+        func test() async {
+            while let foo = await bar() {}
+        }
+        """),
+        Example("""
+        func test() async {
+            async let foo = bar()
+            let baz = await foo
+        }
+        """),
+        Example("""
+        func test() async {
+            async let foo = bar()
+            await foo
+        }
+        """),
     ]
 
     static let triggeringExamples = [
@@ -189,6 +211,21 @@ internal struct AsyncWithoutAwaitRuleExamples {
             }
         }
         """),
+        Example("""
+        func test() ↓async {
+            for foo in bar {}
+        }
+        """),
+        Example("""
+        func test() ↓async {
+            while let foo = bar() {}
+        }
+        """),
+        Example("""
+        func test() ↓async {
+            async let foo = bar()
+        }
+        """),
     ]
 
     static let corrections = [