More improvements to the delay test

lightningkite · May 9, 2016 · 94a19de · 94a19de
1 parent 4762a47
commit 94a19de
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/Example/Tests/LKAlertControllerTests.swift b/Example/Tests/LKAlertControllerTests.swift
@@ -145,18 +145,20 @@ class LKAlertControllerTests: XCTestCase {
 	func testDelay() {
 		let expectation = expectationWithDescription("Delay testing")
 
-		let startTime = NSDate()
+		var startTime = NSDate()
 
 		LKAlertController.overrideShowForTesting { (style, title, message, actions, fields) -> Void in
 
 			let endTime = NSDate()
-			let delayTime = Int(endTime.timeIntervalSinceDate(startTime))
+			let delayTime = Int(round(endTime.timeIntervalSinceDate(startTime)))
 
 			XCTAssertEqual(delayTime, 5)
 
 			expectation.fulfill()
 		}
 
+		startTime = NSDate()
+
 		Alert(title: "Title", message: "Message").addAction("Cancel").delay(5).show()
 
 		waitForExpectationsWithTimeout(6, handler: nil)