Use async sleep in concurrency tests

Author: drexin

test/Concurrency/Runtime/actor_counters.swift

@@ -53,7 +53,7 @@ func runTest(numCounters: Int, numWorkers: Int, numIterations: Int) async {
   for i in 0..<numWorkers {
     workers.append(
       Task.runDetached { [counters] in
-        usleep(UInt32.random(in: 0..<100) * 1000)
+        await Task.sleep(UInt64.random(in: 0..<100) * 1_000_000)
         await worker(identity: i, counters: counters, numIterations: numIterations)
       }
     )

test/Concurrency/Runtime/async_let_fibonacci.swift

@@ -30,12 +30,12 @@ func asyncFib(_ n: Int) async -> Int {
   async let second = await asyncFib(n-1)
 
   // Sleep a random amount of time waiting on the result producing a result.
-  usleep(UInt32.random(in: 0..<100) * 1000)
+  await Task.sleep(UInt64.random(in: 0..<100) * 1_000_000)
 
   let result = await first + second
 
   // Sleep a random amount of time before producing a result.
-  usleep(UInt32.random(in: 0..<100) * 1000)
+  await Task.sleep(UInt64.random(in: 0..<100) * 1_000_000)
 
   return result
 }

test/Concurrency/Runtime/async_task_priority_current.swift

@@ -32,7 +32,7 @@ func test_multiple_lo_indirectly_escalated() async {
   @concurrent
   func loopUntil(priority: Task.Priority) async {
     while (Task.currentPriority != priority) {
-      sleep(1)
+      await Task.sleep(1_000_000_000)
     }
   }
 

test/Concurrency/Runtime/async_taskgroup_is_empty.swift

@@ -25,7 +25,7 @@ func test_taskGroup_isEmpty() async {
       print("before add: isEmpty=\(group.isEmpty)")
 
       await group.add {
-        sleep(2)
+        await Task.sleep(2_000_000_000)
         return await asyncEcho(1)
       }
 

test/Concurrency/Runtime/async_taskgroup_next_not_invoked_cancelAll.swift

@@ -19,7 +19,7 @@ func test_skipCallingNext_butInvokeCancelAll() async {
     for n in numbers {
       print("group.add { \(n) }")
       await group.add { [group] () async -> Int in
-        sleep(2)
+        await Task.sleep(1_000_000_000)
         print("  inside group.add { \(n) }")
         print("  inside group.add { \(n) } (group cancelled: \(group.isCancelled))")
         print("  inside group.add { \(n) } (group child task cancelled: \(Task.isCancelled))")

test/Concurrency/Runtime/async_taskgroup_next_not_invoked_without_cancelAll.swift

@@ -11,15 +11,14 @@ import Darwin
 #elseif canImport(Glibc)
 import Glibc
 #endif
-
 func test_skipCallingNext() async {
   let numbers = [1, 1]
 
   let result = try! await Task.withGroup(resultType: Int.self) { (group) async -> Int in
     for n in numbers {
       print("group.add { \(n) }")
       await group.add { () async -> Int in
-        sleep(1)
+        await Task.sleep(1_000_000_000)
         let c = Task.isCancelled
         print("  inside group.add { \(n) } (canceled: \(c))")
         return n

test/Concurrency/Runtime/async_taskgroup_next_on_completed.swift

@@ -28,7 +28,7 @@ func test_sum_nextOnCompleted() async {
 
     // We specifically want to await on completed child tasks in this test,
     // so give them some time to complete before we hit group.next()
-    sleep(2)
+      await Task.sleep(2_000_000_000)
 
     var sum = 0
     do {

test/Concurrency/Runtime/async_taskgroup_next_on_pending.swift

@@ -13,7 +13,7 @@ import Glibc
 #endif
 
 func completeSlowly(n: Int) async -> Int {
-  sleep(UInt32(n + 1))
+  await Task.sleep(UInt64((n * 1_000_000_000) + 1_000_000_000))
   print("  complete group.add { \(n) }")
   return n
 }

test/Concurrency/Runtime/basic_future.swift

@@ -34,7 +34,7 @@ func testSimple(
     // can complete.
     if doSuspend {
       print("- Future sleeping")
-      sleep(1) // TODO: await Task.sleep would be preferable here
+      await Task.sleep(1_000_000_000)
     }
 
     if (shouldThrow) {
@@ -50,7 +50,7 @@ func testSimple(
   // can complete.
   if !doSuspend {
     print("+ Reader sleeping")
-    sleep(1)
+    await Task.sleep(1_000_000_000)
   }
 
   do {

test/Concurrency/Runtime/future_fibonacci.swift

@@ -37,12 +37,12 @@ func asyncFib(_ n: Int) async -> Int {
   }
 
   // Sleep a random amount of time waiting on the result producing a result.
-  usleep(UInt32.random(in: 0..<100) * 1000)
+  await Task.sleep(UInt64.random(in: 0..<100) * 1_000_000)
 
   let result = try! await first.get() + second.get()
 
   // Sleep a random amount of time before producing a result.
-  usleep(UInt32.random(in: 0..<100) * 1000)
+  await Task.sleep(UInt64.random(in: 0..<100) * 1_000_000)
 
   return result
 }