UploadTests.swift 30 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766
  1. // UploadTests.swift
  2. //
  3. // Copyright (c) 2014–2016 Alamofire Software Foundation (http://alamofire.org/)
  4. //
  5. // Permission is hereby granted, free of charge, to any person obtaining a copy
  6. // of this software and associated documentation files (the "Software"), to deal
  7. // in the Software without restriction, including without limitation the rights
  8. // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  9. // copies of the Software, and to permit persons to whom the Software is
  10. // furnished to do so, subject to the following conditions:
  11. //
  12. // The above copyright notice and this permission notice shall be included in
  13. // all copies or substantial portions of the Software.
  14. //
  15. // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  16. // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  17. // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  18. // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  19. // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  20. // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  21. // THE SOFTWARE.
  22. import Alamofire
  23. import Foundation
  24. import XCTest
  25. class UploadFileInitializationTestCase: BaseTestCase {
  26. func testUploadClassMethodWithMethodURLAndFile() {
  27. // Given
  28. let URLString = "https://httpbin.org/"
  29. let imageURL = URLForResource("rainbow", withExtension: "jpg")
  30. // When
  31. let request = Alamofire.upload(.POST, URLString, file: imageURL)
  32. // Then
  33. XCTAssertNotNil(request.request, "request should not be nil")
  34. XCTAssertEqual(request.request?.HTTPMethod ?? "", "POST", "request HTTP method should be POST")
  35. XCTAssertEqual(request.request?.URLString ?? "", URLString, "request URL string should be equal")
  36. XCTAssertNil(request.response, "response should be nil")
  37. }
  38. func testUploadClassMethodWithMethodURLHeadersAndFile() {
  39. // Given
  40. let URLString = "https://httpbin.org/"
  41. let imageURL = URLForResource("rainbow", withExtension: "jpg")
  42. // When
  43. let request = Alamofire.upload(.POST, URLString, headers: ["Authorization": "123456"], file: imageURL)
  44. // Then
  45. XCTAssertNotNil(request.request, "request should not be nil")
  46. XCTAssertEqual(request.request?.HTTPMethod ?? "", "POST", "request HTTP method should be POST")
  47. XCTAssertEqual(request.request?.URLString ?? "", URLString, "request URL string should be equal")
  48. let authorizationHeader = request.request?.valueForHTTPHeaderField("Authorization") ?? ""
  49. XCTAssertEqual(authorizationHeader, "123456", "Authorization header is incorrect")
  50. XCTAssertNil(request.response, "response should be nil")
  51. }
  52. }
  53. // MARK: -
  54. class UploadDataInitializationTestCase: BaseTestCase {
  55. func testUploadClassMethodWithMethodURLAndData() {
  56. // Given
  57. let URLString = "https://httpbin.org/"
  58. // When
  59. let request = Alamofire.upload(.POST, URLString, data: NSData())
  60. // Then
  61. XCTAssertNotNil(request.request, "request should not be nil")
  62. XCTAssertEqual(request.request?.HTTPMethod ?? "", "POST", "request HTTP method should be POST")
  63. XCTAssertEqual(request.request?.URLString ?? "", URLString, "request URL string should be equal")
  64. XCTAssertNil(request.response, "response should be nil")
  65. }
  66. func testUploadClassMethodWithMethodURLHeadersAndData() {
  67. // Given
  68. let URLString = "https://httpbin.org/"
  69. // When
  70. let request = Alamofire.upload(.POST, URLString, headers: ["Authorization": "123456"], data: NSData())
  71. // Then
  72. XCTAssertNotNil(request.request, "request should not be nil")
  73. XCTAssertEqual(request.request?.HTTPMethod ?? "", "POST", "request HTTP method should be POST")
  74. XCTAssertEqual(request.request?.URLString ?? "", URLString, "request URL string should be equal")
  75. let authorizationHeader = request.request?.valueForHTTPHeaderField("Authorization") ?? ""
  76. XCTAssertEqual(authorizationHeader, "123456", "Authorization header is incorrect")
  77. XCTAssertNil(request.response, "response should be nil")
  78. }
  79. }
  80. // MARK: -
  81. class UploadStreamInitializationTestCase: BaseTestCase {
  82. func testUploadClassMethodWithMethodURLAndStream() {
  83. // Given
  84. let URLString = "https://httpbin.org/"
  85. let imageURL = URLForResource("rainbow", withExtension: "jpg")
  86. let imageStream = NSInputStream(URL: imageURL)!
  87. // When
  88. let request = Alamofire.upload(.POST, URLString, stream: imageStream)
  89. // Then
  90. XCTAssertNotNil(request.request, "request should not be nil")
  91. XCTAssertEqual(request.request?.HTTPMethod ?? "", "POST", "request HTTP method should be POST")
  92. XCTAssertEqual(request.request?.URLString ?? "", URLString, "request URL string should be equal")
  93. XCTAssertNil(request.response, "response should be nil")
  94. }
  95. func testUploadClassMethodWithMethodURLHeadersAndStream() {
  96. // Given
  97. let URLString = "https://httpbin.org/"
  98. let imageURL = URLForResource("rainbow", withExtension: "jpg")
  99. let imageStream = NSInputStream(URL: imageURL)!
  100. // When
  101. let request = Alamofire.upload(.POST, URLString, headers: ["Authorization": "123456"], stream: imageStream)
  102. // Then
  103. XCTAssertNotNil(request.request, "request should not be nil")
  104. XCTAssertEqual(request.request?.HTTPMethod ?? "", "POST", "request HTTP method should be POST")
  105. XCTAssertEqual(request.request?.URLString ?? "", URLString, "request URL string should be equal")
  106. let authorizationHeader = request.request?.valueForHTTPHeaderField("Authorization") ?? ""
  107. XCTAssertEqual(authorizationHeader, "123456", "Authorization header is incorrect")
  108. XCTAssertNil(request.response, "response should be nil")
  109. }
  110. }
  111. // MARK: -
  112. class UploadDataTestCase: BaseTestCase {
  113. func testUploadDataRequest() {
  114. // Given
  115. let URLString = "https://httpbin.org/post"
  116. let data = "Lorem ipsum dolor sit amet".dataUsingEncoding(NSUTF8StringEncoding, allowLossyConversion: false)!
  117. let expectation = expectationWithDescription("Upload request should succeed: \(URLString)")
  118. var request: NSURLRequest?
  119. var response: NSHTTPURLResponse?
  120. var error: NSError?
  121. // When
  122. Alamofire.upload(.POST, URLString, data: data)
  123. .response { responseRequest, responseResponse, _, responseError in
  124. request = responseRequest
  125. response = responseResponse
  126. error = responseError
  127. expectation.fulfill()
  128. }
  129. waitForExpectationsWithTimeout(timeout, handler: nil)
  130. // Then
  131. XCTAssertNotNil(request, "request should not be nil")
  132. XCTAssertNotNil(response, "response should not be nil")
  133. XCTAssertNil(error, "error should be nil")
  134. }
  135. func testUploadDataRequestWithProgress() {
  136. // Given
  137. let URLString = "https://httpbin.org/post"
  138. let data: NSData = {
  139. var text = ""
  140. for _ in 1...3_000 {
  141. text += "Lorem ipsum dolor sit amet, consectetur adipiscing elit. "
  142. }
  143. return text.dataUsingEncoding(NSUTF8StringEncoding, allowLossyConversion: false)!
  144. }()
  145. let expectation = expectationWithDescription("Bytes upload progress should be reported: \(URLString)")
  146. var byteValues: [(bytes: Int64, totalBytes: Int64, totalBytesExpected: Int64)] = []
  147. var progressValues: [(completedUnitCount: Int64, totalUnitCount: Int64)] = []
  148. var responseRequest: NSURLRequest?
  149. var responseResponse: NSHTTPURLResponse?
  150. var responseData: NSData?
  151. var responseError: ErrorType?
  152. // When
  153. let upload = Alamofire.upload(.POST, URLString, data: data)
  154. upload.progress { bytesWritten, totalBytesWritten, totalBytesExpectedToWrite in
  155. let bytes = (bytes: bytesWritten, totalBytes: totalBytesWritten, totalBytesExpected: totalBytesExpectedToWrite)
  156. byteValues.append(bytes)
  157. let progress = (
  158. completedUnitCount: upload.progress.completedUnitCount,
  159. totalUnitCount: upload.progress.totalUnitCount
  160. )
  161. progressValues.append(progress)
  162. }
  163. upload.response { request, response, data, error in
  164. responseRequest = request
  165. responseResponse = response
  166. responseData = data
  167. responseError = error
  168. expectation.fulfill()
  169. }
  170. waitForExpectationsWithTimeout(timeout, handler: nil)
  171. // Then
  172. XCTAssertNotNil(responseRequest, "response request should not be nil")
  173. XCTAssertNotNil(responseResponse, "response response should not be nil")
  174. XCTAssertNotNil(responseData, "response data should not be nil")
  175. XCTAssertNil(responseError, "response error should be nil")
  176. XCTAssertEqual(byteValues.count, progressValues.count, "byteValues count should equal progressValues count")
  177. if byteValues.count == progressValues.count {
  178. for index in 0..<byteValues.count {
  179. let byteValue = byteValues[index]
  180. let progressValue = progressValues[index]
  181. XCTAssertGreaterThan(byteValue.bytes, 0, "reported bytes should always be greater than 0")
  182. XCTAssertEqual(
  183. byteValue.totalBytes,
  184. progressValue.completedUnitCount,
  185. "total bytes should be equal to completed unit count"
  186. )
  187. XCTAssertEqual(
  188. byteValue.totalBytesExpected,
  189. progressValue.totalUnitCount,
  190. "total bytes expected should be equal to total unit count"
  191. )
  192. }
  193. }
  194. if let
  195. lastByteValue = byteValues.last,
  196. lastProgressValue = progressValues.last
  197. {
  198. let byteValueFractionalCompletion = Double(lastByteValue.totalBytes) / Double(lastByteValue.totalBytesExpected)
  199. let progressValueFractionalCompletion = Double(lastProgressValue.0) / Double(lastProgressValue.1)
  200. XCTAssertEqual(byteValueFractionalCompletion, 1.0, "byte value fractional completion should equal 1.0")
  201. XCTAssertEqual(
  202. progressValueFractionalCompletion,
  203. 1.0,
  204. "progress value fractional completion should equal 1.0"
  205. )
  206. } else {
  207. XCTFail("last item in bytesValues and progressValues should not be nil")
  208. }
  209. }
  210. }
  211. // MARK: -
  212. class UploadMultipartFormDataTestCase: BaseTestCase {
  213. // MARK: Tests
  214. func testThatUploadingMultipartFormDataSetsContentTypeHeader() {
  215. // Given
  216. let URLString = "https://httpbin.org/post"
  217. let uploadData = "upload_data".dataUsingEncoding(NSUTF8StringEncoding, allowLossyConversion: false)!
  218. let expectation = expectationWithDescription("multipart form data upload should succeed")
  219. var formData: MultipartFormData?
  220. var request: NSURLRequest?
  221. var response: NSHTTPURLResponse?
  222. var data: NSData?
  223. var error: NSError?
  224. // When
  225. Alamofire.upload(
  226. .POST,
  227. URLString,
  228. multipartFormData: { multipartFormData in
  229. multipartFormData.appendBodyPart(data: uploadData, name: "upload_data")
  230. formData = multipartFormData
  231. },
  232. encodingCompletion: { result in
  233. switch result {
  234. case .Success(let upload, _, _):
  235. upload.response { responseRequest, responseResponse, responseData, responseError in
  236. request = responseRequest
  237. response = responseResponse
  238. data = responseData
  239. error = responseError
  240. expectation.fulfill()
  241. }
  242. case .Failure:
  243. expectation.fulfill()
  244. }
  245. }
  246. )
  247. waitForExpectationsWithTimeout(timeout, handler: nil)
  248. // Then
  249. XCTAssertNotNil(request, "request should not be nil")
  250. XCTAssertNotNil(response, "response should not be nil")
  251. XCTAssertNotNil(data, "data should not be nil")
  252. XCTAssertNil(error, "error should be nil")
  253. if let
  254. request = request,
  255. multipartFormData = formData,
  256. contentType = request.valueForHTTPHeaderField("Content-Type")
  257. {
  258. XCTAssertEqual(contentType, multipartFormData.contentType, "Content-Type header value should match")
  259. } else {
  260. XCTFail("Content-Type header value should not be nil")
  261. }
  262. }
  263. func testThatUploadingMultipartFormDataSucceedsWithDefaultParameters() {
  264. // Given
  265. let URLString = "https://httpbin.org/post"
  266. let french = "français".dataUsingEncoding(NSUTF8StringEncoding, allowLossyConversion: false)!
  267. let japanese = "日本語".dataUsingEncoding(NSUTF8StringEncoding, allowLossyConversion: false)!
  268. let expectation = expectationWithDescription("multipart form data upload should succeed")
  269. var request: NSURLRequest?
  270. var response: NSHTTPURLResponse?
  271. var data: NSData?
  272. var error: NSError?
  273. // When
  274. Alamofire.upload(
  275. .POST,
  276. URLString,
  277. multipartFormData: { multipartFormData in
  278. multipartFormData.appendBodyPart(data: french, name: "french")
  279. multipartFormData.appendBodyPart(data: japanese, name: "japanese")
  280. },
  281. encodingCompletion: { result in
  282. switch result {
  283. case .Success(let upload, _, _):
  284. upload.response { responseRequest, responseResponse, responseData, responseError in
  285. request = responseRequest
  286. response = responseResponse
  287. data = responseData
  288. error = responseError
  289. expectation.fulfill()
  290. }
  291. case .Failure:
  292. expectation.fulfill()
  293. }
  294. }
  295. )
  296. waitForExpectationsWithTimeout(timeout, handler: nil)
  297. // Then
  298. XCTAssertNotNil(request, "request should not be nil")
  299. XCTAssertNotNil(response, "response should not be nil")
  300. XCTAssertNotNil(data, "data should not be nil")
  301. XCTAssertNil(error, "error should be nil")
  302. }
  303. func testThatUploadingMultipartFormDataWhileStreamingFromMemoryMonitorsProgress() {
  304. executeMultipartFormDataUploadRequestWithProgress(streamFromDisk: false)
  305. }
  306. func testThatUploadingMultipartFormDataWhileStreamingFromDiskMonitorsProgress() {
  307. executeMultipartFormDataUploadRequestWithProgress(streamFromDisk: true)
  308. }
  309. func testThatUploadingMultipartFormDataBelowMemoryThresholdStreamsFromMemory() {
  310. // Given
  311. let URLString = "https://httpbin.org/post"
  312. let french = "français".dataUsingEncoding(NSUTF8StringEncoding, allowLossyConversion: false)!
  313. let japanese = "日本語".dataUsingEncoding(NSUTF8StringEncoding, allowLossyConversion: false)!
  314. let expectation = expectationWithDescription("multipart form data upload should succeed")
  315. var streamingFromDisk: Bool?
  316. var streamFileURL: NSURL?
  317. // When
  318. Alamofire.upload(
  319. .POST,
  320. URLString,
  321. multipartFormData: { multipartFormData in
  322. multipartFormData.appendBodyPart(data: french, name: "french")
  323. multipartFormData.appendBodyPart(data: japanese, name: "japanese")
  324. },
  325. encodingCompletion: { result in
  326. switch result {
  327. case let .Success(upload, uploadStreamingFromDisk, uploadStreamFileURL):
  328. streamingFromDisk = uploadStreamingFromDisk
  329. streamFileURL = uploadStreamFileURL
  330. upload.response { _, _, _, _ in
  331. expectation.fulfill()
  332. }
  333. case .Failure:
  334. expectation.fulfill()
  335. }
  336. }
  337. )
  338. waitForExpectationsWithTimeout(timeout, handler: nil)
  339. // Then
  340. XCTAssertNotNil(streamingFromDisk, "streaming from disk should not be nil")
  341. XCTAssertNil(streamFileURL, "stream file URL should be nil")
  342. if let streamingFromDisk = streamingFromDisk {
  343. XCTAssertFalse(streamingFromDisk, "streaming from disk should be false")
  344. }
  345. }
  346. func testThatUploadingMultipartFormDataBelowMemoryThresholdSetsContentTypeHeader() {
  347. // Given
  348. let URLString = "https://httpbin.org/post"
  349. let uploadData = "upload data".dataUsingEncoding(NSUTF8StringEncoding, allowLossyConversion: false)!
  350. let expectation = expectationWithDescription("multipart form data upload should succeed")
  351. var formData: MultipartFormData?
  352. var request: NSURLRequest?
  353. var streamingFromDisk: Bool?
  354. // When
  355. Alamofire.upload(
  356. .POST,
  357. URLString,
  358. multipartFormData: { multipartFormData in
  359. multipartFormData.appendBodyPart(data: uploadData, name: "upload_data")
  360. formData = multipartFormData
  361. },
  362. encodingCompletion: { result in
  363. switch result {
  364. case let .Success(upload, uploadStreamingFromDisk, _):
  365. streamingFromDisk = uploadStreamingFromDisk
  366. upload.response { responseRequest, _, _, _ in
  367. request = responseRequest
  368. expectation.fulfill()
  369. }
  370. case .Failure:
  371. expectation.fulfill()
  372. }
  373. }
  374. )
  375. waitForExpectationsWithTimeout(timeout, handler: nil)
  376. // Then
  377. XCTAssertNotNil(streamingFromDisk, "streaming from disk should not be nil")
  378. if let streamingFromDisk = streamingFromDisk {
  379. XCTAssertFalse(streamingFromDisk, "streaming from disk should be false")
  380. }
  381. if let
  382. request = request,
  383. multipartFormData = formData,
  384. contentType = request.valueForHTTPHeaderField("Content-Type")
  385. {
  386. XCTAssertEqual(contentType, multipartFormData.contentType, "Content-Type header value should match")
  387. } else {
  388. XCTFail("Content-Type header value should not be nil")
  389. }
  390. }
  391. func testThatUploadingMultipartFormDataAboveMemoryThresholdStreamsFromDisk() {
  392. // Given
  393. let URLString = "https://httpbin.org/post"
  394. let french = "français".dataUsingEncoding(NSUTF8StringEncoding, allowLossyConversion: false)!
  395. let japanese = "日本語".dataUsingEncoding(NSUTF8StringEncoding, allowLossyConversion: false)!
  396. let expectation = expectationWithDescription("multipart form data upload should succeed")
  397. var streamingFromDisk: Bool?
  398. var streamFileURL: NSURL?
  399. // When
  400. Alamofire.upload(
  401. .POST,
  402. URLString,
  403. multipartFormData: { multipartFormData in
  404. multipartFormData.appendBodyPart(data: french, name: "french")
  405. multipartFormData.appendBodyPart(data: japanese, name: "japanese")
  406. },
  407. encodingMemoryThreshold: 0,
  408. encodingCompletion: { result in
  409. switch result {
  410. case let .Success(upload, uploadStreamingFromDisk, uploadStreamFileURL):
  411. streamingFromDisk = uploadStreamingFromDisk
  412. streamFileURL = uploadStreamFileURL
  413. upload.response { _, _, _, _ in
  414. expectation.fulfill()
  415. }
  416. case .Failure:
  417. expectation.fulfill()
  418. }
  419. }
  420. )
  421. waitForExpectationsWithTimeout(timeout, handler: nil)
  422. // Then
  423. XCTAssertNotNil(streamingFromDisk, "streaming from disk should not be nil")
  424. XCTAssertNotNil(streamFileURL, "stream file URL should not be nil")
  425. if let
  426. streamingFromDisk = streamingFromDisk,
  427. streamFilePath = streamFileURL?.path
  428. {
  429. XCTAssertTrue(streamingFromDisk, "streaming from disk should be true")
  430. XCTAssertTrue(
  431. NSFileManager.defaultManager().fileExistsAtPath(streamFilePath),
  432. "stream file path should exist"
  433. )
  434. }
  435. }
  436. func testThatUploadingMultipartFormDataAboveMemoryThresholdSetsContentTypeHeader() {
  437. // Given
  438. let URLString = "https://httpbin.org/post"
  439. let uploadData = "upload data".dataUsingEncoding(NSUTF8StringEncoding, allowLossyConversion: false)!
  440. let expectation = expectationWithDescription("multipart form data upload should succeed")
  441. var formData: MultipartFormData?
  442. var request: NSURLRequest?
  443. var streamingFromDisk: Bool?
  444. // When
  445. Alamofire.upload(
  446. .POST,
  447. URLString,
  448. multipartFormData: { multipartFormData in
  449. multipartFormData.appendBodyPart(data: uploadData, name: "upload_data")
  450. formData = multipartFormData
  451. },
  452. encodingMemoryThreshold: 0,
  453. encodingCompletion: { result in
  454. switch result {
  455. case let .Success(upload, uploadStreamingFromDisk, _):
  456. streamingFromDisk = uploadStreamingFromDisk
  457. upload.response { responseRequest, _, _, _ in
  458. request = responseRequest
  459. expectation.fulfill()
  460. }
  461. case .Failure:
  462. expectation.fulfill()
  463. }
  464. }
  465. )
  466. waitForExpectationsWithTimeout(timeout, handler: nil)
  467. // Then
  468. XCTAssertNotNil(streamingFromDisk, "streaming from disk should not be nil")
  469. if let streamingFromDisk = streamingFromDisk {
  470. XCTAssertTrue(streamingFromDisk, "streaming from disk should be true")
  471. }
  472. if let
  473. request = request,
  474. multipartFormData = formData,
  475. contentType = request.valueForHTTPHeaderField("Content-Type")
  476. {
  477. XCTAssertEqual(contentType, multipartFormData.contentType, "Content-Type header value should match")
  478. } else {
  479. XCTFail("Content-Type header value should not be nil")
  480. }
  481. }
  482. func testThatUploadingMultipartFormDataOnBackgroundSessionWritesDataToFileToAvoidCrash() {
  483. // Given
  484. let manager: Manager = {
  485. let identifier = "com.alamofire.uploadtests.\(NSUUID().UUIDString)"
  486. let configuration = NSURLSessionConfiguration.backgroundSessionConfigurationForAllPlatformsWithIdentifier(identifier)
  487. return Manager(configuration: configuration, serverTrustPolicyManager: nil)
  488. }()
  489. let URLString = "https://httpbin.org/post"
  490. let french = "français".dataUsingEncoding(NSUTF8StringEncoding, allowLossyConversion: false)!
  491. let japanese = "日本語".dataUsingEncoding(NSUTF8StringEncoding, allowLossyConversion: false)!
  492. let expectation = expectationWithDescription("multipart form data upload should succeed")
  493. var request: NSURLRequest?
  494. var response: NSHTTPURLResponse?
  495. var data: NSData?
  496. var error: NSError?
  497. var streamingFromDisk: Bool?
  498. // When
  499. manager.upload(
  500. .POST,
  501. URLString,
  502. multipartFormData: { multipartFormData in
  503. multipartFormData.appendBodyPart(data: french, name: "french")
  504. multipartFormData.appendBodyPart(data: japanese, name: "japanese")
  505. },
  506. encodingCompletion: { result in
  507. switch result {
  508. case let .Success(upload, uploadStreamingFromDisk, _):
  509. streamingFromDisk = uploadStreamingFromDisk
  510. upload.response { responseRequest, responseResponse, responseData, responseError in
  511. request = responseRequest
  512. response = responseResponse
  513. data = responseData
  514. error = responseError
  515. expectation.fulfill()
  516. }
  517. case .Failure:
  518. expectation.fulfill()
  519. }
  520. }
  521. )
  522. waitForExpectationsWithTimeout(timeout, handler: nil)
  523. // Then
  524. XCTAssertNotNil(request, "request should not be nil")
  525. XCTAssertNotNil(response, "response should not be nil")
  526. XCTAssertNotNil(data, "data should not be nil")
  527. XCTAssertNil(error, "error should be nil")
  528. if let streamingFromDisk = streamingFromDisk {
  529. XCTAssertTrue(streamingFromDisk, "streaming from disk should be true")
  530. } else {
  531. XCTFail("streaming from disk should not be nil")
  532. }
  533. }
  534. // MARK: Combined Test Execution
  535. private func executeMultipartFormDataUploadRequestWithProgress(streamFromDisk streamFromDisk: Bool) {
  536. // Given
  537. let URLString = "https://httpbin.org/post"
  538. let loremData1: NSData = {
  539. var loremValues: [String] = []
  540. for _ in 1...1_500 {
  541. loremValues.append("Lorem ipsum dolor sit amet, consectetur adipiscing elit.")
  542. }
  543. return loremValues.joinWithSeparator(" ").dataUsingEncoding(NSUTF8StringEncoding, allowLossyConversion: false)!
  544. }()
  545. let loremData2: NSData = {
  546. var loremValues: [String] = []
  547. for _ in 1...1_500 {
  548. loremValues.append("Lorem ipsum dolor sit amet, nam no graeco recusabo appellantur.")
  549. }
  550. return loremValues.joinWithSeparator(" ").dataUsingEncoding(NSUTF8StringEncoding, allowLossyConversion: false)!
  551. }()
  552. let expectation = expectationWithDescription("multipart form data upload should succeed")
  553. var byteValues: [(bytes: Int64, totalBytes: Int64, totalBytesExpected: Int64)] = []
  554. var progressValues: [(completedUnitCount: Int64, totalUnitCount: Int64)] = []
  555. var request: NSURLRequest?
  556. var response: NSHTTPURLResponse?
  557. var data: NSData?
  558. var error: NSError?
  559. // When
  560. Alamofire.upload(
  561. .POST,
  562. URLString,
  563. multipartFormData: { multipartFormData in
  564. multipartFormData.appendBodyPart(data: loremData1, name: "lorem1")
  565. multipartFormData.appendBodyPart(data: loremData2, name: "lorem2")
  566. },
  567. encodingMemoryThreshold: streamFromDisk ? 0 : 100_000_000,
  568. encodingCompletion: { result in
  569. switch result {
  570. case .Success(let upload, _, _):
  571. upload.progress { bytesWritten, totalBytesWritten, totalBytesExpectedToWrite in
  572. let bytes = (
  573. bytes: bytesWritten,
  574. totalBytes: totalBytesWritten,
  575. totalBytesExpected: totalBytesExpectedToWrite
  576. )
  577. byteValues.append(bytes)
  578. let progress = (
  579. completedUnitCount: upload.progress.completedUnitCount,
  580. totalUnitCount: upload.progress.totalUnitCount
  581. )
  582. progressValues.append(progress)
  583. }
  584. upload.response { responseRequest, responseResponse, responseData, responseError in
  585. request = responseRequest
  586. response = responseResponse
  587. data = responseData
  588. error = responseError
  589. expectation.fulfill()
  590. }
  591. case .Failure:
  592. expectation.fulfill()
  593. }
  594. }
  595. )
  596. waitForExpectationsWithTimeout(timeout, handler: nil)
  597. // Then
  598. XCTAssertNotNil(request, "request should not be nil")
  599. XCTAssertNotNil(response, "response should not be nil")
  600. XCTAssertNotNil(data, "data should not be nil")
  601. XCTAssertNil(error, "error should be nil")
  602. XCTAssertEqual(byteValues.count, progressValues.count, "byteValues count should equal progressValues count")
  603. if byteValues.count == progressValues.count {
  604. for index in 0..<byteValues.count {
  605. let byteValue = byteValues[index]
  606. let progressValue = progressValues[index]
  607. XCTAssertGreaterThan(byteValue.bytes, 0, "reported bytes should always be greater than 0")
  608. XCTAssertEqual(
  609. byteValue.totalBytes,
  610. progressValue.completedUnitCount,
  611. "total bytes should be equal to completed unit count"
  612. )
  613. XCTAssertEqual(
  614. byteValue.totalBytesExpected,
  615. progressValue.totalUnitCount,
  616. "total bytes expected should be equal to total unit count"
  617. )
  618. }
  619. }
  620. if let
  621. lastByteValue = byteValues.last,
  622. lastProgressValue = progressValues.last
  623. {
  624. let byteValueFractionalCompletion = Double(lastByteValue.totalBytes) / Double(lastByteValue.totalBytesExpected)
  625. let progressValueFractionalCompletion = Double(lastProgressValue.0) / Double(lastProgressValue.1)
  626. XCTAssertEqual(byteValueFractionalCompletion, 1.0, "byte value fractional completion should equal 1.0")
  627. XCTAssertEqual(
  628. progressValueFractionalCompletion,
  629. 1.0,
  630. "progress value fractional completion should equal 1.0"
  631. )
  632. } else {
  633. XCTFail("last item in bytesValues and progressValues should not be nil")
  634. }
  635. }
  636. }