Improved flush policy interval test.

segmentio · Nov 16, 2023 · 3a42e48 · 3a42e48
1 parent 3ab0792
commit 3a42e48
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/Tests/Segment-Tests/FlushPolicy_Tests.swift b/Tests/Segment-Tests/FlushPolicy_Tests.swift
@@ -137,9 +137,15 @@ class FlushPolicyTests: XCTestCase {
 
         XCTAssertTrue(analytics.hasUnsentEvents)
 
-        // sleep for 4 seconds for 2 second flush policy
-        RunLoop.main.run(until: Date.init(timeIntervalSinceNow: 4))
-
-        XCTAssertFalse(analytics.hasUnsentEvents)
+        @Atomic var flushSent = false
+        while !flushSent {
+            RunLoop.main.run(until: Date.distantPast)
+            if analytics.pendingUploads!.count > 0 {
+                // flush was triggered
+                flushSent = true
+            }
+        }
+
+        XCTAssertTrue(flushSent)
     }
 }