async_taskgroup_is_asyncsequence.swift

// RUN: %target-run-simple-swift( -target %target-swift-5.1-abi-triple -parse-as-library) | %FileCheck %s --dump-input=always

// REQUIRES: executable_test
// REQUIRES: concurrency

// rdar://76038845
// REQUIRES: concurrency_runtime
// UNSUPPORTED: back_deployment_runtime

@available(SwiftStdlib 5.1,*)
func test_taskGroup_is_asyncSequence()async{
print(#function)

letsum=awaitwithTaskGroup(of:Int.self, returning:Int.self){ group in
fornin1...10{
 group.spawn{
print("add \(n)")
return n
}
}

varsum=0
forawaitrin group { // here
print("next: \(r)")
 sum += r
}

return sum
}

print("result: \(sum)")
}

@available(SwiftStdlib 5.1,*)
func test_throwingTaskGroup_is_asyncSequence()asyncthrows{
print(#function)

letsum=tryawaitwithThrowingTaskGroup(of:Int.self, returning:Int.self){ group in
fornin1...10{
 group.spawn{
print("add \(n)")
return n
}
}

varsum=0
fortryawaitrin group { // here
print("next: \(r)")
 sum += r
}

return sum
}

print("result: \(sum)")
}

typealiasLabelledTuple=(x:Int, y:Int)

@available(SwiftStdlib 5.1,*)
func test_asyncSequence_labelledTuples()async{
print(#function)

letsum=awaitwithTaskGroup(of:LabelledTuple.self, returning:Int.self){ group in
fornin1...10{
 group.spawn{
print("add (x: \(n), y: 1)")
return(x: n, y:1)
}
}

varsum=0
forawait(x, y)in group { // here
print("next: (x:\(x), y:\(y))")
 sum += x + y
}

return sum
}

print("result: \(sum)")
}

@available(SwiftStdlib 5.1,*)
@mainstructMain{
staticfunc main()async{
awaittest_taskGroup_is_asyncSequence()
 // CHECK: test_taskGroup_is_asyncSequence()
 // CHECK: result: 55

try!awaittest_throwingTaskGroup_is_asyncSequence()
 // CHECK: test_throwingTaskGroup_is_asyncSequence()
 // CHECK: result: 55

awaittest_asyncSequence_labelledTuples()
 // CHECK: test_asyncSequence_labelledTuples()
 // CHECK: result: 65
}
}