use withExtendedLifetime to preserve the producer side

apple · Apr 11, 2023 · c7863f6 · c7863f6
1 parent b1af3eb
commit c7863f6
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/Tests/NIOCoreTests/AsyncSequences/NIOThrowingAsyncSequenceTests.swift b/Tests/NIOCoreTests/AsyncSequences/NIOThrowingAsyncSequenceTests.swift
@@ -487,7 +487,10 @@ final class NIOThrowingAsyncSequenceProducerTests: XCTestCase {
         task.cancel()
         let result = await task.result
         XCTAssertEqualWithoutAutoclosure(await delegate.events.prefix(1).collect(), [.didTerminate])
-        XCTAssertNil(try result.get())
+
+        try withExtendedLifetime(new.source) {
+            XCTAssertNil(try result.get())
+        }
     }
 
     func testTaskCancel_whenStreaming_andNotSuspended() async throws {
@@ -569,9 +572,11 @@ final class NIOThrowingAsyncSequenceProducerTests: XCTestCase {
         }
 
         task.cancel()
-
+        
         let result = await task.result
-        XCTAssertNil(try result.get())
+        try withExtendedLifetime(new.source) {
+            XCTAssertNil(try result.get())
+        }
     }
 
     // MARK: - Next