- fixed bug

Author: Alex Belozierov

Sources/SwiftCoroutine/CoFuture/Operators/CoFuture2+await.swift

@@ -28,9 +28,16 @@ extension CoFuture {
             mutex?.unlock()
             return try result.get()
         }
-        return try Coroutine.await { (callback: @escaping (Result<Value, Error>) -> Void) in
-                self.append(callback: callback)
-                self.mutex?.unlock()
+        let coroutine: CoroutineProtocol
+        do {
+            coroutine = try Coroutine.current()
+        } catch {
+            mutex?.unlock()
+            throw error
+        }
+        return try coroutine.await { (callback: @escaping (Result<Value, Error>) -> Void) in
+            self.append(callback: callback)
+            self.mutex?.unlock()
         }.get()
     }
 

Sources/SwiftCoroutine/Coroutine/TaskScheduler/CoroutineScheduler.swift

@@ -51,7 +51,7 @@ extension CoroutineScheduler {
     /// - Parameter task: The closure that will be executed inside coroutine.
     /// If the task throws an error, then the coroutine will be terminated.
     @inlinable public func startCoroutine(_ task: @escaping () throws -> Void) {
-        startCoroutine { do { try task() } catch { print(error) } }
+        startCoroutine { try? task() }
     }
 
     /// Start a coroutine and await its result. Must be called inside other coroutine.

Tests/SwiftCoroutineTests/CoFutureTests/CoFutureCombineTests.swift

@@ -20,8 +20,8 @@ class CoFutureCombineTests: XCTestCase {
 
     func testSubscribe() {
         let exp = expectation(description: "testSubscribe")
-        exp.expectedFulfillmentCount = 100
-        for i in 0..<100 {
+        exp.expectedFulfillmentCount = 1000
+        for i in 0..<1000 {
             let future = Future<Int, Never> { promise in
                 DispatchQueue.global().asyncAfter(deadline: .now() + .milliseconds(100)) {
                     promise(.success(i))
@@ -32,7 +32,7 @@ class CoFutureCombineTests: XCTestCase {
                 exp.fulfill()
             }
         }
-        wait(for: [exp], timeout: 3)
+        wait(for: [exp], timeout: 5)
     }
 
     func testSubscription() {