瀏覽代碼

Added extra progress check to increase reliability of download tests.

Christian Noon 9 年之前
父節點
當前提交
2e1cbb8193
共有 1 個文件被更改,包括 5 次插入5 次删除
  1. 5 5
      Tests/DownloadTests.swift

+ 5 - 5
Tests/DownloadTests.swift

@@ -403,8 +403,8 @@ class DownloadResumeDataTestCase: BaseTestCase {
 
         // When
         let download = Alamofire.download(urlString)
-        download.downloadProgress { _, _, _ in
-            download.cancel()
+        download.downloadProgress { _, totalBytesReceived, _ in
+            if totalBytesReceived > 10_000 { download.cancel() }
         }
         download.response { resp in
             response = resp
@@ -420,7 +420,7 @@ class DownloadResumeDataTestCase: BaseTestCase {
         XCTAssertNotNil(response?.resumeData)
         XCTAssertNotNil(response?.error)
 
-        XCTAssertNotNil(download.resumeData, "resume data should not be nil")
+        XCTAssertNotNil(download.resumeData)
 
         if let responseResumeData = response?.resumeData, let resumeData = download.resumeData {
             XCTAssertEqual(responseResumeData, resumeData)
@@ -436,8 +436,8 @@ class DownloadResumeDataTestCase: BaseTestCase {
 
         // When
         let download = Alamofire.download(urlString)
-        download.downloadProgress { _, _, _ in
-            download.cancel()
+        download.downloadProgress { _, totalBytesReceived, _ in
+            if totalBytesReceived > 10_000 { download.cancel() }
         }
         download.responseJSON { resp in
             response = resp