Fix cancellation support

badoo · Dec 4, 2023 · 28e0eaa · 28e0eaa
1 parent e7b929a
commit 28e0eaa
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/Chatto/Tests/ObservableAsyncStreamTests.swift b/Chatto/Tests/ObservableAsyncStreamTests.swift
@@ -52,7 +52,13 @@ final class ObservableAsyncStreamTests: XCTestCase {
         }
         observable = nil
 
-        try await withTimeout { await iterateAllStreamValuesTask.value }
+        try await withTimeout {
+            await withTaskCancellationHandler {
+                await iterateAllStreamValuesTask.value
+            } onCancel: {
+                iterateAllStreamValuesTask.cancel()
+            }
+        }
     }
 
     // MARK: - Timeout utils