From 28e0eaaeeba2821c813085a501e9c38766bb0616 Mon Sep 17 00:00:00 2001
From: Victor Shamanov <wiruzx@gmail.com>
Date: Sun, 19 Nov 2023 21:22:55 +0000
Subject: [PATCH] Fix cancellation support

---
 Chatto/Tests/ObservableAsyncStreamTests.swift | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/Chatto/Tests/ObservableAsyncStreamTests.swift b/Chatto/Tests/ObservableAsyncStreamTests.swift
index 0d7d8f52..0bf22d83 100644
--- 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