From 2f1b859a033dd66cc01309c807692423988b8a32 Mon Sep 17 00:00:00 2001
From: Sami Samhuri <sami@samhuri.net>
Date: Fri, 4 Apr 2025 10:12:48 -0700
Subject: [PATCH] Stop relying on a specific delay in cancellation test

---
 .../SJSAssetExportSessionTests.swift                      | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/Tests/SJSAssetExportSessionTests/SJSAssetExportSessionTests.swift b/Tests/SJSAssetExportSessionTests/SJSAssetExportSessionTests.swift
index 9084e02..c2c5511 100644
--- a/Tests/SJSAssetExportSessionTests/SJSAssetExportSessionTests.swift
+++ b/Tests/SJSAssetExportSessionTests/SJSAssetExportSessionTests.swift
@@ -296,11 +296,11 @@ final class ExportSessionTests: BaseTests {
     @Test func test_export_cancellation() async throws {
         let sourceURL = resourceURL(named: "test-720p-h264-24fps.mov")
         let destinationURL💥 = makeTemporaryURL()
+        let subject = ExportSession()
         let task = Task {
             let sourceAsset = AVURLAsset(url: sourceURL, options: [
                 AVURLAssetPreferPreciseDurationAndTimingKey: true,
             ])
-            let subject = ExportSession()
             try await subject.export(
                 asset: sourceAsset,
                 video: .codec(.h264, width: 1280, height: 720),
@@ -309,8 +309,10 @@ final class ExportSessionTests: BaseTests {
             )
             Issue.record("Task should be cancelled long before we get here")
         }
-        NSLog("Sleeping for 0.3s")
-        try await Task.sleep(for: .milliseconds(300))
+        NSLog("Waiting for encoding to begin...")
+        for await progress in subject.progressStream where progress > 0 {
+            break
+        }
         NSLog("Cancelling task")
         task.cancel()
         try? await task.value // Wait for task to complete