Add % encoding in ClientRequest.init, remove % encoding in HTTPServer…

…Request.init
Kitura · Feb 20, 2019 · 4b12bfb · 4b12bfb
1 parent a54fffe
commit 4b12bfb
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 2 deletions.
diff --git a/Sources/KituraNet/ClientRequest.swift b/Sources/KituraNet/ClientRequest.swift
@@ -238,7 +238,7 @@ public class ClientRequest {
                 if thePath.first != "/" {
                     thePath = "/" + thePath
                 }
-                path = thePath
+                path = thePath.addingPercentEncoding(withAllowedCharacters: NSCharacterSet.urlQueryAllowed) ?? thePath
                 self.path = path
             case .username(let userName):
                 self.userName = userName

diff --git a/Sources/KituraNet/HTTP/HTTPServerRequest.swift b/Sources/KituraNet/HTTP/HTTPServerRequest.swift
@@ -148,7 +148,7 @@ public class HTTPServerRequest: ServerRequest {
         self.method = requestHead.method.string()
         self.httpVersionMajor = requestHead.version.major
         self.httpVersionMinor = requestHead.version.minor
-        self._urlString = requestHead.uri
+        self._urlString = requestHead.uri.removingPercentEncoding ?? requestHead.uri
         self.enableSSL = enableSSL
     }
 

diff --git a/Tests/KituraNetTests/ClientRequestTests.swift b/Tests/KituraNetTests/ClientRequestTests.swift
@@ -98,6 +98,20 @@ class ClientRequestTests: KituraNetTest {
         let options = ClientRequest.parse("https://username:password@66o.tech:8080/path?key=value")
         let testRequest = ClientRequest(options: options, callback: testCallback)
         XCTAssertEqual(testRequest.url, "https://username:password@66o.tech:8080/path?key=value")
+
+        let options1: [ClientRequest.Options] = [ .schema("https"),
+                                                  .hostname("66o.tech"),
+                                                  .path("/view/matching?key=\"viewTest\"")
+        ]
+        let testRequest1 = ClientRequest(options: options1, callback: testCallback)
+        XCTAssertEqual(testRequest1.url, "https://66o.tech/view/matching?key=%22viewTest%22")
+
+        let options2: [ClientRequest.Options] = [ .schema("https"),
+                                                  .hostname("66o.tech"),
+                                                  .path("/view/match?key?=value?")
+        ]   
+        let testRequest2 = ClientRequest(options: options2, callback: testCallback)
+        XCTAssertEqual(testRequest2.url, "https://66o.tech/view/match?key?=value?")
     }
 
     func testClientRequestBasicAuthentcation() {