这是我的下载功能:
// 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)")
}
但是,没有输出“下载完成!” 或“失败......”即使我通过了单元测试,所以我猜在这个测试用例中没有调用completionHandler。
我的问题是如何让单元测试方法等到下载任务完成?