Swift: CFG test for for-try-await

Author: Robert Marsh

swift/ql/test/library-tests/controlflow/graph/Cfg.expected

@@ -1458,44 +1458,26 @@ cfg.swift:
 #  141| case ...
 #-----|  -> =~ ...
 
-#  141| $match
-#-----|  -> ... ~=(_:_:) ...
-
-#  141| ... ~=(_:_:) ...
-#-----|  -> =~ ...
-
 #  141| 0
-#-----|  -> $match
+#-----|  -> =~ ...
 
 #  141| =~ ...
-#-----|  -> ~=(_:_:)
+#-----|  -> 0
 
 #  141| =~ ...
 #-----| no-match -> =~ ...
 #-----| match -> true
 
-#  141| ~=(_:_:)
-#-----|  -> 0
-
-#  141| $match
-#-----|  -> ... ~=(_:_:) ...
-
-#  141| ... ~=(_:_:) ...
-#-----|  -> =~ ...
-
 #  141| 1
-#-----|  -> $match
+#-----|  -> =~ ...
 
 #  141| =~ ...
-#-----|  -> ~=(_:_:)
+#-----|  -> 1
 
 #  141| =~ ...
 #-----| match -> true
 #-----| no-match -> case ...
 
-#  141| ~=(_:_:)
-#-----|  -> 1
-
 #  142| return ...
 #-----| return -> exit patterns(x:) (normal)
 
@@ -1505,24 +1487,15 @@ cfg.swift:
 #  144| case ...
 #-----|  -> =~ ... where ...
 
-#  144| $match
-#-----|  -> ... ~=(_:_:) ...
-
-#  144| ... ~=(_:_:) ...
-#-----|  -> =~ ...
-
 #  144| =~ ...
 #-----| match, no-match -> ... .&&(_:_:) ...
 #-----| no-match -> case ...
 
 #  144| x
-#-----|  -> $match
-
-#  144| ~=(_:_:)
-#-----|  -> x
+#-----|  -> =~ ...
 
 #  144| =~ ... where ...
-#-----|  -> ~=(_:_:)
+#-----|  -> x
 
 #  144| ... .&&(_:_:) ...
 
@@ -5992,3 +5965,170 @@ cfg.swift:
 
 #  519| x
 #-----|  -> (Int) ...
+
+#  522| enter testAsyncFor()
+#-----|  -> testAsyncFor()
+
+#  522| exit testAsyncFor()
+
+#  522| exit testAsyncFor() (normal)
+#-----|  -> exit testAsyncFor()
+
+#  522| testAsyncFor()
+#-----|  -> stream
+
+#  523| var ... = ...
+#-----|  -> stream
+
+#  523| stream
+#-----| match -> AsyncStream<Element>.init(_:bufferingPolicy:_:)
+
+#  523| AsyncStream<Element>.init(_:bufferingPolicy:_:)
+#-----|  -> AsyncStream<Int>.Type
+
+#  523| AsyncStream<Int>.Type
+#-----|  -> Int.Type
+
+#  523| call to AsyncStream<Element>.init(_:bufferingPolicy:_:)
+#-----|  -> var ... = ...
+
+#  523| Int.Type
+#-----|  -> .self
+
+#  523| .self
+#-----|  -> .bufferingNewest
+
+#  523| AsyncStream<Int>.Continuation.BufferingPolicy.Type
+#-----|  -> 5
+
+#  523| .bufferingNewest
+#-----|  -> AsyncStream<Int>.Continuation.BufferingPolicy.Type
+
+#  523| call to ...
+#-----|  -> { ... }
+
+#  523| 5
+#-----|  -> call to ...
+
+#  523| enter { ... }
+#-----|  -> { ... }
+
+#  523| exit { ... }
+
+#  523| exit { ... } (normal)
+#-----|  -> exit { ... }
+
+#  523| { ... }
+#-----|  -> continuation
+
+#  523| { ... }
+#-----|  -> call to AsyncStream<Element>.init(_:bufferingPolicy:_:)
+
+#  524| continuation
+#-----|  -> .detached(priority:operation:)
+
+#  525| Task<(), Never>.Type
+#-----|  -> default priority
+
+#  525| return
+#-----| return -> exit { ... } (normal)
+
+#  525| .detached(priority:operation:)
+#-----|  -> Task<(), Never>.Type
+
+#  525| call to detached(priority:operation:)
+#-----|  -> return
+
+#  525| default priority
+#-----|  -> { ... }
+
+#  525| enter { ... }
+#-----|  -> { ... }
+
+#  525| exit { ... }
+
+#  525| exit { ... } (normal)
+#-----|  -> exit { ... }
+
+#  525| { ... }
+#-----|  -> ....(_:_:)
+
+#  525| { ... }
+#-----|  -> call to detached(priority:operation:)
+
+#  526| for ... in ... { ... }
+#-----| non-empty -> i
+#-----| empty -> .finish()
+
+#  526| i
+#-----| match -> .yield(_:)
+
+#  526| 1
+#-----|  -> 100
+
+#  526| ... ....(_:_:) ...
+#-----|  -> for ... in ... { ... }
+
+#  526| ....(_:_:)
+#-----|  -> Int.Type
+
+#  526| Int.Type
+#-----|  -> 1
+
+#  526| 100
+#-----|  -> ... ....(_:_:) ...
+
+#  527| continuation
+#-----|  -> i
+
+#  527| .yield(_:)
+#-----|  -> continuation
+
+#  527| call to yield(_:)
+#-----|  -> for ... in ... { ... }
+
+#  527| i
+#-----|  -> call to yield(_:)
+
+#  529| continuation
+#-----|  -> call to finish()
+
+#  529| .finish()
+#-----|  -> continuation
+
+#  529| call to finish()
+#-----|  -> exit { ... } (normal)
+
+#  533| for ... in ... { ... }
+#-----| empty -> exit testAsyncFor() (normal)
+#-----| non-empty -> i
+
+#  533| i
+#-----| match -> print(_:separator:terminator:)
+
+#  533| stream
+#-----|  -> for ... in ... { ... }
+
+#  534| print(_:separator:terminator:)
+#-----|  -> i
+
+#  534| call to print(_:separator:terminator:)
+#-----|  -> for ... in ... { ... }
+
+#  534| default separator
+#-----|  -> default terminator
+
+#  534| default terminator
+#-----|  -> call to print(_:separator:terminator:)
+
+#  534| (Any) ...
+#-----|  -> [...]
+
+#  534| [...]
+#-----|  -> [...]
+
+#  534| [...]
+#-----|  -> default separator
+
+#  534| i
+#-----|  -> (Any) ...

swift/ql/test/library-tests/controlflow/graph/cfg.swift

@@ -518,3 +518,19 @@ func testAvailable() -> Int {
 
   return x
 }
+
+func testAsyncFor () async {
+    var stream = AsyncStream(Int.self, bufferingPolicy: .bufferingNewest(5), {
+        continuation in
+            Task.detached {
+                for i in 1...100 {
+                    continuation.yield(i)
+                }
+                continuation.finish()
+            }
+    })
+
+    for try await i in stream {
+        print(i)
+    }
+}