UploadTests.swift 30 KB

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