Add WaitUnitll

deda9 · Jul 17, 2019 · b1737a7 · b1737a7
1 parent 28ef53c
commit b1737a7
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/WaitUntill.swift b/WaitUntill.swift
@@ -0,0 +1,11 @@
+import XCTest
+
+extension XCTestCase {
+    func waitUntil(timeout: TimeInterval = 5, done: @escaping (@escaping () -> Void) -> Void) {
+        let expectation = self.expectation(description: "fromAtoB expectation")
+        done {
+            expectation.fulfill()
+        }
+        waitForExpectations(timeout: timeout, handler: nil)
+    }
+}