Question

在Swift中怎么测试UIAlertController方法

Answer 1

import UIKit

class ViewController: UIViewController {
var actionString: String?
@IBAction func showAlert(sender: UIButton) {
let alertViewController = UIAlertController(title: "Test Title", message: "Message", preferredStyle: .Alert)
let okAction = UIAlertAction(title: "OK", style: .Default) { (action) -> Void in
self.actionString = "OK"
}
let cancelAction = UIAlertAction(title: "Cancel", style: .Cancel) { (action) -> Void in
self.actionString = "Cancel"
}
alertViewController.addAction(cancelAction)
alertViewController.addAction(okAction)
presentViewController(alertViewController, animated: true, completion: nil)
}
}

注意，在这个例子中弹窗动作没有做什么具体的操作，他们只表示能验证单元测试。

让我们开始一个简单的测试：测试这个弹窗控制器的标题和消息内容。

测试的代码如下：

import XCTest
@testable import TestingAlertExperiment
class TestingAlertExperimentTests: XCTestCase {
var sut: ViewController!
override func setUp() {
super.setUp()
sut = UIStoryboard(name: "Main", bundle: nil).instantiateInitialViewController() as! ViewController
UIApplication.sharedApplication().keyWindow?.rootViewController = sut
}
override func tearDown() {
// Put teardown code here. This method is called after the invocation of each test method in the class.
super.tearDown()
}
}
```

我们需要设置 sut 为根视图控制器，否则视图控制器不能弹出这个弹窗视图控制器。

添加 UIAlertController 测试标题的代码如下：

?

```Swift
func testAlert_HasTitle() {
sut.showAlert(UIButton())
XCTAssertTrue(sut.presentedViewController is UIAlertController)
XCTAssertEqual(sut.presentedViewController?.title, "Test Title")
}
```

这很简单。现在让我们测试 UIAlertController 的取消按钮。这里有一个问题：无法获取弹窗动作的闭包。因此我们需要模拟弹窗动作，为了存储这个
handler 并在测试中调用它，看弹窗动作是否和我们预期的一样。在测试用例中添加这样一个类：

?

```Swift
class MockAlertAction : UIAlertAction {
typealias Handler = ((UIAlertAction) -> Void)
var handler: Handler?
var mockTitle: String?
var mockStyle: UIAlertActionStyle
convenience init(title: String?, style: UIAlertActionStyle, handler: ((UIAlertAction) -> Void)?) {
self.init()
mockTitle = title
mockStyle = style
self.handler = handler
}
override init() {
mockStyle = .Default
super.init()
}
}

这个模拟类的主要工作是捕获 handler 块，以备后用。现在我们需要将这个模拟的类插入到实现代码中。将视图控制器中的代码换成下面这个：

import UIKit
class ViewController: UIViewController {
var Action = UIAlertAction.self
var actionString: String?
@IBAction func showAlert(sender: UIButton) {
let alertViewController = UIAlertController(title: "Test Title", message: "Message", preferredStyle: .Alert)
let okAction = Action.init(title: "OK", style: .Default) { (action) -> Void in
self.actionString = "OK"
}
let cancelAction = Action.init(title: "Cancel", style: .Cancel) { (action) -> Void in
self.actionString = "Cancel"
}
alertViewController.addAction(cancelAction)
alertViewController.addAction(okAction)
presentViewController(alertViewController, animated: true, completion: nil)
}
}

```

我们添加了一个类变量`Action`，并设置为`UIAlertAction.self`。这个变量我们会在初始化弹窗动作时使用。这就能让我们在测试时可以重写它。像这样：

```Swift
func testAlert_FirstActionStoresCancel() {
sut.Action = MockAlertAction.self
sut.showAlert(UIButton())
let alertController = sut.presentedViewController as! UIAlertController
let action = alertController.actions.first as! MockAlertAction
action.handler!(action)
XCTAssertEqual(sut.actionString, "Cancel")
}

首先我们插入了这个弹窗动作。之后我们调用代码弹出弹窗视图控制器。我们从呈现的视图控制器中获取了取消动作，并且成功调用了捕获的 handler
块。最后一步就是去断言当前的动作是否和我们预期的一样。