譯自：swiftandpainless.com/testing-pus…

---

上個月我寫的關于使用 Swift 進行測試驅動開發的書終于出版了，我會在本文和接下來的一些博文中介紹這本書撰寫過程中的一些心得和體會。

在本文中，我將會展示一種很好的用來測試一個視圖控制器是否因為某個事件而被壓入導航棧的方法。

假設我們在視圖控制器上有一個按鈕。當用戶點擊按鈕時，一個新的視圖控制器應該被壓入導航棧中。我們怎么測試這個呢？

簡單！我們可以通過對導航欄控制器進行 Mock 的方式來達到我們的目的，就像接下來這樣：

class MockNavigationController: UINavigationController {var pushedViewController: UIViewController?override func pushViewController(viewController: UIViewController, animated: Bool) {pushedViewController = viewControllersuper.pushViewController(viewController, animated: true)}
}
復制代碼

確切地說，這是部分 Mock。這里我們建立了一個 UINavigationController 的子類，并且它只重寫了父類的一個方法。這個 Mock 操作會在 pushViewController(_:animated:) 被調用時被觸發，并且會將傳入的視圖控制器也就是第一個參數保存下來。

測試代碼如下所示：

func testTappingPushButton_PushesDetailViewControllerOntoStack() {let viewController = ViewController()let navigationController = MockNavigationController(rootViewController: viewController)UIApplication.sharedApplication().keyWindow?.rootViewController = navigationControllerguard let view = viewController.view as? View else { XCTFail(); return }view.button.sendActionsForControlEvents(.TouchUpInside)XCTAssertTrue(navigationController.pushedViewController is DetailViewController)
}
復制代碼

首先，我們將一個視圖控制器設為我們 Mock 導航欄控制器實例的 rootViewController。然后我們將導航欄控制器設為 UIApplication 單例的 keyWindow 的 rootViewController。為了能夠將一個視圖控制器壓入導航棧中，視圖控制器的視圖必須位于視圖層次里，這一步是必須的。

下一步，我們獲取按鈕并且向它發送 .TouchUpInside 事件。最后我們斷言彈出的視圖控制器是 DetailViewController 類型的。

你可以在 GitHub 上找到相應的示例代碼。

你可以在我的書中找到更多貼近實際的測試例子。當你讀這本書時，請別忘了告訴我你對這本書的看法。

---

如有任何知識產權、版權問題或理論錯誤，還請指正。
https://juejin.im/post/5a3217cd6fb9a045204c3ce5
轉載請注明原作者及以上信息。