单元测试WKNavigationDelegate函数

时间:2018-03-14 12:34:07

标签: swift io webkit

我有一个实现WKNavigationDelegate函数的UIViewController,我想在这些函数中对逻辑进行单元测试。这是一个例子:

func webView(_ webView: WKWebView,
             decidePolicyFor navigationAction: WKNavigationAction,
             decisionHandler: @escaping (WKNavigationActionPolicy) -> Void) {
    guard let url = navigationAction.request.url else {
        decisionHandler(.cancel)
        return
    }

    if url.absoluteString != "https://my-approved-url" {
        decisionHandler(.cancel)
        return
    }

    decisionHandler(.allow)
}

我希望我的单元测试能够确保使用基于WKNavigationAction的request.url的正确的WKNavigationActionPolicy调用decisionHandler。

但是,我无法弄清楚如何测试这个功能。当我运行我的测试项目时,在webview上调用.load()不会触发委托函数。我也试图直接调用这个函数来测试它,但似乎不可能实例化我自己的新WKNavigationAction(.request是只读的)。

在WKNavigationDelegate函数中单元测试逻辑的正确方法是什么?

1 个答案:

答案 0 :(得分:7)

直接调用委托方法是单元测试环境中最合适的方法。您可以继承WKNavigationAction,并将该类的实例作为输入参数传递给委托方法:

class FakeNavigationAction: WKNavigationAction {
    let testRequest: URLRequest
    override var request: URLRequest {
        return testRequest
    }

    init(testRequest: URLRequest) {
        self.testRequest = testRequest
        super.init()
    }
}

稍后,在单元测试中:

// setup
var receivedPolicy: WKNavigationActionPolicy?
let fakeAction = FakeNavigationAction(testRequest: ...)

// act
delegateObject.webView(webView, decidePolicyFor: fakeAction, decisionHandler: { receivedPolicy = $0 })

// assert
XCTAssertEqual(receivedPolicy, theExpectedValue)

另一种方法是调整request的getter,因为WKNavigationAction是Objective-C类,但这更像是一个hacky解决方案。