Adding test for the delay

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