Fix flaky test by inserting a mega yield

supabase · Oct 26, 2023 · d102867 · d102867
1 parent 6e79b0c
commit d102867
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/Sources/_Helpers/Task.swift b/Sources/_Helpers/Task.swift
@@ -0,0 +1,17 @@
+//
+//  File.swift
+//  
+//
+//  Created by Guilherme Souza on 26/10/23.
+//
+
+import Foundation
+
+extension Task where Success == Never, Failure == Never {
+  @_spi(Internal)
+  public static func megaYield(count: Int = 20) async {
+    for _ in 0..<count {
+      await Task<Void, Never>.detached(priority: .background) { await Task.yield() }.value
+    }
+  }
+}
diff --git a/Tests/GoTrueTests/GoTrueClientTests.swift b/Tests/GoTrueTests/GoTrueClientTests.swift
@@ -45,6 +45,8 @@ final class GoTrueClientTests: XCTestCase {
 
     streamTask.cancel()
 
+    await Task.megaYield()
+
     listeners = await sut.authChangeListeners
     XCTAssertEqual(listeners.count, 0)
   }