Fix flaky testCancelAsyncImageTask

kean · Apr 21, 2024 · e4d214f · e4d214f
1 parent 3776cae
commit e4d214f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/Tests/NukeTests/ImagePipelineTests/ImagePipelineAsyncAwaitTests.swift b/Tests/NukeTests/ImagePipelineTests/ImagePipelineAsyncAwaitTests.swift
@@ -124,11 +124,12 @@ class ImagePipelineAsyncAwaitTests: XCTestCase, @unchecked Sendable {
     func testCancelAsyncImageTask() async throws {
         dataLoader.queue.isSuspended = true
 
+        pipeline.queue.suspend()
         let task = pipeline.imageTask(with: Test.url)
-
         observer = NotificationCenter.default.addObserver(forName: MockDataLoader.DidStartTask, object: dataLoader, queue: OperationQueue()) { _ in
             task.cancel()
         }
+        pipeline.queue.resume()
 
         var caughtError: Error?
         do {