SessionManagerTests.swift 30 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862
  1. //
  2. // SessionManagerTests.swift
  3. //
  4. // Copyright (c) 2014-2018 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. @testable import Alamofire
  25. import Foundation
  26. import XCTest
  27. class SessionManagerTestCase: BaseTestCase {
  28. // MARK: Helper Types
  29. private class HTTPMethodAdapter: RequestAdapter {
  30. let method: HTTPMethod
  31. let throwsError: Bool
  32. init(method: HTTPMethod, throwsError: Bool = false) {
  33. self.method = method
  34. self.throwsError = throwsError
  35. }
  36. func adapt(_ urlRequest: URLRequest) throws -> URLRequest {
  37. guard !throwsError else { throw AFError.invalidURL(url: "") }
  38. var urlRequest = urlRequest
  39. urlRequest.httpMethod = method.rawValue
  40. return urlRequest
  41. }
  42. }
  43. private class RequestHandler: RequestAdapter, RequestRetrier {
  44. var adaptedCount = 0
  45. var retryCount = 0
  46. var retryErrors: [Error] = []
  47. var shouldApplyAuthorizationHeader = false
  48. var throwsErrorOnSecondAdapt = false
  49. func adapt(_ urlRequest: URLRequest) throws -> URLRequest {
  50. if throwsErrorOnSecondAdapt && adaptedCount == 1 {
  51. throwsErrorOnSecondAdapt = false
  52. throw AFError.invalidURL(url: "")
  53. }
  54. var urlRequest = urlRequest
  55. adaptedCount += 1
  56. if shouldApplyAuthorizationHeader && adaptedCount > 1 {
  57. if let header = Request.authorizationHeader(user: "user", password: "password") {
  58. urlRequest.setValue(header.value, forHTTPHeaderField: header.key)
  59. }
  60. }
  61. return urlRequest
  62. }
  63. func should(_ manager: SessionManager, retry request: Request, with error: Error, completion: @escaping RequestRetryCompletion) {
  64. retryCount += 1
  65. retryErrors.append(error)
  66. if retryCount < 2 {
  67. completion(true, 0.0)
  68. } else {
  69. completion(false, 0.0)
  70. }
  71. }
  72. }
  73. private class UploadHandler: RequestAdapter, RequestRetrier {
  74. var adaptedCount = 0
  75. var retryCount = 0
  76. var retryErrors: [Error] = []
  77. func adapt(_ urlRequest: URLRequest) throws -> URLRequest {
  78. adaptedCount += 1
  79. if adaptedCount == 1 { throw AFError.invalidURL(url: "") }
  80. return urlRequest
  81. }
  82. func should(_ manager: SessionManager, retry request: Request, with error: Error, completion: @escaping RequestRetryCompletion) {
  83. retryCount += 1
  84. retryErrors.append(error)
  85. completion(true, 0.0)
  86. }
  87. }
  88. // MARK: Tests - Initialization
  89. func testInitializerWithDefaultArguments() {
  90. // Given, When
  91. let manager = SessionManager()
  92. // Then
  93. XCTAssertNotNil(manager.session.delegate, "session delegate should not be nil")
  94. XCTAssertTrue(manager.delegate === manager.session.delegate, "manager delegate should equal session delegate")
  95. XCTAssertNil(manager.session.serverTrustPolicyManager, "session server trust policy manager should be nil")
  96. }
  97. func testInitializerWithSpecifiedArguments() {
  98. // Given
  99. let configuration = URLSessionConfiguration.default
  100. let delegate = SessionDelegate()
  101. let serverTrustPolicyManager = ServerTrustPolicyManager(policies: [:])
  102. // When
  103. let manager = SessionManager(
  104. configuration: configuration,
  105. delegate: delegate,
  106. serverTrustPolicyManager: serverTrustPolicyManager
  107. )
  108. // Then
  109. XCTAssertNotNil(manager.session.delegate, "session delegate should not be nil")
  110. XCTAssertTrue(manager.delegate === manager.session.delegate, "manager delegate should equal session delegate")
  111. XCTAssertNotNil(manager.session.serverTrustPolicyManager, "session server trust policy manager should not be nil")
  112. }
  113. func testThatFailableInitializerSucceedsWithDefaultArguments() {
  114. // Given
  115. let delegate = SessionDelegate()
  116. let session: URLSession = {
  117. let configuration = URLSessionConfiguration.default
  118. return URLSession(configuration: configuration, delegate: delegate, delegateQueue: nil)
  119. }()
  120. // When
  121. let manager = SessionManager(session: session, delegate: delegate)
  122. // Then
  123. if let manager = manager {
  124. XCTAssertTrue(manager.delegate === manager.session.delegate, "manager delegate should equal session delegate")
  125. XCTAssertNil(manager.session.serverTrustPolicyManager, "session server trust policy manager should be nil")
  126. } else {
  127. XCTFail("manager should not be nil")
  128. }
  129. }
  130. func testThatFailableInitializerSucceedsWithSpecifiedArguments() {
  131. // Given
  132. let delegate = SessionDelegate()
  133. let session: URLSession = {
  134. let configuration = URLSessionConfiguration.default
  135. return URLSession(configuration: configuration, delegate: delegate, delegateQueue: nil)
  136. }()
  137. let serverTrustPolicyManager = ServerTrustPolicyManager(policies: [:])
  138. // When
  139. let manager = SessionManager(session: session, delegate: delegate, serverTrustPolicyManager: serverTrustPolicyManager)
  140. // Then
  141. if let manager = manager {
  142. XCTAssertTrue(manager.delegate === manager.session.delegate, "manager delegate should equal session delegate")
  143. XCTAssertNotNil(manager.session.serverTrustPolicyManager, "session server trust policy manager should not be nil")
  144. } else {
  145. XCTFail("manager should not be nil")
  146. }
  147. }
  148. func testThatFailableInitializerFailsWithWhenDelegateDoesNotEqualSessionDelegate() {
  149. // Given
  150. let delegate = SessionDelegate()
  151. let session: URLSession = {
  152. let configuration = URLSessionConfiguration.default
  153. return URLSession(configuration: configuration, delegate: SessionDelegate(), delegateQueue: nil)
  154. }()
  155. // When
  156. let manager = SessionManager(session: session, delegate: delegate)
  157. // Then
  158. XCTAssertNil(manager, "manager should be nil")
  159. }
  160. func testThatFailableInitializerFailsWhenSessionDelegateIsNil() {
  161. // Given
  162. let delegate = SessionDelegate()
  163. let session: URLSession = {
  164. let configuration = URLSessionConfiguration.default
  165. return URLSession(configuration: configuration, delegate: nil, delegateQueue: nil)
  166. }()
  167. // When
  168. let manager = SessionManager(session: session, delegate: delegate)
  169. // Then
  170. XCTAssertNil(manager, "manager should be nil")
  171. }
  172. // MARK: Tests - Default HTTP Headers
  173. func testDefaultUserAgentHeader() {
  174. // Given, When
  175. let userAgent = SessionManager.defaultHTTPHeaders["User-Agent"]
  176. // Then
  177. let osNameVersion: String = {
  178. let version = ProcessInfo.processInfo.operatingSystemVersion
  179. let versionString = "\(version.majorVersion).\(version.minorVersion).\(version.patchVersion)"
  180. let osName: String = {
  181. #if os(iOS)
  182. return "iOS"
  183. #elseif os(watchOS)
  184. return "watchOS"
  185. #elseif os(tvOS)
  186. return "tvOS"
  187. #elseif os(macOS)
  188. return "OS X"
  189. #elseif os(Linux)
  190. return "Linux"
  191. #else
  192. return "Unknown"
  193. #endif
  194. }()
  195. return "\(osName) \(versionString)"
  196. }()
  197. let alamofireVersion: String = {
  198. guard
  199. let afInfo = Bundle(for: SessionManager.self).infoDictionary,
  200. let build = afInfo["CFBundleShortVersionString"]
  201. else { return "Unknown" }
  202. return "Alamofire/\(build)"
  203. }()
  204. let expectedUserAgent = "Unknown/Unknown (Unknown; build:Unknown; \(osNameVersion)) \(alamofireVersion)"
  205. XCTAssertEqual(userAgent, expectedUserAgent)
  206. }
  207. // MARK: Tests - Start Requests Immediately
  208. func testSetStartRequestsImmediatelyToFalseAndResumeRequest() {
  209. // Given
  210. let manager = SessionManager()
  211. manager.startRequestsImmediately = false
  212. let url = URL(string: "https://httpbin.org/get")!
  213. let urlRequest = URLRequest(url: url)
  214. let expectation = self.expectation(description: "\(url)")
  215. var response: HTTPURLResponse?
  216. // When
  217. manager.request(urlRequest)
  218. .response { resp in
  219. response = resp.response
  220. expectation.fulfill()
  221. }
  222. .resume()
  223. waitForExpectations(timeout: timeout, handler: nil)
  224. // Then
  225. XCTAssertNotNil(response, "response should not be nil")
  226. XCTAssertTrue(response?.statusCode == 200, "response status code should be 200")
  227. }
  228. // MARK: Tests - Deinitialization
  229. func testReleasingManagerWithPendingRequestDeinitializesSuccessfully() {
  230. // Given
  231. var manager: SessionManager? = SessionManager()
  232. manager?.startRequestsImmediately = false
  233. let url = URL(string: "https://httpbin.org/get")!
  234. let urlRequest = URLRequest(url: url)
  235. // When
  236. let request = manager?.request(urlRequest)
  237. manager = nil
  238. // Then
  239. XCTAssertTrue(request?.task?.state == .suspended, "request task state should be '.Suspended'")
  240. XCTAssertNil(manager, "manager should be nil")
  241. }
  242. func testReleasingManagerWithPendingCanceledRequestDeinitializesSuccessfully() {
  243. // Given
  244. var manager: SessionManager? = SessionManager()
  245. manager!.startRequestsImmediately = false
  246. let url = URL(string: "https://httpbin.org/get")!
  247. let urlRequest = URLRequest(url: url)
  248. // When
  249. let request = manager!.request(urlRequest)
  250. request.cancel()
  251. manager = nil
  252. // Then
  253. let state = request.task?.state
  254. XCTAssertTrue(state == .canceling || state == .completed, "state should be .Canceling or .Completed")
  255. XCTAssertNil(manager, "manager should be nil")
  256. }
  257. // MARK: Tests - Bad Requests
  258. func testThatDataRequestWithInvalidURLStringThrowsResponseHandlerError() {
  259. // Given
  260. let sessionManager = SessionManager()
  261. let expectation = self.expectation(description: "Request should fail with error")
  262. var response: DefaultDataResponse?
  263. // When
  264. sessionManager.request("https://httpbin.org/get/äëïöü").response { resp in
  265. response = resp
  266. expectation.fulfill()
  267. }
  268. waitForExpectations(timeout: timeout, handler: nil)
  269. // Then
  270. XCTAssertNil(response?.request)
  271. XCTAssertNil(response?.response)
  272. XCTAssertNotNil(response?.data)
  273. XCTAssertEqual(response?.data?.count, 0)
  274. XCTAssertNotNil(response?.error)
  275. if let error = response?.error as? AFError {
  276. XCTAssertTrue(error.isInvalidURLError)
  277. XCTAssertEqual(error.urlConvertible as? String, "https://httpbin.org/get/äëïöü")
  278. } else {
  279. XCTFail("error should not be nil")
  280. }
  281. }
  282. func testThatDownloadRequestWithInvalidURLStringThrowsResponseHandlerError() {
  283. // Given
  284. let sessionManager = SessionManager()
  285. let expectation = self.expectation(description: "Download should fail with error")
  286. var response: DefaultDownloadResponse?
  287. // When
  288. sessionManager.download("https://httpbin.org/get/äëïöü").response { resp in
  289. response = resp
  290. expectation.fulfill()
  291. }
  292. waitForExpectations(timeout: timeout, handler: nil)
  293. // Then
  294. XCTAssertNil(response?.request)
  295. XCTAssertNil(response?.response)
  296. XCTAssertNil(response?.temporaryURL)
  297. XCTAssertNil(response?.destinationURL)
  298. XCTAssertNil(response?.resumeData)
  299. XCTAssertNotNil(response?.error)
  300. if let error = response?.error as? AFError {
  301. XCTAssertTrue(error.isInvalidURLError)
  302. XCTAssertEqual(error.urlConvertible as? String, "https://httpbin.org/get/äëïöü")
  303. } else {
  304. XCTFail("error should not be nil")
  305. }
  306. }
  307. func testThatUploadDataRequestWithInvalidURLStringThrowsResponseHandlerError() {
  308. // Given
  309. let sessionManager = SessionManager()
  310. let expectation = self.expectation(description: "Upload should fail with error")
  311. var response: DefaultDataResponse?
  312. // When
  313. sessionManager.upload(Data(), to: "https://httpbin.org/get/äëïöü").response { resp in
  314. response = resp
  315. expectation.fulfill()
  316. }
  317. waitForExpectations(timeout: timeout, handler: nil)
  318. // Then
  319. XCTAssertNil(response?.request)
  320. XCTAssertNil(response?.response)
  321. XCTAssertNotNil(response?.data)
  322. XCTAssertEqual(response?.data?.count, 0)
  323. XCTAssertNotNil(response?.error)
  324. if let error = response?.error as? AFError {
  325. XCTAssertTrue(error.isInvalidURLError)
  326. XCTAssertEqual(error.urlConvertible as? String, "https://httpbin.org/get/äëïöü")
  327. } else {
  328. XCTFail("error should not be nil")
  329. }
  330. }
  331. func testThatUploadFileRequestWithInvalidURLStringThrowsResponseHandlerError() {
  332. // Given
  333. let sessionManager = SessionManager()
  334. let expectation = self.expectation(description: "Upload should fail with error")
  335. var response: DefaultDataResponse?
  336. // When
  337. sessionManager.upload(URL(fileURLWithPath: "/invalid"), to: "https://httpbin.org/get/äëïöü").response { resp in
  338. response = resp
  339. expectation.fulfill()
  340. }
  341. waitForExpectations(timeout: timeout, handler: nil)
  342. // Then
  343. XCTAssertNil(response?.request)
  344. XCTAssertNil(response?.response)
  345. XCTAssertNotNil(response?.data)
  346. XCTAssertEqual(response?.data?.count, 0)
  347. XCTAssertNotNil(response?.error)
  348. if let error = response?.error as? AFError {
  349. XCTAssertTrue(error.isInvalidURLError)
  350. XCTAssertEqual(error.urlConvertible as? String, "https://httpbin.org/get/äëïöü")
  351. } else {
  352. XCTFail("error should not be nil")
  353. }
  354. }
  355. func testThatUploadStreamRequestWithInvalidURLStringThrowsResponseHandlerError() {
  356. // Given
  357. let sessionManager = SessionManager()
  358. let expectation = self.expectation(description: "Upload should fail with error")
  359. var response: DefaultDataResponse?
  360. // When
  361. sessionManager.upload(InputStream(data: Data()), to: "https://httpbin.org/get/äëïöü").response { resp in
  362. response = resp
  363. expectation.fulfill()
  364. }
  365. waitForExpectations(timeout: timeout, handler: nil)
  366. // Then
  367. XCTAssertNil(response?.request)
  368. XCTAssertNil(response?.response)
  369. XCTAssertNotNil(response?.data)
  370. XCTAssertEqual(response?.data?.count, 0)
  371. XCTAssertNotNil(response?.error)
  372. if let error = response?.error as? AFError {
  373. XCTAssertTrue(error.isInvalidURLError)
  374. XCTAssertEqual(error.urlConvertible as? String, "https://httpbin.org/get/äëïöü")
  375. } else {
  376. XCTFail("error should not be nil")
  377. }
  378. }
  379. // MARK: Tests - Request Adapter
  380. func testThatSessionManagerCallsRequestAdapterWhenCreatingDataRequest() {
  381. // Given
  382. let adapter = HTTPMethodAdapter(method: .post)
  383. let sessionManager = SessionManager()
  384. sessionManager.adapter = adapter
  385. sessionManager.startRequestsImmediately = false
  386. // When
  387. let request = sessionManager.request("https://httpbin.org/get")
  388. // Then
  389. XCTAssertEqual(request.task?.originalRequest?.httpMethod, adapter.method.rawValue)
  390. }
  391. func testThatSessionManagerCallsRequestAdapterWhenCreatingDownloadRequest() {
  392. // Given
  393. let adapter = HTTPMethodAdapter(method: .post)
  394. let sessionManager = SessionManager()
  395. sessionManager.adapter = adapter
  396. sessionManager.startRequestsImmediately = false
  397. // When
  398. let destination = DownloadRequest.suggestedDownloadDestination()
  399. let request = sessionManager.download("https://httpbin.org/get", to: destination)
  400. // Then
  401. XCTAssertEqual(request.task?.originalRequest?.httpMethod, adapter.method.rawValue)
  402. }
  403. func testThatSessionManagerCallsRequestAdapterWhenCreatingUploadRequestWithData() {
  404. // Given
  405. let adapter = HTTPMethodAdapter(method: .get)
  406. let sessionManager = SessionManager()
  407. sessionManager.adapter = adapter
  408. sessionManager.startRequestsImmediately = false
  409. // When
  410. let request = sessionManager.upload("data".data(using: .utf8)!, to: "https://httpbin.org/post")
  411. // Then
  412. XCTAssertEqual(request.task?.originalRequest?.httpMethod, adapter.method.rawValue)
  413. }
  414. func testThatSessionManagerCallsRequestAdapterWhenCreatingUploadRequestWithFile() {
  415. // Given
  416. let adapter = HTTPMethodAdapter(method: .get)
  417. let sessionManager = SessionManager()
  418. sessionManager.adapter = adapter
  419. sessionManager.startRequestsImmediately = false
  420. // When
  421. let fileURL = URL(fileURLWithPath: "/path/to/some/file.txt")
  422. let request = sessionManager.upload(fileURL, to: "https://httpbin.org/post")
  423. // Then
  424. XCTAssertEqual(request.task?.originalRequest?.httpMethod, adapter.method.rawValue)
  425. }
  426. func testThatSessionManagerCallsRequestAdapterWhenCreatingUploadRequestWithInputStream() {
  427. // Given
  428. let adapter = HTTPMethodAdapter(method: .get)
  429. let sessionManager = SessionManager()
  430. sessionManager.adapter = adapter
  431. sessionManager.startRequestsImmediately = false
  432. // When
  433. let inputStream = InputStream(data: "data".data(using: .utf8)!)
  434. let request = sessionManager.upload(inputStream, to: "https://httpbin.org/post")
  435. // Then
  436. XCTAssertEqual(request.task?.originalRequest?.httpMethod, adapter.method.rawValue)
  437. }
  438. func testThatRequestAdapterErrorThrowsResponseHandlerError() {
  439. // Given
  440. let adapter = HTTPMethodAdapter(method: .post, throwsError: true)
  441. let sessionManager = SessionManager()
  442. sessionManager.adapter = adapter
  443. sessionManager.startRequestsImmediately = false
  444. // When
  445. let request = sessionManager.request("https://httpbin.org/get")
  446. // Then
  447. if let error = request.delegate.error as? AFError {
  448. XCTAssertTrue(error.isInvalidURLError)
  449. XCTAssertEqual(error.urlConvertible as? String, "")
  450. } else {
  451. XCTFail("error should not be nil")
  452. }
  453. }
  454. // MARK: Tests - Request Retrier
  455. func testThatSessionManagerCallsRequestRetrierWhenRequestEncountersError() {
  456. // Given
  457. let handler = RequestHandler()
  458. let sessionManager = SessionManager()
  459. sessionManager.adapter = handler
  460. sessionManager.retrier = handler
  461. let expectation = self.expectation(description: "request should eventually fail")
  462. var response: DataResponse<Any>?
  463. // When
  464. let request = sessionManager.request("https://httpbin.org/basic-auth/user/password")
  465. .validate()
  466. .responseJSON { jsonResponse in
  467. response = jsonResponse
  468. expectation.fulfill()
  469. }
  470. waitForExpectations(timeout: timeout, handler: nil)
  471. // Then
  472. XCTAssertEqual(handler.adaptedCount, 2)
  473. XCTAssertEqual(handler.retryCount, 2)
  474. XCTAssertEqual(request.retryCount, 1)
  475. XCTAssertEqual(response?.result.isSuccess, false)
  476. XCTAssertTrue(sessionManager.delegate.requests.isEmpty)
  477. }
  478. func testThatSessionManagerCallsRequestRetrierWhenRequestInitiallyEncountersAdaptError() {
  479. // Given
  480. let handler = RequestHandler()
  481. handler.adaptedCount = 1
  482. handler.throwsErrorOnSecondAdapt = true
  483. handler.shouldApplyAuthorizationHeader = true
  484. let sessionManager = SessionManager()
  485. sessionManager.adapter = handler
  486. sessionManager.retrier = handler
  487. let expectation = self.expectation(description: "request should eventually fail")
  488. var response: DataResponse<Any>?
  489. // When
  490. sessionManager.request("https://httpbin.org/basic-auth/user/password")
  491. .validate()
  492. .responseJSON { jsonResponse in
  493. response = jsonResponse
  494. expectation.fulfill()
  495. }
  496. waitForExpectations(timeout: timeout, handler: nil)
  497. // Then
  498. XCTAssertEqual(handler.adaptedCount, 2)
  499. XCTAssertEqual(handler.retryCount, 1)
  500. XCTAssertEqual(response?.result.isSuccess, true)
  501. XCTAssertTrue(sessionManager.delegate.requests.isEmpty)
  502. handler.retryErrors.forEach { XCTAssertFalse($0 is AdaptError) }
  503. }
  504. func testThatSessionManagerCallsRequestRetrierWhenDownloadInitiallyEncountersAdaptError() {
  505. // Given
  506. let handler = RequestHandler()
  507. handler.adaptedCount = 1
  508. handler.throwsErrorOnSecondAdapt = true
  509. handler.shouldApplyAuthorizationHeader = true
  510. let sessionManager = SessionManager()
  511. sessionManager.adapter = handler
  512. sessionManager.retrier = handler
  513. let expectation = self.expectation(description: "request should eventually fail")
  514. var response: DownloadResponse<Any>?
  515. let destination: DownloadRequest.DownloadFileDestination = { _, _ in
  516. let fileURL = self.testDirectoryURL.appendingPathComponent("test-output.json")
  517. return (fileURL, [.removePreviousFile])
  518. }
  519. // When
  520. sessionManager.download("https://httpbin.org/basic-auth/user/password", to: destination)
  521. .validate()
  522. .responseJSON { jsonResponse in
  523. response = jsonResponse
  524. expectation.fulfill()
  525. }
  526. waitForExpectations(timeout: timeout, handler: nil)
  527. // Then
  528. XCTAssertEqual(handler.adaptedCount, 2)
  529. XCTAssertEqual(handler.retryCount, 1)
  530. XCTAssertEqual(response?.result.isSuccess, true)
  531. XCTAssertTrue(sessionManager.delegate.requests.isEmpty)
  532. handler.retryErrors.forEach { XCTAssertFalse($0 is AdaptError) }
  533. }
  534. func testThatSessionManagerCallsRequestRetrierWhenUploadInitiallyEncountersAdaptError() {
  535. // Given
  536. let handler = UploadHandler()
  537. let sessionManager = SessionManager()
  538. sessionManager.adapter = handler
  539. sessionManager.retrier = handler
  540. let expectation = self.expectation(description: "request should eventually fail")
  541. var response: DataResponse<Any>?
  542. let uploadData = "upload data".data(using: .utf8, allowLossyConversion: false)!
  543. // When
  544. sessionManager.upload(uploadData, to: "https://httpbin.org/post")
  545. .validate()
  546. .responseJSON { jsonResponse in
  547. response = jsonResponse
  548. expectation.fulfill()
  549. }
  550. waitForExpectations(timeout: timeout, handler: nil)
  551. // Then
  552. XCTAssertEqual(handler.adaptedCount, 2)
  553. XCTAssertEqual(handler.retryCount, 1)
  554. XCTAssertEqual(response?.result.isSuccess, true)
  555. XCTAssertTrue(sessionManager.delegate.requests.isEmpty)
  556. handler.retryErrors.forEach { XCTAssertFalse($0 is AdaptError) }
  557. }
  558. func testThatSessionManagerCallsAdapterWhenRequestIsRetried() {
  559. // Given
  560. let handler = RequestHandler()
  561. handler.shouldApplyAuthorizationHeader = true
  562. let sessionManager = SessionManager()
  563. sessionManager.adapter = handler
  564. sessionManager.retrier = handler
  565. let expectation = self.expectation(description: "request should eventually fail")
  566. var response: DataResponse<Any>?
  567. // When
  568. let request = sessionManager.request("https://httpbin.org/basic-auth/user/password")
  569. .validate()
  570. .responseJSON { jsonResponse in
  571. response = jsonResponse
  572. expectation.fulfill()
  573. }
  574. waitForExpectations(timeout: timeout, handler: nil)
  575. // Then
  576. XCTAssertEqual(handler.adaptedCount, 2)
  577. XCTAssertEqual(handler.retryCount, 1)
  578. XCTAssertEqual(request.retryCount, 1)
  579. XCTAssertEqual(response?.result.isSuccess, true)
  580. XCTAssertTrue(sessionManager.delegate.requests.isEmpty)
  581. }
  582. func testThatRequestAdapterErrorThrowsResponseHandlerErrorWhenRequestIsRetried() {
  583. // Given
  584. let handler = RequestHandler()
  585. handler.throwsErrorOnSecondAdapt = true
  586. let sessionManager = SessionManager()
  587. sessionManager.adapter = handler
  588. sessionManager.retrier = handler
  589. let expectation = self.expectation(description: "request should eventually fail")
  590. var response: DataResponse<Any>?
  591. // When
  592. let request = sessionManager.request("https://httpbin.org/basic-auth/user/password")
  593. .validate()
  594. .responseJSON { jsonResponse in
  595. response = jsonResponse
  596. expectation.fulfill()
  597. }
  598. waitForExpectations(timeout: timeout, handler: nil)
  599. // Then
  600. XCTAssertEqual(handler.adaptedCount, 1)
  601. XCTAssertEqual(handler.retryCount, 1)
  602. XCTAssertEqual(request.retryCount, 0)
  603. XCTAssertEqual(response?.result.isSuccess, false)
  604. XCTAssertTrue(sessionManager.delegate.requests.isEmpty)
  605. if let error = response?.result.error as? AFError {
  606. XCTAssertTrue(error.isInvalidURLError)
  607. XCTAssertEqual(error.urlConvertible as? String, "")
  608. } else {
  609. XCTFail("error should not be nil")
  610. }
  611. }
  612. }
  613. // MARK: -
  614. class SessionManagerConfigurationHeadersTestCase: BaseTestCase {
  615. enum ConfigurationType {
  616. case `default`, ephemeral, background
  617. }
  618. func testThatDefaultConfigurationHeadersAreSentWithRequest() {
  619. // Given, When, Then
  620. executeAuthorizationHeaderTest(for: .default)
  621. }
  622. func testThatEphemeralConfigurationHeadersAreSentWithRequest() {
  623. // Given, When, Then
  624. executeAuthorizationHeaderTest(for: .ephemeral)
  625. }
  626. #if os(macOS)
  627. func testThatBackgroundConfigurationHeadersAreSentWithRequest() {
  628. // Given, When, Then
  629. executeAuthorizationHeaderTest(for: .background)
  630. }
  631. #endif
  632. private func executeAuthorizationHeaderTest(for type: ConfigurationType) {
  633. // Given
  634. let manager: SessionManager = {
  635. let configuration: URLSessionConfiguration = {
  636. let configuration: URLSessionConfiguration
  637. switch type {
  638. case .default:
  639. configuration = .default
  640. case .ephemeral:
  641. configuration = .ephemeral
  642. case .background:
  643. let identifier = "org.alamofire.test.manager-configuration-tests"
  644. configuration = .background(withIdentifier: identifier)
  645. }
  646. var headers = SessionManager.defaultHTTPHeaders
  647. headers["Authorization"] = "Bearer 123456"
  648. configuration.httpAdditionalHeaders = headers
  649. return configuration
  650. }()
  651. return SessionManager(configuration: configuration)
  652. }()
  653. let expectation = self.expectation(description: "request should complete successfully")
  654. var response: DataResponse<Any>?
  655. // When
  656. manager.request("https://httpbin.org/headers")
  657. .responseJSON { closureResponse in
  658. response = closureResponse
  659. expectation.fulfill()
  660. }
  661. waitForExpectations(timeout: timeout, handler: nil)
  662. // Then
  663. if let response = response {
  664. XCTAssertNotNil(response.request, "request should not be nil")
  665. XCTAssertNotNil(response.response, "response should not be nil")
  666. XCTAssertNotNil(response.data, "data should not be nil")
  667. XCTAssertTrue(response.result.isSuccess, "result should be a success")
  668. if
  669. let response = response.result.value as? [String: Any],
  670. let headers = response["headers"] as? [String: String],
  671. let authorization = headers["Authorization"]
  672. {
  673. XCTAssertEqual(authorization, "Bearer 123456", "authorization header value does not match")
  674. } else {
  675. XCTFail("failed to extract authorization header value")
  676. }
  677. } else {
  678. XCTFail("response should not be nil")
  679. }
  680. }
  681. }