Make ExecutorTests less flaky.

BoltsFramework · Mar 17, 2016 · 17c72c7 · 17c72c7
1 parent 9547946
commit 17c72c7
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/Tests/ExecutorTests.swift b/Tests/ExecutorTests.swift
@@ -54,14 +54,18 @@ class ExecutorTests: XCTestCase {
     func testQueueExecute() {
         let expectation = expectationWithDescription(currentTestName)
         var finished = false
+        let semaphore = dispatch_semaphore_create(0)
 
         Executor.Queue(dispatch_get_global_queue(DISPATCH_QUEUE_PRIORITY_DEFAULT, 0)).execute {
-            expectation.fulfill()
+            dispatch_semaphore_wait(semaphore, DISPATCH_TIME_FOREVER)
             finished = true
+            expectation.fulfill()
         }
 
         XCTAssertFalse(finished)
+        dispatch_semaphore_signal(semaphore)
         waitForExpectationsWithTimeout(0.5, handler: nil)
+        XCTAssertTrue(finished)
     }
 
     func testClosureExecute() {
@@ -79,15 +83,19 @@ class ExecutorTests: XCTestCase {
     func testOperationQueueExecute() {
         let expectation = expectationWithDescription(currentTestName)
         let operationQueue = NSOperationQueue()
+        let semaphore = dispatch_semaphore_create(0)
         var finished = false
 
         Executor.OperationQueue(operationQueue).execute {
-            expectation.fulfill()
+            dispatch_semaphore_wait(semaphore, DISPATCH_TIME_FOREVER)
             finished = true
+            expectation.fulfill()
         }
 
         XCTAssertFalse(finished)
+        dispatch_semaphore_signal(semaphore)
         waitForExpectationsWithTimeout(0.5, handler: nil)
+        XCTAssertTrue(finished)
     }
 
     // MARK: Descriptions