ParameterEncoderTests.swift 32 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020
  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.success, "a=a")
  187. }
  188. func testEncoderCanEncodeDecimal() {
  189. // Given
  190. let encoder = URLEncodedFormEncoder()
  191. let decimal: Decimal = 1.0
  192. let parameters = ["a": decimal]
  193. // When
  194. let result = Result<String, Error> { try encoder.encode(parameters) }
  195. // Then
  196. XCTAssertEqual(result.success, "a=1")
  197. }
  198. func testEncoderCanEncodeDecimalWithHighPrecision() {
  199. // Given
  200. let encoder = URLEncodedFormEncoder()
  201. let decimal: Decimal = 1.123456
  202. let parameters = ["a": decimal]
  203. // When
  204. let result = Result<String, Error> { try encoder.encode(parameters) }
  205. // Then
  206. XCTAssertEqual(result.success, "a=1.123456")
  207. }
  208. func testEncoderCanEncodeDouble() {
  209. // Given
  210. let encoder = URLEncodedFormEncoder()
  211. let parameters = ["a": 1.0]
  212. // When
  213. let result = Result<String, Error> { try encoder.encode(parameters) }
  214. // Then
  215. XCTAssertEqual(result.success, "a=1.0")
  216. }
  217. func testEncoderCanEncodeFloat() {
  218. // Given
  219. let encoder = URLEncodedFormEncoder()
  220. let parameters: [String: Float] = ["a": 1.0]
  221. // When
  222. let result = Result<String, Error> { try encoder.encode(parameters) }
  223. // Then
  224. XCTAssertEqual(result.success, "a=1.0")
  225. }
  226. func testEncoderCanEncodeInt8() {
  227. // Given
  228. let encoder = URLEncodedFormEncoder()
  229. let parameters: [String: Int8] = ["a": 1]
  230. // When
  231. let result = Result<String, Error> { try encoder.encode(parameters) }
  232. // Then
  233. XCTAssertEqual(result.success, "a=1")
  234. }
  235. func testEncoderCanEncodeInt16() {
  236. // Given
  237. let encoder = URLEncodedFormEncoder()
  238. let parameters: [String: Int16] = ["a": 1]
  239. // When
  240. let result = Result<String, Error> { try encoder.encode(parameters) }
  241. // Then
  242. XCTAssertEqual(result.success, "a=1")
  243. }
  244. func testEncoderCanEncodeInt32() {
  245. // Given
  246. let encoder = URLEncodedFormEncoder()
  247. let parameters: [String: Int32] = ["a": 1]
  248. // When
  249. let result = Result<String, Error> { try encoder.encode(parameters) }
  250. // Then
  251. XCTAssertEqual(result.success, "a=1")
  252. }
  253. func testEncoderCanEncodeInt64() {
  254. // Given
  255. let encoder = URLEncodedFormEncoder()
  256. let parameters: [String: Int64] = ["a": 1]
  257. // When
  258. let result = Result<String, Error> { try encoder.encode(parameters) }
  259. // Then
  260. XCTAssertEqual(result.success, "a=1")
  261. }
  262. func testEncoderCanEncodeUInt() {
  263. // Given
  264. let encoder = URLEncodedFormEncoder()
  265. let parameters: [String: UInt] = ["a": 1]
  266. // When
  267. let result = Result<String, Error> { try encoder.encode(parameters) }
  268. // Then
  269. XCTAssertEqual(result.success, "a=1")
  270. }
  271. func testEncoderCanEncodeUInt8() {
  272. // Given
  273. let encoder = URLEncodedFormEncoder()
  274. let parameters: [String: UInt8] = ["a": 1]
  275. // When
  276. let result = Result<String, Error> { try encoder.encode(parameters) }
  277. // Then
  278. XCTAssertEqual(result.success, "a=1")
  279. }
  280. func testEncoderCanEncodeUInt16() {
  281. // Given
  282. let encoder = URLEncodedFormEncoder()
  283. let parameters: [String: UInt16] = ["a": 1]
  284. // When
  285. let result = Result<String, Error> { try encoder.encode(parameters) }
  286. // Then
  287. XCTAssertEqual(result.success, "a=1")
  288. }
  289. func testEncoderCanEncodeUInt32() {
  290. // Given
  291. let encoder = URLEncodedFormEncoder()
  292. let parameters: [String: UInt32] = ["a": 1]
  293. // When
  294. let result = Result<String, Error> { try encoder.encode(parameters) }
  295. // Then
  296. XCTAssertEqual(result.success, "a=1")
  297. }
  298. func testEncoderCanEncodeUInt64() {
  299. // Given
  300. let encoder = URLEncodedFormEncoder()
  301. let parameters: [String: UInt64] = ["a": 1]
  302. // When
  303. let result = Result<String, Error> { try encoder.encode(parameters) }
  304. // Then
  305. XCTAssertEqual(result.success, "a=1")
  306. }
  307. func testThatNestedDictionariesHaveBracketedKeys() {
  308. // Given
  309. let encoder = URLEncodedFormEncoder()
  310. let parameters = ["a": ["b": "b"]]
  311. // When
  312. let result = Result<String, Error> { try encoder.encode(parameters) }
  313. // Then
  314. XCTAssertEqual(result.success, "a%5Bb%5D=b")
  315. }
  316. func testThatEncodableStructCanBeEncoded() {
  317. // Given
  318. let encoder = URLEncodedFormEncoder()
  319. let parameters = EncodableStruct()
  320. // When
  321. let result = Result<String, Error> { try encoder.encode(parameters) }
  322. // Then
  323. let expected = "five%5Ba%5D=a&four%5B%5D=1&four%5B%5D=2&four%5B%5D=3&one=one&seven%5Ba%5D=a&six%5Ba%5D%5Bb%5D=b&three=1&two=2"
  324. XCTAssertEqual(result.success, expected)
  325. }
  326. func testThatManuallyEncodableStructCanBeEncoded() {
  327. // Given
  328. let encoder = URLEncodedFormEncoder()
  329. let parameters = ManuallyEncodableStruct()
  330. // When
  331. let result = Result<String, Error> { try encoder.encode(parameters) }
  332. // Then
  333. let expected = "root%5B%5D%5B%5D%5B%5D=1&root%5B%5D%5B%5D%5B%5D=2&root%5B%5D%5B%5D%5B%5D=3&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"
  334. XCTAssertEqual(result.success, expected)
  335. }
  336. func testThatEncodableClassWithNoInheritanceCanBeEncoded() {
  337. // Given
  338. let encoder = URLEncodedFormEncoder()
  339. let parameters = EncodableSuperclass()
  340. // When
  341. let result = Result<String, Error> { try encoder.encode(parameters) }
  342. // Then
  343. XCTAssertEqual(result.success, "one=one&three=1&two=2")
  344. }
  345. func testThatEncodableSubclassCanBeEncoded() {
  346. // Given
  347. let encoder = URLEncodedFormEncoder()
  348. let parameters = EncodableSubclass()
  349. // When
  350. let result = Result<String, Error> { try encoder.encode(parameters) }
  351. // Then
  352. let expected = "five%5Ba%5D=a&five%5Bb%5D=b&four%5B%5D=1&four%5B%5D=2&four%5B%5D=3&one=one&three=1&two=2"
  353. XCTAssertEqual(result.success, expected)
  354. }
  355. func testThatEncodableSubclassCanBeEncodedInImplementationOrderWhenAlphabetizeKeysIsFalse() {
  356. // Given
  357. let encoder = URLEncodedFormEncoder(alphabetizeKeyValuePairs: false)
  358. let parameters = EncodableStruct()
  359. // When
  360. let result = Result<String, Error> { try encoder.encode(parameters) }
  361. // Then
  362. let expected = "one=one&two=2&three=1&four%5B%5D=1&four%5B%5D=2&four%5B%5D=3&five%5Ba%5D=a&six%5Ba%5D%5Bb%5D=b&seven%5Ba%5D=a"
  363. XCTAssertEqual(result.success, expected)
  364. }
  365. func testThatManuallyEncodableSubclassCanBeEncoded() {
  366. // Given
  367. let encoder = URLEncodedFormEncoder()
  368. let parameters = ManuallyEncodableSubclass()
  369. // When
  370. let result = Result<String, Error> { try encoder.encode(parameters) }
  371. // Then
  372. let expected = "five%5Ba%5D=a&five%5Bb%5D=b&four%5Bfive%5D=2&four%5Bfour%5D=one"
  373. XCTAssertEqual(result.success, expected)
  374. }
  375. func testThatARootArrayCannotBeEncoded() {
  376. // Given
  377. let encoder = URLEncodedFormEncoder()
  378. let parameters = [1]
  379. // When
  380. let result = Result<String, Error> { try encoder.encode(parameters) }
  381. // Then
  382. XCTAssertFalse(result.isSuccess)
  383. }
  384. func testThatARootValueCannotBeEncoded() {
  385. // Given
  386. let encoder = URLEncodedFormEncoder()
  387. let parameters = "string"
  388. // When
  389. let result = Result<String, Error> { try encoder.encode(parameters) }
  390. // Then
  391. XCTAssertFalse(result.isSuccess)
  392. }
  393. func testThatOptionalValuesCannotBeEncoded() {
  394. // Given
  395. let encoder = URLEncodedFormEncoder()
  396. let parameters: [String: String?] = ["string": nil]
  397. // When
  398. let result = Result<String, Error> { try encoder.encode(parameters) }
  399. // Then
  400. XCTAssertFalse(result.isSuccess)
  401. }
  402. func testThatArraysCanBeEncodedWithoutBrackets() {
  403. // Given
  404. let encoder = URLEncodedFormEncoder(arrayEncoding: .noBrackets)
  405. let parameters = ["array": [1, 2]]
  406. // When
  407. let result = Result<String, Error> { try encoder.encode(parameters) }
  408. // Then
  409. XCTAssertEqual(result.success, "array=1&array=2")
  410. }
  411. func testThatBoolsCanBeLiteralEncoded() {
  412. // Given
  413. let encoder = URLEncodedFormEncoder(boolEncoding: .literal)
  414. let parameters = ["bool": true]
  415. // When
  416. let result = Result<String, Error> { try encoder.encode(parameters) }
  417. // Then
  418. XCTAssertEqual(result.success, "bool=true")
  419. }
  420. func testThatDataCanBeEncoded() {
  421. // Given
  422. let encoder = URLEncodedFormEncoder()
  423. let parameters = ["data": Data("data".utf8)]
  424. // When
  425. let result = Result<String, Error> { try encoder.encode(parameters) }
  426. // Then
  427. XCTAssertEqual(result.success, "data=ZGF0YQ%3D%3D")
  428. }
  429. func testThatCustomDataEncodingFailsWhenErrorIsThrown() {
  430. // Given
  431. struct DataEncodingError: Error {}
  432. let encoder = URLEncodedFormEncoder(dataEncoding: .custom { _ in throw DataEncodingError() })
  433. let parameters = ["data": Data("data".utf8)]
  434. // When
  435. let result = Result<String, Error> { try encoder.encode(parameters) }
  436. // Then
  437. XCTAssertTrue(result.isFailure)
  438. XCTAssertTrue(result.failure is DataEncodingError)
  439. }
  440. func testThatDatesCanBeEncoded() {
  441. // Given
  442. let encoder = URLEncodedFormEncoder(dateEncoding: .deferredToDate)
  443. let parameters = ["date": Date(timeIntervalSinceReferenceDate: 123.456)]
  444. // When
  445. let result = Result<String, Error> { try encoder.encode(parameters) }
  446. // Then
  447. XCTAssertEqual(result.success, "date=123.456")
  448. }
  449. func testThatDatesCanBeEncodedAsSecondsSince1970() {
  450. // Given
  451. let encoder = URLEncodedFormEncoder(dateEncoding: .secondsSince1970)
  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.success, "date=978307323.456")
  457. }
  458. func testThatDatesCanBeEncodedAsMillisecondsSince1970() {
  459. // Given
  460. let encoder = URLEncodedFormEncoder(dateEncoding: .millisecondsSince1970)
  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.success, "date=978307323456.0")
  466. }
  467. func testThatDatesCanBeEncodedAsISO8601Formatted() {
  468. // Given
  469. let encoder = URLEncodedFormEncoder(dateEncoding: .iso8601)
  470. let parameters = ["date": Date(timeIntervalSinceReferenceDate: 123.456)]
  471. // When
  472. let result = Result<String, Error> { try encoder.encode(parameters) }
  473. // Then
  474. XCTAssertEqual(result.success, "date=2001-01-01T00%3A02%3A03Z")
  475. }
  476. func testThatDatesCanBeEncodedAsFormatted() {
  477. // Given
  478. let dateFormatter = DateFormatter()
  479. dateFormatter.dateFormat = "yyyy-MM-dd HH:mm:ss.SSSS"
  480. dateFormatter.timeZone = TimeZone(secondsFromGMT: 0)
  481. let encoder = URLEncodedFormEncoder(dateEncoding: .formatted(dateFormatter))
  482. let parameters = ["date": Date(timeIntervalSinceReferenceDate: 123.456)]
  483. // When
  484. let result = Result<String, Error> { try encoder.encode(parameters) }
  485. // Then
  486. XCTAssertEqual(result.success, "date=2001-01-01%2000%3A02%3A03.4560")
  487. }
  488. func testThatDatesCanBeEncodedAsCustomFormatted() {
  489. // Given
  490. let encoder = URLEncodedFormEncoder(dateEncoding: .custom { "\($0.timeIntervalSinceReferenceDate)" })
  491. let parameters = ["date": Date(timeIntervalSinceReferenceDate: 123.456)]
  492. // When
  493. let result = Result<String, Error> { try encoder.encode(parameters) }
  494. // Then
  495. XCTAssertEqual(result.success, "date=123.456")
  496. }
  497. func testEncoderThrowsErrorWhenCustomDateEncodingFails() {
  498. // Given
  499. struct DateEncodingError: Error {}
  500. let encoder = URLEncodedFormEncoder(dateEncoding: .custom { _ in throw DateEncodingError() })
  501. let parameters = ["date": Date(timeIntervalSinceReferenceDate: 123.456)]
  502. // When
  503. let result = Result<String, Error> { try encoder.encode(parameters) }
  504. // Then
  505. XCTAssertTrue(result.isFailure)
  506. XCTAssertTrue(result.failure is DateEncodingError)
  507. }
  508. func testThatKeysCanBeEncodedIntoSnakeCase() {
  509. // Given
  510. let encoder = URLEncodedFormEncoder(keyEncoding: .convertToSnakeCase)
  511. let parameters = ["oneTwoThree": "oneTwoThree"]
  512. // When
  513. let result = Result<String, Error> { try encoder.encode(parameters) }
  514. // Then
  515. XCTAssertEqual(result.success, "one_two_three=oneTwoThree")
  516. }
  517. func testThatKeysCanBeEncodedIntoKebabCase() {
  518. // Given
  519. let encoder = URLEncodedFormEncoder(keyEncoding: .convertToKebabCase)
  520. let parameters = ["oneTwoThree": "oneTwoThree"]
  521. // When
  522. let result = Result<String, Error> { try encoder.encode(parameters) }
  523. // Then
  524. XCTAssertEqual(result.success, "one-two-three=oneTwoThree")
  525. }
  526. func testThatKeysCanBeEncodedIntoACapitalizedString() {
  527. // Given
  528. let encoder = URLEncodedFormEncoder(keyEncoding: .capitalized)
  529. let parameters = ["oneTwoThree": "oneTwoThree"]
  530. // When
  531. let result = Result<String, Error> { try encoder.encode(parameters) }
  532. // Then
  533. XCTAssertEqual(result.success, "OneTwoThree=oneTwoThree")
  534. }
  535. func testThatKeysCanBeEncodedIntoALowercasedString() {
  536. // Given
  537. let encoder = URLEncodedFormEncoder(keyEncoding: .lowercased)
  538. let parameters = ["oneTwoThree": "oneTwoThree"]
  539. // When
  540. let result = Result<String, Error> { try encoder.encode(parameters) }
  541. // Then
  542. XCTAssertEqual(result.success, "onetwothree=oneTwoThree")
  543. }
  544. func testThatKeysCanBeEncodedIntoAnUppercasedString() {
  545. // Given
  546. let encoder = URLEncodedFormEncoder(keyEncoding: .uppercased)
  547. let parameters = ["oneTwoThree": "oneTwoThree"]
  548. // When
  549. let result = Result<String, Error> { try encoder.encode(parameters) }
  550. // Then
  551. XCTAssertEqual(result.success, "ONETWOTHREE=oneTwoThree")
  552. }
  553. func testThatKeysCanBeCustomEncoded() {
  554. // Given
  555. let encoder = URLEncodedFormEncoder(keyEncoding: .custom { _ in "A" })
  556. let parameters = ["oneTwoThree": "oneTwoThree"]
  557. // When
  558. let result = Result<String, Error> { try encoder.encode(parameters) }
  559. // Then
  560. XCTAssertEqual(result.success, "A=oneTwoThree")
  561. }
  562. func testThatSpacesCanBeEncodedAsPluses() {
  563. // Given
  564. let encoder = URLEncodedFormEncoder(spaceEncoding: .plusReplaced)
  565. let parameters = ["spaces": "replace with spaces"]
  566. // When
  567. let result = Result<String, Error> { try encoder.encode(parameters) }
  568. // Then
  569. XCTAssertEqual(result.success, "spaces=replace+with+spaces")
  570. }
  571. func testThatEscapedCharactersCanBeCustomized() {
  572. // Given
  573. var allowed = CharacterSet.afURLQueryAllowed
  574. allowed.remove(charactersIn: "?/")
  575. let encoder = URLEncodedFormEncoder(allowedCharacters: allowed)
  576. let parameters = ["allowed": "?/"]
  577. // When
  578. let result = Result<String, Error> { try encoder.encode(parameters) }
  579. // Then
  580. XCTAssertEqual(result.success, "allowed=%3F%2F")
  581. }
  582. func testThatUnreservedCharactersAreNotPercentEscaped() {
  583. // Given
  584. let encoder = URLEncodedFormEncoder()
  585. let parameters = ["lowercase": "abcdefghijklmnopqrstuvwxyz",
  586. "numbers": "0123456789",
  587. "uppercase": "ABCDEFGHIJKLMNOPQRSTUVWXYZ"]
  588. // When
  589. let result = Result<String, Error> { try encoder.encode(parameters) }
  590. // Then
  591. let expected = "lowercase=abcdefghijklmnopqrstuvwxyz&numbers=0123456789&uppercase=ABCDEFGHIJKLMNOPQRSTUVWXYZ"
  592. XCTAssertEqual(result.success, expected)
  593. }
  594. func testThatReservedCharactersArePercentEscaped() {
  595. // Given
  596. let encoder = URLEncodedFormEncoder()
  597. let generalDelimiters = ":#[]@"
  598. let subDelimiters = "!$&'()*+,;="
  599. let parameters = ["reserved": "\(generalDelimiters)\(subDelimiters)"]
  600. // When
  601. let result = Result<String, Error> { try encoder.encode(parameters) }
  602. // Then
  603. XCTAssertEqual(result.success, "reserved=%3A%23%5B%5D%40%21%24%26%27%28%29%2A%2B%2C%3B%3D")
  604. }
  605. func testThatIllegalASCIICharactersArePercentEscaped() {
  606. // Given
  607. let encoder = URLEncodedFormEncoder()
  608. let parameters = ["illegal": " \"#%<>[]\\^`{}|"]
  609. // When
  610. let result = Result<String, Error> { try encoder.encode(parameters) }
  611. // Then
  612. XCTAssertEqual(result.success, "illegal=%20%22%23%25%3C%3E%5B%5D%5C%5E%60%7B%7D%7C")
  613. }
  614. func testThatAmpersandsInKeysAndValuesArePercentEscaped() {
  615. // Given
  616. let encoder = URLEncodedFormEncoder()
  617. let parameters = ["foo&bar": "baz&qux", "foobar": "bazqux"]
  618. // When
  619. let result = Result<String, Error> { try encoder.encode(parameters) }
  620. // Then
  621. XCTAssertEqual(result.success, "foo%26bar=baz%26qux&foobar=bazqux")
  622. }
  623. func testThatQuestionMarksInKeysAndValuesAreNotPercentEscaped() {
  624. // Given
  625. let encoder = URLEncodedFormEncoder()
  626. let parameters = ["?foo?": "?bar?"]
  627. // When
  628. let result = Result<String, Error> { try encoder.encode(parameters) }
  629. // Then
  630. XCTAssertEqual(result.success, "?foo?=?bar?")
  631. }
  632. func testThatSlashesInKeysAndValuesAreNotPercentEscaped() {
  633. // Given
  634. let encoder = URLEncodedFormEncoder()
  635. let parameters = ["foo": "/bar/baz/qux"]
  636. // When
  637. let result = Result<String, Error> { try encoder.encode(parameters) }
  638. // Then
  639. XCTAssertEqual(result.success, "foo=/bar/baz/qux")
  640. }
  641. func testThatSpacesInKeysAndValuesArePercentEscaped() {
  642. // Given
  643. let encoder = URLEncodedFormEncoder()
  644. let parameters = [" foo ": " bar "]
  645. // When
  646. let result = Result<String, Error> { try encoder.encode(parameters) }
  647. // Then
  648. XCTAssertEqual(result.success, "%20foo%20=%20bar%20")
  649. }
  650. func testThatPlusesInKeysAndValuesArePercentEscaped() {
  651. // Given
  652. let encoder = URLEncodedFormEncoder()
  653. let parameters = ["+foo+": "+bar+"]
  654. // When
  655. let result = Result<String, Error> { try encoder.encode(parameters) }
  656. // Then
  657. XCTAssertEqual(result.success, "%2Bfoo%2B=%2Bbar%2B")
  658. }
  659. func testThatPercentsInKeysAndValuesArePercentEscaped() {
  660. // Given
  661. let encoder = URLEncodedFormEncoder()
  662. let parameters = ["percent%": "%25"]
  663. // When
  664. let result = Result<String, Error> { try encoder.encode(parameters) }
  665. // Then
  666. XCTAssertEqual(result.success, "percent%25=%2525")
  667. }
  668. func testThatNonLatinCharactersArePercentEscaped() {
  669. // Given
  670. let encoder = URLEncodedFormEncoder()
  671. let parameters = ["french": "français",
  672. "japanese": "日本語",
  673. "arabic": "العربية",
  674. "emoji": "😃"]
  675. // When
  676. let result = Result<String, Error> { try encoder.encode(parameters) }
  677. // Then
  678. let expectedParameterValues = ["arabic=%D8%A7%D9%84%D8%B9%D8%B1%D8%A8%D9%8A%D8%A9",
  679. "emoji=%F0%9F%98%83",
  680. "french=fran%C3%A7ais",
  681. "japanese=%E6%97%A5%E6%9C%AC%E8%AA%9E"].joined(separator: "&")
  682. XCTAssertEqual(result.success, expectedParameterValues)
  683. }
  684. func testStringWithThousandsOfChineseCharactersIsPercentEscaped() {
  685. // Given
  686. let encoder = URLEncodedFormEncoder()
  687. let repeatedCount = 2000
  688. let parameters = ["chinese": String(repeating: "一二三四五六七八九十", count: repeatedCount)]
  689. // When
  690. let result = Result<String, Error> { try encoder.encode(parameters) }
  691. // Then
  692. 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",
  693. count: repeatedCount)
  694. let expected = "chinese=\(escaped)"
  695. XCTAssertEqual(result.success, expected)
  696. }
  697. }
  698. private struct EncodableStruct: Encodable {
  699. let one = "one"
  700. let two = 2
  701. let three = true
  702. let four = [1, 2, 3]
  703. let five = ["a": "a"]
  704. let six = ["a": ["b": "b"]]
  705. let seven = NestedEncodableStruct()
  706. }
  707. private struct NestedEncodableStruct: Encodable {
  708. let a = "a"
  709. }
  710. private class EncodableSuperclass: Encodable {
  711. let one = "one"
  712. let two = 2
  713. let three = true
  714. }
  715. private final class EncodableSubclass: EncodableSuperclass {
  716. let four = [1, 2, 3]
  717. let five = ["a": "a", "b": "b"]
  718. private enum CodingKeys: String, CodingKey {
  719. case four, five
  720. }
  721. override func encode(to encoder: Encoder) throws {
  722. try super.encode(to: encoder)
  723. var container = encoder.container(keyedBy: CodingKeys.self)
  724. try container.encode(four, forKey: .four)
  725. try container.encode(five, forKey: .five)
  726. }
  727. }
  728. private final class ManuallyEncodableSubclass: EncodableSuperclass {
  729. let four = [1, 2, 3]
  730. let five = ["a": "a", "b": "b"]
  731. private enum CodingKeys: String, CodingKey {
  732. case four, five
  733. }
  734. override func encode(to encoder: Encoder) throws {
  735. var keyedContainer = encoder.container(keyedBy: CodingKeys.self)
  736. try keyedContainer.encode(four, forKey: .four)
  737. try keyedContainer.encode(five, forKey: .five)
  738. let superEncoder = keyedContainer.superEncoder()
  739. var superContainer = superEncoder.container(keyedBy: CodingKeys.self)
  740. try superContainer.encode(one, forKey: .four)
  741. let keyedSuperEncoder = keyedContainer.superEncoder(forKey: .four)
  742. var superKeyedContainer = keyedSuperEncoder.container(keyedBy: CodingKeys.self)
  743. try superKeyedContainer.encode(two, forKey: .five)
  744. var unkeyedContainer = keyedContainer.nestedUnkeyedContainer(forKey: .four)
  745. let unkeyedSuperEncoder = unkeyedContainer.superEncoder()
  746. var keyedUnkeyedSuperContainer = unkeyedSuperEncoder.container(keyedBy: CodingKeys.self)
  747. try keyedUnkeyedSuperContainer.encode(one, forKey: .four)
  748. }
  749. }
  750. private struct ManuallyEncodableStruct: Encodable {
  751. let a = ["string": "string"]
  752. let b = [1, 2, 3]
  753. private enum RootKey: String, CodingKey {
  754. case root
  755. }
  756. private enum TypeKeys: String, CodingKey {
  757. case a, b
  758. }
  759. func encode(to encoder: Encoder) throws {
  760. var container = encoder.container(keyedBy: RootKey.self)
  761. var nestedKeyedContainer = container.nestedContainer(keyedBy: TypeKeys.self, forKey: .root)
  762. try nestedKeyedContainer.encode(a, forKey: .a)
  763. var nestedUnkeyedContainer = container.nestedUnkeyedContainer(forKey: .root)
  764. try nestedUnkeyedContainer.encode(b)
  765. var nestedUnkeyedKeyedContainer = nestedUnkeyedContainer.nestedContainer(keyedBy: TypeKeys.self)
  766. try nestedUnkeyedKeyedContainer.encode(a, forKey: .a)
  767. var nestedUnkeyedUnkeyedContainer = nestedUnkeyedContainer.nestedUnkeyedContainer()
  768. try nestedUnkeyedUnkeyedContainer.encode(b)
  769. }
  770. }
  771. private struct FailingOptionalStruct: Encodable {
  772. enum TestedContainer {
  773. case keyed, unkeyed
  774. }
  775. enum CodingKeys: String, CodingKey { case a }
  776. let testedContainer: TestedContainer
  777. func encode(to encoder: Encoder) throws {
  778. var container = encoder.container(keyedBy: CodingKeys.self)
  779. switch testedContainer {
  780. case .keyed:
  781. var nested = container.nestedContainer(keyedBy: CodingKeys.self, forKey: .a)
  782. try nested.encodeNil(forKey: .a)
  783. case .unkeyed:
  784. var nested = container.nestedUnkeyedContainer(forKey: .a)
  785. try nested.encodeNil()
  786. }
  787. }
  788. }