Cancellation

Author: swhitty

Sources/Timeout.swift

@@ -56,7 +56,11 @@ public func withThrowingTimeout<T>(
             throw TimeoutError(timeout: seconds)
         }
 
-        let bodyResult = await bodyTask.result
+        let bodyResult = await withTaskCancellationHandler {
+            await bodyTask.result
+        } onCancel: {
+            bodyTask.cancel()
+        }
         timeoutTask.cancel()
         let timeoutResult = await timeoutTask.result
 

Tests/TimeoutTests.swift

@@ -90,6 +90,21 @@ struct TimeoutTests {
             }
         }
     }
+
+    @Test
+    func timeout_cancels() async {
+        let task = Task {
+            try await withThrowingTimeout(seconds: 1) {
+                try await Task.sleep(nanoseconds: 1_000_000_000)
+            }
+        }
+
+        task.cancel()
+
+        await #expect(throws: CancellationError.self) {
+            try await task.value
+        }
+    }
 }
 
 public struct NonSendable<T> {

Tests/TimeoutXCTests.swift

@@ -91,6 +91,23 @@ final class TimeoutTests: XCTestCase {
             XCTAssertTrue(error is TimeoutError)
         }
     }
+
+    func testTimeout_Cancels() async {
+        let task = Task {
+            try await withThrowingTimeout(seconds: 1) {
+                try await Task.sleep(nanoseconds: 1_000_000_000)
+            }
+        }
+
+        task.cancel()
+
+        do {
+            _ = try await task.value
+            XCTFail("Expected Error")
+        } catch {
+            XCTAssertTrue(error is CancellationError)
+        }
+    }
 }
 
 public struct NonSendable<T> {