ParameterEncoderTests.swift 31 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992
  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 = Result<String, Error> { 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 = Result<String, Error> { 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 = Result<String, Error> { 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 = Result<String, Error> { 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 = Result<String, Error> { 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 = Result<String, Error> { 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 = Result<String, Error> { 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 = Result<String, Error> { 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 = Result<String, Error> { 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 = Result<String, Error> { 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 = Result<String, Error> { 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 = Result<String, Error> { 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 = Result<String, Error> { 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 = Result<String, Error> { 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 = Result<String, Error> { 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 = Result<String, Error> { 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 = Result<String, Error> { 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 = Result<String, Error> { 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 = Result<String, Error> { 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 = Result<String, Error> { 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 = Result<String, Error> { 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 = Result<String, Error> { 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 = Result<String, Error> { try encoder.encode(parameters) }
  369. // Then
  370. XCTAssertFalse(result.isSuccess)
  371. }
  372. func testThatArraysCanBeEncodedWithoutBrackets() {
  373. // Given
  374. let encoder = URLEncodedFormEncoder(arrayEncoding: .noBrackets)
  375. let parameters = ["array": [1, 2]]
  376. // When
  377. let result = Result<String, Error> { try encoder.encode(parameters) }
  378. // Then
  379. XCTAssertQueryEqual(result.value, "array=1&array=2")
  380. }
  381. func testThatBoolsCanBeLiteralEncoded() {
  382. // Given
  383. let encoder = URLEncodedFormEncoder(boolEncoding: .literal)
  384. let parameters = ["bool": true]
  385. // When
  386. let result = Result<String, Error> { try encoder.encode(parameters) }
  387. // Then
  388. XCTAssertEqual(result.value, "bool=true")
  389. }
  390. func testThatDataCanBeEncoded() {
  391. // Given
  392. let encoder = URLEncodedFormEncoder()
  393. let parameters = ["data": Data("data".utf8)]
  394. // When
  395. let result = Result<String, Error> { try encoder.encode(parameters) }
  396. // Then
  397. XCTAssertEqual(result.value, "data=ZGF0YQ%3D%3D")
  398. }
  399. func testThatCustomDataEncodingFailsWhenErrorIsThrown() {
  400. // Given
  401. struct DataEncodingError: Error {}
  402. let encoder = URLEncodedFormEncoder(dataEncoding: .custom({ _ in throw DataEncodingError() }))
  403. let parameters = ["data": Data("data".utf8)]
  404. // When
  405. let result = Result<String, Error> { try encoder.encode(parameters) }
  406. // Then
  407. XCTAssertTrue(result.isFailure)
  408. XCTAssertTrue(result.error is DataEncodingError)
  409. }
  410. func testThatDatesCanBeEncoded() {
  411. // Given
  412. let encoder = URLEncodedFormEncoder(dateEncoding: .deferredToDate)
  413. let parameters = ["date": Date(timeIntervalSinceReferenceDate: 123.456)]
  414. // When
  415. let result = Result<String, Error> { try encoder.encode(parameters) }
  416. // Then
  417. XCTAssertEqual(result.value, "date=123.456")
  418. }
  419. func testThatDatesCanBeEncodedAsSecondsSince1970() {
  420. // Given
  421. let encoder = URLEncodedFormEncoder(dateEncoding: .secondsSince1970)
  422. let parameters = ["date": Date(timeIntervalSinceReferenceDate: 123.456)]
  423. // When
  424. let result = Result<String, Error> { try encoder.encode(parameters) }
  425. // Then
  426. XCTAssertEqual(result.value, "date=978307323.456")
  427. }
  428. func testThatDatesCanBeEncodedAsMillisecondsSince1970() {
  429. // Given
  430. let encoder = URLEncodedFormEncoder(dateEncoding: .millisecondsSince1970)
  431. let parameters = ["date": Date(timeIntervalSinceReferenceDate: 123.456)]
  432. // When
  433. let result = Result<String, Error> { try encoder.encode(parameters) }
  434. // Then
  435. XCTAssertEqual(result.value, "date=978307323456.0")
  436. }
  437. func testThatDatesCanBeEncodedAsISO8601Formatted() {
  438. // Given
  439. let encoder = URLEncodedFormEncoder(dateEncoding: .iso8601)
  440. let parameters = ["date": Date(timeIntervalSinceReferenceDate: 123.456)]
  441. // When
  442. let result = Result<String, Error> { try encoder.encode(parameters) }
  443. // Then
  444. XCTAssertEqual(result.value, "date=2001-01-01T00%3A02%3A03Z")
  445. }
  446. func testThatDatesCanBeEncodedAsFormatted() {
  447. // Given
  448. let dateFormatter = DateFormatter()
  449. dateFormatter.dateFormat = "yyyy-MM-dd HH:mm:ss.SSSS"
  450. dateFormatter.timeZone = TimeZone(secondsFromGMT: 0)
  451. let encoder = URLEncodedFormEncoder(dateEncoding: .formatted(dateFormatter))
  452. let parameters = ["date": Date(timeIntervalSinceReferenceDate: 123.456)]
  453. // When
  454. let result = Result<String, Error> { try encoder.encode(parameters) }
  455. // Then
  456. XCTAssertEqual(result.value, "date=2001-01-01%2000%3A02%3A03.4560")
  457. }
  458. func testThatDatesCanBeEncodedAsCustomFormatted() {
  459. // Given
  460. let encoder = URLEncodedFormEncoder(dateEncoding: .custom({ "\($0.timeIntervalSinceReferenceDate)" }))
  461. let parameters = ["date": Date(timeIntervalSinceReferenceDate: 123.456)]
  462. // When
  463. let result = Result<String, Error> { try encoder.encode(parameters) }
  464. // Then
  465. XCTAssertEqual(result.value, "date=123.456")
  466. }
  467. func testEncoderThrowsErrorWhenCustomDateEncodingFails() {
  468. // Given
  469. struct DateEncodingError: Error {}
  470. let encoder = URLEncodedFormEncoder(dateEncoding: .custom({ _ in throw DateEncodingError() }))
  471. let parameters = ["date": Date(timeIntervalSinceReferenceDate: 123.456)]
  472. // When
  473. let result = Result<String, Error> { try encoder.encode(parameters) }
  474. // Then
  475. XCTAssertTrue(result.isFailure)
  476. XCTAssertTrue(result.error is DateEncodingError)
  477. }
  478. func testThatKeysCanBeEncodedIntoSnakeCase() {
  479. // Given
  480. let encoder = URLEncodedFormEncoder(keyEncoding: .convertToSnakeCase)
  481. let paramters = ["oneTwoThree": "oneTwoThree"]
  482. // When
  483. let result = Result<String, Error> { try encoder.encode(paramters) }
  484. // Then
  485. XCTAssertEqual(result.value, "one_two_three=oneTwoThree")
  486. }
  487. func testThatKeysCanBeEncodedIntoKebabCase() {
  488. // Given
  489. let encoder = URLEncodedFormEncoder(keyEncoding: .convertToKebabCase)
  490. let paramters = ["oneTwoThree": "oneTwoThree"]
  491. // When
  492. let result = Result<String, Error> { try encoder.encode(paramters) }
  493. // Then
  494. XCTAssertEqual(result.value, "one-two-three=oneTwoThree")
  495. }
  496. func testThatKeysCanBeEncodedIntoACapitalizedString() {
  497. // Given
  498. let encoder = URLEncodedFormEncoder(keyEncoding: .capitalized)
  499. let paramters = ["oneTwoThree": "oneTwoThree"]
  500. // When
  501. let result = Result<String, Error> { try encoder.encode(paramters) }
  502. // Then
  503. XCTAssertEqual(result.value, "OneTwoThree=oneTwoThree")
  504. }
  505. func testThatKeysCanBeEncodedIntoALowercasedString() {
  506. // Given
  507. let encoder = URLEncodedFormEncoder(keyEncoding: .lowercased)
  508. let paramters = ["oneTwoThree": "oneTwoThree"]
  509. // When
  510. let result = Result<String, Error> { try encoder.encode(paramters) }
  511. // Then
  512. XCTAssertEqual(result.value, "onetwothree=oneTwoThree")
  513. }
  514. func testThatKeysCanBeEncodedIntoAnUppercasedString() {
  515. // Given
  516. let encoder = URLEncodedFormEncoder(keyEncoding: .uppercased)
  517. let paramters = ["oneTwoThree": "oneTwoThree"]
  518. // When
  519. let result = Result<String, Error> { try encoder.encode(paramters) }
  520. // Then
  521. XCTAssertEqual(result.value, "ONETWOTHREE=oneTwoThree")
  522. }
  523. func testThatKeysCanBeCustomEncoded() {
  524. // Given
  525. let encoder = URLEncodedFormEncoder(keyEncoding: .custom({ _ in "A" }))
  526. let paramters = ["oneTwoThree": "oneTwoThree"]
  527. // When
  528. let result = Result<String, Error> { try encoder.encode(paramters) }
  529. // Then
  530. XCTAssertEqual(result.value, "A=oneTwoThree")
  531. }
  532. func testThatSpacesCanBeEncodedAsPluses() {
  533. // Given
  534. let encoder = URLEncodedFormEncoder(spaceEncoding: .plusReplaced)
  535. let parameters = ["spaces": "replace with spaces"]
  536. // When
  537. let result = Result<String, Error> { try encoder.encode(parameters) }
  538. // Then
  539. XCTAssertEqual(result.value, "spaces=replace+with+spaces")
  540. }
  541. func testThatEscapedCharactersCanBeCustomized() {
  542. // Given
  543. var allowed = CharacterSet.afURLQueryAllowed
  544. allowed.remove(charactersIn: "?/")
  545. let encoder = URLEncodedFormEncoder(allowedCharacters: allowed)
  546. let parameters = ["allowed": "?/"]
  547. // When
  548. let result = Result<String, Error> { try encoder.encode(parameters) }
  549. // Then
  550. XCTAssertEqual(result.value, "allowed=%3F%2F")
  551. }
  552. func testThatUnreservedCharactersAreNotPercentEscaped() {
  553. // Given
  554. let encoder = URLEncodedFormEncoder()
  555. let parameters = ["lowercase": "abcdefghijklmnopqrstuvwxyz",
  556. "uppercase": "ABCDEFGHIJKLMNOPQRSTUVWXYZ",
  557. "numbers": "0123456789"]
  558. // When
  559. let result = Result<String, Error> { try encoder.encode(parameters) }
  560. // Then
  561. XCTAssertQueryEqual(result.value,
  562. "uppercase=ABCDEFGHIJKLMNOPQRSTUVWXYZ&numbers=0123456789&lowercase=abcdefghijklmnopqrstuvwxyz")
  563. }
  564. func testThatReseredCharactersArePercentEscaped() {
  565. // Given
  566. let encoder = URLEncodedFormEncoder()
  567. let generalDelimiters = ":#[]@"
  568. let subDelimiters = "!$&'()*+,;="
  569. let parameters = ["reserved": "\(generalDelimiters)\(subDelimiters)"]
  570. // When
  571. let result = Result<String, Error> { try encoder.encode(parameters) }
  572. // Then
  573. XCTAssertEqual(result.value, "reserved=%3A%23%5B%5D%40%21%24%26%27%28%29%2A%2B%2C%3B%3D")
  574. }
  575. func testThatIllegalASCIICharactersArePercentEscaped() {
  576. // Given
  577. let encoder = URLEncodedFormEncoder()
  578. let parameters = ["illegal": " \"#%<>[]\\^`{}|"]
  579. // When
  580. let result = Result<String, Error> { try encoder.encode(parameters) }
  581. // Then
  582. XCTAssertEqual(result.value, "illegal=%20%22%23%25%3C%3E%5B%5D%5C%5E%60%7B%7D%7C")
  583. }
  584. func testThatAmpersandsInKeysAndValuesArePercentEscaped() {
  585. // Given
  586. let encoder = URLEncodedFormEncoder()
  587. let parameters = ["foo&bar": "baz&qux", "foobar": "bazqux"]
  588. // When
  589. let result = Result<String, Error> { try encoder.encode(parameters) }
  590. // Then
  591. XCTAssertQueryEqual(result.value, "foobar=bazqux&foo%26bar=baz%26qux")
  592. }
  593. func testThatQuestionMarksInKeysAndValuesAreNotPercentEscaped() {
  594. // Given
  595. let encoder = URLEncodedFormEncoder()
  596. let parameters = ["?foo?": "?bar?"]
  597. // When
  598. let result = Result<String, Error> { try encoder.encode(parameters) }
  599. // Then
  600. XCTAssertEqual(result.value, "?foo?=?bar?")
  601. }
  602. func testThatSlashesInKeysAndValuesAreNotPercentEscaped() {
  603. // Given
  604. let encoder = URLEncodedFormEncoder()
  605. let parameters = ["foo": "/bar/baz/qux"]
  606. // When
  607. let result = Result<String, Error> { try encoder.encode(parameters) }
  608. // Then
  609. XCTAssertEqual(result.value, "foo=/bar/baz/qux")
  610. }
  611. func testThatSpacesInKeysAndValuesArePercentEscaped() {
  612. // Given
  613. let encoder = URLEncodedFormEncoder()
  614. let parameters = [" foo ": " bar "]
  615. // When
  616. let result = Result<String, Error> { try encoder.encode(parameters) }
  617. // Then
  618. XCTAssertEqual(result.value, "%20foo%20=%20bar%20")
  619. }
  620. func testThatPlusesInKeysAndValuesArePercentEscaped() {
  621. // Given
  622. let encoder = URLEncodedFormEncoder()
  623. let parameters = ["+foo+": "+bar+"]
  624. // When
  625. let result = Result<String, Error> { try encoder.encode(parameters) }
  626. // Then
  627. XCTAssertEqual(result.value, "%2Bfoo%2B=%2Bbar%2B")
  628. }
  629. func testThatPercentsInKeysAndValuesArePercentEscaped() {
  630. // Given
  631. let encoder = URLEncodedFormEncoder()
  632. let parameters = ["percent%": "%25"]
  633. // When
  634. let result = Result<String, Error> { try encoder.encode(parameters) }
  635. // Then
  636. XCTAssertEqual(result.value, "percent%25=%2525")
  637. }
  638. func testThatNonLatinCharactersArePercentEscaped() {
  639. // Given
  640. let encoder = URLEncodedFormEncoder()
  641. let parameters = [
  642. "french": "français",
  643. "japanese": "日本語",
  644. "arabic": "العربية",
  645. "emoji": "😃"
  646. ]
  647. // When
  648. let result = Result<String, Error> { try encoder.encode(parameters) }
  649. // Then
  650. let expectedParameterValues = [
  651. "arabic=%D8%A7%D9%84%D8%B9%D8%B1%D8%A8%D9%8A%D8%A9",
  652. "japanese=%E6%97%A5%E6%9C%AC%E8%AA%9E",
  653. "french=fran%C3%A7ais",
  654. "emoji=%F0%9F%98%83"
  655. ].joined(separator: "&")
  656. XCTAssertQueryEqual(result.value, expectedParameterValues)
  657. }
  658. func testStringWithThousandsOfChineseCharactersIsPercentEscaped() {
  659. // Given
  660. let encoder = URLEncodedFormEncoder()
  661. let repeatedCount = 2_000
  662. let parameters = ["chinese": String(repeating: "一二三四五六七八九十", count: repeatedCount)]
  663. // When
  664. let result = Result<String, Error> { try encoder.encode(parameters) }
  665. // Then
  666. 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",
  667. count: repeatedCount)
  668. let expected = "chinese=\(escaped)"
  669. XCTAssertEqual(result.value, expected)
  670. }
  671. }
  672. private struct EncodableStruct: Encodable {
  673. let one = "one"
  674. let two = 2
  675. let three = true
  676. let four = [1, 2, 3]
  677. let five = ["a": "a"]
  678. let six = ["a": ["b": "b"]]
  679. let seven = NestedEncodableStruct()
  680. }
  681. private struct NestedEncodableStruct: Encodable {
  682. let a = "a"
  683. }
  684. private class EncodableSuperclass: Encodable {
  685. let one = "one"
  686. let two = 2
  687. let three = true
  688. }
  689. private final class EncodableSubclass: EncodableSuperclass {
  690. let four = [1, 2, 3]
  691. let five = ["a": "a", "b": "b"]
  692. private enum CodingKeys: String, CodingKey {
  693. case four, five
  694. }
  695. override func encode(to encoder: Encoder) throws {
  696. try super.encode(to: encoder)
  697. var container = encoder.container(keyedBy: CodingKeys.self)
  698. try container.encode(four, forKey: .four)
  699. try container.encode(five, forKey: .five)
  700. }
  701. }
  702. private final class ManuallyEncodableSubclass: EncodableSuperclass {
  703. let four = [1, 2, 3]
  704. let five = ["a": "a", "b": "b"]
  705. private enum CodingKeys: String, CodingKey {
  706. case four, five
  707. }
  708. override func encode(to encoder: Encoder) throws {
  709. var keyedContainer = encoder.container(keyedBy: CodingKeys.self)
  710. try keyedContainer.encode(four, forKey: .four)
  711. try keyedContainer.encode(five, forKey: .five)
  712. let superEncoder = keyedContainer.superEncoder()
  713. var superContainer = superEncoder.container(keyedBy: CodingKeys.self)
  714. try superContainer.encode(one, forKey: .four)
  715. let keyedSuperEncoder = keyedContainer.superEncoder(forKey: .four)
  716. var superKeyedContainer = keyedSuperEncoder.container(keyedBy: CodingKeys.self)
  717. try superKeyedContainer.encode(two, forKey: .five)
  718. var unkeyedContainer = keyedContainer.nestedUnkeyedContainer(forKey: .four)
  719. let unkeyedSuperEncoder = unkeyedContainer.superEncoder()
  720. var keyedUnkeyedSuperContainer = unkeyedSuperEncoder.container(keyedBy: CodingKeys.self)
  721. try keyedUnkeyedSuperContainer.encode(one, forKey: .four)
  722. }
  723. }
  724. private struct ManuallyEncodableStruct: Encodable {
  725. let a = ["string": "string"]
  726. let b = [1, 2, 3]
  727. private enum RootKey: String, CodingKey {
  728. case root
  729. }
  730. private enum TypeKeys: String, CodingKey {
  731. case a, b
  732. }
  733. func encode(to encoder: Encoder) throws {
  734. var container = encoder.container(keyedBy: RootKey.self)
  735. var nestedKeyedContainer = container.nestedContainer(keyedBy: TypeKeys.self, forKey: .root)
  736. try nestedKeyedContainer.encode(a, forKey: .a)
  737. var nestedUnkeyedContainer = container.nestedUnkeyedContainer(forKey: .root)
  738. try nestedUnkeyedContainer.encode(b)
  739. var nestedUnkeyedKeyedContainer = nestedUnkeyedContainer.nestedContainer(keyedBy: TypeKeys.self)
  740. try nestedUnkeyedKeyedContainer.encode(a, forKey: .a)
  741. var nestedUnkeyedUnkeyedContainer = nestedUnkeyedContainer.nestedUnkeyedContainer()
  742. try nestedUnkeyedUnkeyedContainer.encode(b)
  743. }
  744. }
  745. private struct FailingOptionalStruct: Encodable {
  746. enum TestedContainer {
  747. case keyed, unkeyed
  748. }
  749. enum CodingKeys: String, CodingKey { case a }
  750. let testedContainer: TestedContainer
  751. func encode(to encoder: Encoder) throws {
  752. var container = encoder.container(keyedBy: CodingKeys.self)
  753. switch testedContainer {
  754. case .keyed:
  755. var nested = container.nestedContainer(keyedBy: CodingKeys.self, forKey: .a)
  756. try nested.encodeNil(forKey: .a)
  757. case .unkeyed:
  758. var nested = container.nestedUnkeyedContainer(forKey: .a)
  759. try nested.encodeNil()
  760. }
  761. }
  762. }
  763. private func XCTAssertQueryEqual(_ query1: String?, _ query2: String?) {
  764. let items1 = query1?.split(separator: "&").sorted()
  765. let items2 = query2?.split(separator: "&").sorted()
  766. XCTAssertEqual(items1, items2)
  767. }