Question

这是我的下载功能：

// Download a file from the url to the local directory
class func downloadUrl(url: URL, to dirUrl: URL, completion: (() -> ())?){
    let sessionConfig = URLSessionConfiguration.default
    let session = URLSession(configuration: sessionConfig)
    let request = URLRequest(url: url)

    let task = session.downloadTask(with: request) {(tempLocalUrl, response, error) in
        if let tempLocalUrl = tempLocalUrl, error == nil {
            // Success, copy the downloaded file from the memory to the disk
            print("Finished downloading!")
            do {
                try FileManager.default.copyItem(at: tempLocalUrl,to:
                    dirUrl.appendingPathComponent((response?.suggestedFilename)!))
                if completion != nil {
                    completion!()
                }
            } catch (let writeError) {
                print("Fliled to write file \(dirUrl) : \(writeError)")
            }
        } else {
            print("Failure: \(String(describing: error?.localizedDescription))")
        }
    }
    task.resume()
}

我想编写一个单元测试方法来测试它是否将文件从url下载到dirUrl。

func testDownloadUrl(){
    let fm = FileManager.default
    let url = URL(string: "https://raw.githubusercontent.com/apple/swift/master/README.md")
    fileDownloader.downloadUrl(url: url!, to: fm.temporaryDirectory, completion: nil)

    // Check the contents of temp file
    let tempContents = try? fm.contentsOfDirectory(atPath: fm.temporaryDirectory.path)
        print("Contents: \(tempContents)")
    }

然而，没有输出＆＃34;完成下载！＆＃34;或者＆＃34;失败......＆＃34;即使我通过了单元测试，所以我猜测在这个测试用例中没有调用completionHandler。

我的问题是如何使单元测试方法等到下载任务完成？

Answer 1

标准习语是使用Xcode asynchronous testing APIs，主要是XCTestExpectation类。这些都是从Xcode 6开始提供的，所以，如果您使用的是最新版本，那么您应该受到保护;）

常规异步测试模板如下：

func testAsyncFunction() {
    let asyncDone = expectation(description: "Async function")
    ...
    someAsyncFunction(...) {
        ...
        asyncDone.fulfill()
    }
    wait(for: [asyncDone], timeout: 10)
    /* Test the results here */
}

这将阻止您的测试执行，直到上述功能完成（或指定的超时已经过去）。

如何用`completionHandler`编写`downloadTask`的单元测试？

1 个答案: