ParameterEncoderTests.swift 28 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893
  1. //
  2. // ParameterEncoderTests.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. import Alamofire
  25. import XCTest
  26. final class JSONParameterEncoderTests: BaseTestCase {
  27. func testThatDataIsProperlyEncodedAndProperContentTypeIsSet() throws {
  28. // Given
  29. let encoder = JSONParameterEncoder()
  30. let request = URLRequest.makeHTTPBinRequest()
  31. // When
  32. let newRequest = try encoder.encode(HTTPBinParameters.default, into: request)
  33. // Then
  34. XCTAssertEqual(newRequest.headers["Content-Type"], "application/json")
  35. XCTAssertEqual(newRequest.httpBody?.asString, "{\"property\":\"property\"}")
  36. }
  37. func testThatDataIsProperlyEncodedButContentTypeIsNotSetIfRequestAlreadyHasAContentType() throws {
  38. // Given
  39. let encoder = JSONParameterEncoder()
  40. var request = URLRequest.makeHTTPBinRequest()
  41. request.headers.update(.contentType("type"))
  42. // When
  43. let newRequest = try encoder.encode(HTTPBinParameters.default, into: request)
  44. // Then
  45. XCTAssertEqual(newRequest.headers["Content-Type"], "type")
  46. XCTAssertEqual(newRequest.httpBody?.asString, "{\"property\":\"property\"}")
  47. }
  48. func testThatJSONEncoderCanBeCustomized() throws {
  49. // Given
  50. let jsonEncoder = JSONEncoder()
  51. jsonEncoder.outputFormatting = .prettyPrinted
  52. let encoder = JSONParameterEncoder(encoder: jsonEncoder)
  53. let request = URLRequest.makeHTTPBinRequest()
  54. // When
  55. let newRequest = try encoder.encode(HTTPBinParameters.default, into: request)
  56. // Then
  57. let expected = """
  58. {
  59. "property" : "property"
  60. }
  61. """
  62. XCTAssertEqual(newRequest.httpBody?.asString, expected)
  63. }
  64. func testThatJSONEncoderDefaultWorks() throws {
  65. // Given
  66. let encoder = JSONParameterEncoder.default
  67. let request = URLRequest.makeHTTPBinRequest()
  68. // When
  69. let encoded = try encoder.encode(HTTPBinParameters.default, into: request)
  70. // Then
  71. let expected = """
  72. {"property":"property"}
  73. """
  74. XCTAssertEqual(encoded.httpBody?.asString, expected)
  75. }
  76. func testThatJSONEncoderPrettyPrintedPrintsPretty() throws {
  77. // Given
  78. let encoder = JSONParameterEncoder.prettyPrinted
  79. let request = URLRequest.makeHTTPBinRequest()
  80. // When
  81. let encoded = try encoder.encode(HTTPBinParameters.default, into: request)
  82. // Then
  83. let expected = """
  84. {
  85. "property" : "property"
  86. }
  87. """
  88. XCTAssertEqual(encoded.httpBody?.asString, expected)
  89. }
  90. }
  91. final class SortedKeysJSONParameterEncoderTests: BaseTestCase {
  92. func testTestJSONEncoderSortedKeysHasSortedKeys() throws {
  93. guard #available(macOS 10.13, iOS 11.0, tvOS 11.0, watchOS 4.0, *) else { return }
  94. // Given
  95. let encoder = JSONParameterEncoder.sortedKeys
  96. let request = URLRequest.makeHTTPBinRequest()
  97. // When
  98. let encoded = try encoder.encode(["z": "z", "a": "a", "p": "p"], into: request)
  99. // Then
  100. let expected = """
  101. {"a":"a","p":"p","z":"z"}
  102. """
  103. XCTAssertEqual(encoded.httpBody?.asString, expected)
  104. }
  105. }
  106. final class URLEncodedFormParameterEncoderTests: BaseTestCase {
  107. func testThatQueryIsBodyEncodedAndProperContentTypeIsSetForPOSTRequest() throws {
  108. // Given
  109. let encoder = URLEncodedFormParameterEncoder()
  110. let request = URLRequest.makeHTTPBinRequest(method: .post)
  111. // When
  112. let newRequest = try encoder.encode(HTTPBinParameters.default, into: request)
  113. // Then
  114. XCTAssertEqual(newRequest.headers["Content-Type"], "application/x-www-form-urlencoded; charset=utf-8")
  115. XCTAssertEqual(newRequest.httpBody?.asString, "property=property")
  116. }
  117. func testThatQueryIsBodyEncodedButContentTypeIsNotSetWhenRequestAlreadyHasContentType() throws {
  118. // Given
  119. let encoder = URLEncodedFormParameterEncoder()
  120. var request = URLRequest.makeHTTPBinRequest(method: .post)
  121. request.headers.update(.contentType("type"))
  122. // When
  123. let newRequest = try encoder.encode(HTTPBinParameters.default, into: request)
  124. // Then
  125. XCTAssertEqual(newRequest.headers["Content-Type"], "type")
  126. XCTAssertEqual(newRequest.httpBody?.asString, "property=property")
  127. }
  128. func testThatEncoderCanBeCustomized() throws {
  129. // Given
  130. let urlEncoder = URLEncodedFormEncoder(boolEncoding: .literal)
  131. let encoder = URLEncodedFormParameterEncoder(encoder: urlEncoder)
  132. let request = URLRequest.makeHTTPBinRequest()
  133. // When
  134. let newRequest = try encoder.encode(["bool": true], into: request)
  135. // Then
  136. let components = URLComponents(url: newRequest.url!, resolvingAgainstBaseURL: false)
  137. XCTAssertEqual(components?.percentEncodedQuery, "bool=true")
  138. }
  139. func testThatQueryIsInURLWhenDestinationIsURLAndMethodIsPOST() throws {
  140. // Given
  141. let encoder = URLEncodedFormParameterEncoder(destination: .queryString)
  142. let request = URLRequest.makeHTTPBinRequest(method: .post)
  143. // When
  144. let newRequest = try encoder.encode(HTTPBinParameters.default, into: request)
  145. // Then
  146. let components = URLComponents(url: newRequest.url!, resolvingAgainstBaseURL: false)
  147. XCTAssertEqual(components?.percentEncodedQuery, "property=property")
  148. }
  149. func testThatQueryIsNilWhenEncodableResultsInAnEmptyString() throws {
  150. // Given
  151. let encoder = URLEncodedFormParameterEncoder(destination: .queryString)
  152. let request = URLRequest.makeHTTPBinRequest()
  153. // When
  154. let newRequest = try encoder.encode([String: String](), into: request)
  155. // Then
  156. let components = URLComponents(url: newRequest.url!, resolvingAgainstBaseURL: false)
  157. XCTAssertNil(components?.percentEncodedQuery)
  158. }
  159. }
  160. final class URLEncodedFormEncoderTests: BaseTestCase {
  161. func testEncoderThrowsErrorWhenAttemptingToEncodeNilInKeyedContainer() {
  162. // Given
  163. let encoder = URLEncodedFormEncoder()
  164. let parameters = FailingOptionalStruct(testedContainer: .keyed)
  165. // When
  166. let result = AFResult<String> { try encoder.encode(parameters) }
  167. // Then
  168. XCTAssertTrue(result.isFailure)
  169. }
  170. func testEncoderThrowsErrorWhenAttemptingToEncodeNilInUnkeyedContainer() {
  171. // Given
  172. let encoder = URLEncodedFormEncoder()
  173. let parameters = FailingOptionalStruct(testedContainer: .unkeyed)
  174. // When
  175. let result = AFResult<String> { try encoder.encode(parameters) }
  176. // Then
  177. XCTAssertTrue(result.isFailure)
  178. }
  179. func testEncoderCanEncodeDictionary() {
  180. // Given
  181. let encoder = URLEncodedFormEncoder()
  182. let parameters = ["a": "a"]
  183. // When
  184. let result = AFResult<String> { try encoder.encode(parameters) }
  185. // Then
  186. XCTAssertEqual(result.value, "a=a")
  187. }
  188. func testEncoderCanEncodeDouble() {
  189. // Given
  190. let encoder = URLEncodedFormEncoder()
  191. let parameters = ["a": 1.0]
  192. // When
  193. let result = AFResult<String> { try encoder.encode(parameters) }
  194. // Then
  195. XCTAssertEqual(result.value, "a=1.0")
  196. }
  197. func testEncoderCanEncodeFloat() {
  198. // Given
  199. let encoder = URLEncodedFormEncoder()
  200. let parameters: [String: Float] = ["a": 1.0]
  201. // When
  202. let result = AFResult<String> { try encoder.encode(parameters) }
  203. // Then
  204. XCTAssertEqual(result.value, "a=1.0")
  205. }
  206. func testEncoderCanEncodeInt8() {
  207. // Given
  208. let encoder = URLEncodedFormEncoder()
  209. let parameters: [String: Int8] = ["a": 1]
  210. // When
  211. let result = AFResult<String> { try encoder.encode(parameters) }
  212. // Then
  213. XCTAssertEqual(result.value, "a=1")
  214. }
  215. func testEncoderCanEncodeInt16() {
  216. // Given
  217. let encoder = URLEncodedFormEncoder()
  218. let parameters: [String: Int16] = ["a": 1]
  219. // When
  220. let result = AFResult<String> { try encoder.encode(parameters) }
  221. // Then
  222. XCTAssertEqual(result.value, "a=1")
  223. }
  224. func testEncoderCanEncodeInt32() {
  225. // Given
  226. let encoder = URLEncodedFormEncoder()
  227. let parameters: [String: Int32] = ["a": 1]
  228. // When
  229. let result = AFResult<String> { try encoder.encode(parameters) }
  230. // Then
  231. XCTAssertEqual(result.value, "a=1")
  232. }
  233. func testEncoderCanEncodeInt64() {
  234. // Given
  235. let encoder = URLEncodedFormEncoder()
  236. let parameters: [String: Int64] = ["a": 1]
  237. // When
  238. let result = AFResult<String> { try encoder.encode(parameters) }
  239. // Then
  240. XCTAssertEqual(result.value, "a=1")
  241. }
  242. func testEncoderCanEncodeUInt() {
  243. // Given
  244. let encoder = URLEncodedFormEncoder()
  245. let parameters: [String: UInt] = ["a": 1]
  246. // When
  247. let result = AFResult<String> { try encoder.encode(parameters) }
  248. // Then
  249. XCTAssertEqual(result.value, "a=1")
  250. }
  251. func testEncoderCanEncodeUInt8() {
  252. // Given
  253. let encoder = URLEncodedFormEncoder()
  254. let parameters: [String: UInt8] = ["a": 1]
  255. // When
  256. let result = AFResult<String> { try encoder.encode(parameters) }
  257. // Then
  258. XCTAssertEqual(result.value, "a=1")
  259. }
  260. func testEncoderCanEncodeUInt16() {
  261. // Given
  262. let encoder = URLEncodedFormEncoder()
  263. let parameters: [String: UInt16] = ["a": 1]
  264. // When
  265. let result = AFResult<String> { try encoder.encode(parameters) }
  266. // Then
  267. XCTAssertEqual(result.value, "a=1")
  268. }
  269. func testEncoderCanEncodeUInt32() {
  270. // Given
  271. let encoder = URLEncodedFormEncoder()
  272. let parameters: [String: UInt32] = ["a": 1]
  273. // When
  274. let result = AFResult<String> { try encoder.encode(parameters) }
  275. // Then
  276. XCTAssertEqual(result.value, "a=1")
  277. }
  278. func testEncoderCanEncodeUInt64() {
  279. // Given
  280. let encoder = URLEncodedFormEncoder()
  281. let parameters: [String: UInt64] = ["a": 1]
  282. // When
  283. let result = AFResult<String> { try encoder.encode(parameters) }
  284. // Then
  285. XCTAssertEqual(result.value, "a=1")
  286. }
  287. func testThatNestedDictionariesHaveBracketedKeys() {
  288. // Given
  289. let encoder = URLEncodedFormEncoder()
  290. let parameters = ["a": ["b": "b"]]
  291. // When
  292. let result = AFResult<String> { try encoder.encode(parameters) }
  293. // Then
  294. XCTAssertEqual(result.value, "a%5Bb%5D=b")
  295. }
  296. func testThatEncodableStructCanBeEncoded() {
  297. // Given
  298. let encoder = URLEncodedFormEncoder()
  299. let parameters = EncodableStruct()
  300. // When
  301. let result = AFResult<String> { try encoder.encode(parameters) }
  302. // Then
  303. let expected = "four%5B%5D=1&four%5B%5D=2&four%5B%5D=3&three=1&one=one&two=2&five%5Ba%5D=a&six%5Ba%5D%5Bb%5D=b&seven%5Ba%5D=a"
  304. XCTAssertQueryEqual(result.value, expected)
  305. }
  306. func testThatManuallyEncodableStructCanBeEncoded() {
  307. // Given
  308. let encoder = URLEncodedFormEncoder()
  309. let parameters = ManuallyEncodableStruct()
  310. // When
  311. let result = AFResult<String> { try encoder.encode(parameters) }
  312. // Then
  313. let expected = "root%5B%5D%5B%5D=1&root%5B%5D%5B%5D=2&root%5B%5D%5B%5D=3&root%5B%5D%5Ba%5D%5Bstring%5D=string&root%5B%5D%5B%5D%5B%5D=1&root%5B%5D%5B%5D%5B%5D=2&root%5B%5D%5B%5D%5B%5D=3"
  314. XCTAssertQueryEqual(result.value, expected)
  315. }
  316. func testThatEncodableClassWithNoInheritanceCanBeEncoded() {
  317. // Given
  318. let encoder = URLEncodedFormEncoder()
  319. let parameters = EncodableSuperclass()
  320. // When
  321. let result = AFResult<String> { try encoder.encode(parameters) }
  322. // Then
  323. XCTAssertQueryEqual(result.value, "two=2&one=one&three=1")
  324. }
  325. func testThatEncodableSubclassCanBeEncoded() {
  326. // Given
  327. let encoder = URLEncodedFormEncoder()
  328. let parameters = EncodableSubclass()
  329. // When
  330. let result = AFResult<String> { try encoder.encode(parameters) }
  331. // Then
  332. let expected = "four%5B%5D=1&four%5B%5D=2&four%5B%5D=3&two=2&five%5Ba%5D=a&five%5Bb%5D=b&three=1&one=one"
  333. XCTAssertQueryEqual(result.value, expected)
  334. }
  335. func testThatManuallyEncodableSubclassCanBeEncoded() {
  336. // Given
  337. let encoder = URLEncodedFormEncoder()
  338. let parameters = ManuallyEncodableSubclass()
  339. // When
  340. let result = AFResult<String> { try encoder.encode(parameters) }
  341. // Then
  342. let expected = "five%5Ba%5D=a&five%5Bb%5D=b&four%5Bfour%5D=one&four%5Bfive%5D=2"
  343. XCTAssertQueryEqual(result.value, expected)
  344. }
  345. func testThatARootArrayCannotBeEncoded() {
  346. // Given
  347. let encoder = URLEncodedFormEncoder()
  348. let parameters = [1]
  349. // When
  350. let result = AFResult<String> { try encoder.encode(parameters) }
  351. // Then
  352. XCTAssertFalse(result.isSuccess)
  353. }
  354. func testThatARootValueCannotBeEncoded() {
  355. // Given
  356. let encoder = URLEncodedFormEncoder()
  357. let parameters = "string"
  358. // When
  359. let result = AFResult<String> { try encoder.encode(parameters) }
  360. // Then
  361. XCTAssertFalse(result.isSuccess)
  362. }
  363. func testThatOptionalValuesCannotBeEncoded() {
  364. // Givenp
  365. let encoder = URLEncodedFormEncoder()
  366. let parameters: [String: String?] = ["string": nil]
  367. // When
  368. let result = AFResult<String> { try encoder.encode(parameters) }
  369. // Then
  370. XCTAssertFalse(result.isSuccess)
  371. }
  372. func testThatBoolsCanBeLiteralEncoded() {
  373. // Given
  374. let encoder = URLEncodedFormEncoder(boolEncoding: .literal)
  375. let parameters = ["bool": true]
  376. // When
  377. let result = AFResult<String> { try encoder.encode(parameters) }
  378. // Then
  379. XCTAssertEqual(result.value, "bool=true")
  380. }
  381. func testThatDatesCanBeEncoded() {
  382. // Given
  383. let encoder = URLEncodedFormEncoder(dateEncoding: .deferredToDate)
  384. let parameters = ["date": Date(timeIntervalSinceReferenceDate: 123.456)]
  385. // When
  386. let result = AFResult<String> { try encoder.encode(parameters) }
  387. // Then
  388. XCTAssertEqual(result.value, "date=123.456")
  389. }
  390. func testThatDatesCanBeEncodedAsSecondsSince1970() {
  391. // Given
  392. let encoder = URLEncodedFormEncoder(dateEncoding: .secondsSince1970)
  393. let parameters = ["date": Date(timeIntervalSinceReferenceDate: 123.456)]
  394. // When
  395. let result = AFResult<String> { try encoder.encode(parameters) }
  396. // Then
  397. XCTAssertEqual(result.value, "date=978307323.456")
  398. }
  399. func testThatDatesCanBeEncodedAsMillisecondsSince1970() {
  400. // Given
  401. let encoder = URLEncodedFormEncoder(dateEncoding: .millisecondsSince1970)
  402. let parameters = ["date": Date(timeIntervalSinceReferenceDate: 123.456)]
  403. // When
  404. let result = AFResult<String> { try encoder.encode(parameters) }
  405. // Then
  406. XCTAssertEqual(result.value, "date=978307323456.0")
  407. }
  408. func testThatDatesCanBeEncodedAsISO8601Formatted() {
  409. // Given
  410. let encoder = URLEncodedFormEncoder(dateEncoding: .iso8601)
  411. let parameters = ["date": Date(timeIntervalSinceReferenceDate: 123.456)]
  412. // When
  413. let result = AFResult<String> { try encoder.encode(parameters) }
  414. // Then
  415. XCTAssertEqual(result.value, "date=2001-01-01T00%3A02%3A03Z")
  416. }
  417. func testThatDatesCanBeEncodedAsFormatted() {
  418. // Given
  419. let dateFormatter = DateFormatter()
  420. dateFormatter.dateFormat = "yyyy-MM-dd HH:mm:ss.SSSS"
  421. dateFormatter.timeZone = TimeZone(secondsFromGMT: 0)
  422. let encoder = URLEncodedFormEncoder(dateEncoding: .formatted(dateFormatter))
  423. let parameters = ["date": Date(timeIntervalSinceReferenceDate: 123.456)]
  424. // When
  425. let result = AFResult<String> { try encoder.encode(parameters) }
  426. // Then
  427. XCTAssertEqual(result.value, "date=2001-01-01%2000%3A02%3A03.4560")
  428. }
  429. func testThatDatesCanBeEncodedAsCustomFormatted() {
  430. // Given
  431. let encoder = URLEncodedFormEncoder(dateEncoding: .custom({ "\($0.timeIntervalSinceReferenceDate)" }))
  432. let parameters = ["date": Date(timeIntervalSinceReferenceDate: 123.456)]
  433. // When
  434. let result = AFResult<String> { try encoder.encode(parameters) }
  435. // Then
  436. XCTAssertEqual(result.value, "date=123.456")
  437. }
  438. func testEncoderThrowsErrorWhenCustomDateEncodingFails() {
  439. // Given
  440. struct DateEncodingError: Error {}
  441. let encoder = URLEncodedFormEncoder(dateEncoding: .custom({ _ in throw DateEncodingError() }))
  442. let parameters = ["date": Date(timeIntervalSinceReferenceDate: 123.456)]
  443. // When
  444. let result = AFResult<String> { try encoder.encode(parameters) }
  445. // Then
  446. XCTAssertTrue(result.isFailure)
  447. XCTAssertTrue(result.error is DateEncodingError)
  448. }
  449. func testThatArraysCanBeEncodedWithoutBrackets() {
  450. // Given
  451. let encoder = URLEncodedFormEncoder(arrayEncoding: .noBrackets)
  452. let parameters = ["array": [1, 2]]
  453. // When
  454. let result = AFResult<String> { try encoder.encode(parameters) }
  455. // Then
  456. XCTAssertQueryEqual(result.value, "array=1&array=2")
  457. }
  458. func testThatSpacesCanBeEncodedAsPluses() {
  459. // Given
  460. let encoder = URLEncodedFormEncoder(spaceEncoding: .plusReplaced)
  461. let parameters = ["spaces": "replace with spaces"]
  462. // When
  463. let result = AFResult<String> { try encoder.encode(parameters) }
  464. // Then
  465. XCTAssertEqual(result.value, "spaces=replace+with+spaces")
  466. }
  467. func testThatEscapedCharactersCanBeCustomized() {
  468. // Given
  469. var allowed = CharacterSet.afURLQueryAllowed
  470. allowed.remove(charactersIn: "?/")
  471. let encoder = URLEncodedFormEncoder(allowedCharacters: allowed)
  472. let parameters = ["allowed": "?/"]
  473. // When
  474. let result = AFResult<String> { try encoder.encode(parameters) }
  475. // Then
  476. XCTAssertEqual(result.value, "allowed=%3F%2F")
  477. }
  478. func testThatUnreservedCharactersAreNotPercentEscaped() {
  479. // Given
  480. let encoder = URLEncodedFormEncoder()
  481. let parameters = ["lowercase": "abcdefghijklmnopqrstuvwxyz",
  482. "uppercase": "ABCDEFGHIJKLMNOPQRSTUVWXYZ",
  483. "numbers": "0123456789"]
  484. // When
  485. let result = AFResult<String> { try encoder.encode(parameters) }
  486. // Then
  487. XCTAssertQueryEqual(result.value,
  488. "uppercase=ABCDEFGHIJKLMNOPQRSTUVWXYZ&numbers=0123456789&lowercase=abcdefghijklmnopqrstuvwxyz")
  489. }
  490. func testThatReseredCharactersArePercentEscaped() {
  491. // Given
  492. let encoder = URLEncodedFormEncoder()
  493. let generalDelimiters = ":#[]@"
  494. let subDelimiters = "!$&'()*+,;="
  495. let parameters = ["reserved": "\(generalDelimiters)\(subDelimiters)"]
  496. // When
  497. let result = AFResult<String> { try encoder.encode(parameters) }
  498. // Then
  499. XCTAssertEqual(result.value, "reserved=%3A%23%5B%5D%40%21%24%26%27%28%29%2A%2B%2C%3B%3D")
  500. }
  501. func testThatIllegalASCIICharactersArePercentEscaped() {
  502. // Given
  503. let encoder = URLEncodedFormEncoder()
  504. let parameters = ["illegal": " \"#%<>[]\\^`{}|"]
  505. // When
  506. let result = AFResult<String> { try encoder.encode(parameters) }
  507. // Then
  508. XCTAssertEqual(result.value, "illegal=%20%22%23%25%3C%3E%5B%5D%5C%5E%60%7B%7D%7C")
  509. }
  510. func testThatAmpersandsInKeysAndValuesArePercentEscaped() {
  511. // Given
  512. let encoder = URLEncodedFormEncoder()
  513. let parameters = ["foo&bar": "baz&qux", "foobar": "bazqux"]
  514. // When
  515. let result = AFResult<String> { try encoder.encode(parameters) }
  516. // Then
  517. XCTAssertQueryEqual(result.value, "foobar=bazqux&foo%26bar=baz%26qux")
  518. }
  519. func testThatQuestionMarksInKeysAndValuesAreNotPercentEscaped() {
  520. // Given
  521. let encoder = URLEncodedFormEncoder()
  522. let parameters = ["?foo?": "?bar?"]
  523. // When
  524. let result = AFResult<String> { try encoder.encode(parameters) }
  525. // Then
  526. XCTAssertEqual(result.value, "?foo?=?bar?")
  527. }
  528. func testThatSlashesInKeysAndValuesAreNotPercentEscaped() {
  529. // Given
  530. let encoder = URLEncodedFormEncoder()
  531. let parameters = ["foo": "/bar/baz/qux"]
  532. // When
  533. let result = AFResult<String> { try encoder.encode(parameters) }
  534. // Then
  535. XCTAssertEqual(result.value, "foo=/bar/baz/qux")
  536. }
  537. func testThatSpacesInKeysAndValuesArePercentEscaped() {
  538. // Given
  539. let encoder = URLEncodedFormEncoder()
  540. let parameters = [" foo ": " bar "]
  541. // When
  542. let result = AFResult<String> { try encoder.encode(parameters) }
  543. // Then
  544. XCTAssertEqual(result.value, "%20foo%20=%20bar%20")
  545. }
  546. func testThatPlusesInKeysAndValuesArePercentEscaped() {
  547. // Given
  548. let encoder = URLEncodedFormEncoder()
  549. let parameters = ["+foo+": "+bar+"]
  550. // When
  551. let result = AFResult<String> { try encoder.encode(parameters) }
  552. // Then
  553. XCTAssertEqual(result.value, "%2Bfoo%2B=%2Bbar%2B")
  554. }
  555. func testThatPercentsInKeysAndValuesArePercentEscaped() {
  556. // Given
  557. let encoder = URLEncodedFormEncoder()
  558. let parameters = ["percent%": "%25"]
  559. // When
  560. let result = AFResult<String> { try encoder.encode(parameters) }
  561. // Then
  562. XCTAssertEqual(result.value, "percent%25=%2525")
  563. }
  564. func testThatNonLatinCharactersArePercentEscaped() {
  565. // Given
  566. let encoder = URLEncodedFormEncoder()
  567. let parameters = [
  568. "french": "français",
  569. "japanese": "日本語",
  570. "arabic": "العربية",
  571. "emoji": "😃"
  572. ]
  573. // When
  574. let result = AFResult<String> { try encoder.encode(parameters) }
  575. // Then
  576. let expectedParameterValues = [
  577. "arabic=%D8%A7%D9%84%D8%B9%D8%B1%D8%A8%D9%8A%D8%A9",
  578. "japanese=%E6%97%A5%E6%9C%AC%E8%AA%9E",
  579. "french=fran%C3%A7ais",
  580. "emoji=%F0%9F%98%83"
  581. ].joined(separator: "&")
  582. XCTAssertQueryEqual(result.value, expectedParameterValues)
  583. }
  584. func testStringWithThousandsOfChineseCharactersIsPercentEscaped() {
  585. // Given
  586. let encoder = URLEncodedFormEncoder()
  587. let repeatedCount = 2_000
  588. let parameters = ["chinese": String(repeating: "一二三四五六七八九十", count: repeatedCount)]
  589. // When
  590. let result = AFResult<String> { try encoder.encode(parameters) }
  591. // Then
  592. let escaped = String(repeating: "%E4%B8%80%E4%BA%8C%E4%B8%89%E5%9B%9B%E4%BA%94%E5%85%AD%E4%B8%83%E5%85%AB%E4%B9%9D%E5%8D%81",
  593. count: repeatedCount)
  594. let expected = "chinese=\(escaped)"
  595. XCTAssertEqual(result.value, expected)
  596. }
  597. }
  598. private struct EncodableStruct: Encodable {
  599. let one = "one"
  600. let two = 2
  601. let three = true
  602. let four = [1, 2, 3]
  603. let five = ["a": "a"]
  604. let six = ["a": ["b": "b"]]
  605. let seven = NestedEncodableStruct()
  606. }
  607. private struct NestedEncodableStruct: Encodable {
  608. let a = "a"
  609. }
  610. private class EncodableSuperclass: Encodable {
  611. let one = "one"
  612. let two = 2
  613. let three = true
  614. }
  615. private final class EncodableSubclass: EncodableSuperclass {
  616. let four = [1, 2, 3]
  617. let five = ["a": "a", "b": "b"]
  618. private enum CodingKeys: String, CodingKey {
  619. case four, five
  620. }
  621. override func encode(to encoder: Encoder) throws {
  622. try super.encode(to: encoder)
  623. var container = encoder.container(keyedBy: CodingKeys.self)
  624. try container.encode(four, forKey: .four)
  625. try container.encode(five, forKey: .five)
  626. }
  627. }
  628. private final class ManuallyEncodableSubclass: EncodableSuperclass {
  629. let four = [1, 2, 3]
  630. let five = ["a": "a", "b": "b"]
  631. private enum CodingKeys: String, CodingKey {
  632. case four, five
  633. }
  634. override func encode(to encoder: Encoder) throws {
  635. var keyedContainer = encoder.container(keyedBy: CodingKeys.self)
  636. try keyedContainer.encode(four, forKey: .four)
  637. try keyedContainer.encode(five, forKey: .five)
  638. let superEncoder = keyedContainer.superEncoder()
  639. var superContainer = superEncoder.container(keyedBy: CodingKeys.self)
  640. try superContainer.encode(one, forKey: .four)
  641. let keyedSuperEncoder = keyedContainer.superEncoder(forKey: .four)
  642. var superKeyedContainer = keyedSuperEncoder.container(keyedBy: CodingKeys.self)
  643. try superKeyedContainer.encode(two, forKey: .five)
  644. var unkeyedContainer = keyedContainer.nestedUnkeyedContainer(forKey: .four)
  645. let unkeyedSuperEncoder = unkeyedContainer.superEncoder()
  646. var keyedUnkeyedSuperContainer = unkeyedSuperEncoder.container(keyedBy: CodingKeys.self)
  647. try keyedUnkeyedSuperContainer.encode(one, forKey: .four)
  648. }
  649. }
  650. private struct ManuallyEncodableStruct: Encodable {
  651. let a = ["string": "string"]
  652. let b = [1, 2, 3]
  653. private enum RootKey: String, CodingKey {
  654. case root
  655. }
  656. private enum TypeKeys: String, CodingKey {
  657. case a, b
  658. }
  659. func encode(to encoder: Encoder) throws {
  660. var container = encoder.container(keyedBy: RootKey.self)
  661. var nestedKeyedContainer = container.nestedContainer(keyedBy: TypeKeys.self, forKey: .root)
  662. try nestedKeyedContainer.encode(a, forKey: .a)
  663. var nestedUnkeyedContainer = container.nestedUnkeyedContainer(forKey: .root)
  664. try nestedUnkeyedContainer.encode(b)
  665. var nestedUnkeyedKeyedContainer = nestedUnkeyedContainer.nestedContainer(keyedBy: TypeKeys.self)
  666. try nestedUnkeyedKeyedContainer.encode(a, forKey: .a)
  667. var nestedUnkeyedUnkeyedContainer = nestedUnkeyedContainer.nestedUnkeyedContainer()
  668. try nestedUnkeyedUnkeyedContainer.encode(b)
  669. }
  670. }
  671. private struct FailingOptionalStruct: Encodable {
  672. enum TestedContainer {
  673. case keyed, unkeyed
  674. }
  675. enum CodingKeys: String, CodingKey { case a }
  676. let testedContainer: TestedContainer
  677. func encode(to encoder: Encoder) throws {
  678. var container = encoder.container(keyedBy: CodingKeys.self)
  679. switch testedContainer {
  680. case .keyed:
  681. var nested = container.nestedContainer(keyedBy: CodingKeys.self, forKey: .a)
  682. try nested.encodeNil(forKey: .a)
  683. case .unkeyed:
  684. var nested = container.nestedUnkeyedContainer(forKey: .a)
  685. try nested.encodeNil()
  686. }
  687. }
  688. }
  689. private func XCTAssertQueryEqual(_ query1: String?, _ query2: String?) {
  690. let items1 = query1?.split(separator: "&").sorted()
  691. let items2 = query2?.split(separator: "&").sorted()
  692. XCTAssertEqual(items1, items2)
  693. }