Merge pull request #13 from hyperoslo/feature/clean-argument-keys

Feature clean argument keys
hyperoslo · Nov 23, 2015 · dfec953 · dfec953
2 parents 6d02ff3 + 7e4c0b3
commit dfec953
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/Pod/Tests/TestCompass.swift b/Pod/Tests/TestCompass.swift
@@ -23,7 +23,7 @@ class TestCompass: XCTestCase {
     let url = NSURL(string: "compassTests://profile:testUser")!
     Compass.parse(url) { route, arguments in
       XCTAssertEqual("profile:{user}", route)
-      XCTAssertEqual(arguments["{user}"], "testUser")
+      XCTAssertEqual(arguments["user"], "testUser")
     }
   }
 

diff --git a/Source/Compass.swift b/Source/Compass.swift
@@ -32,7 +32,7 @@ public struct Compass {
 
         if queryArguments.count == routeArguments.count {
           for (index, key) in routeArguments.enumerate() {
-            arguments[key] = index <= queryArguments.count && "\(query):" != prefix
+            arguments[String(key.characters.dropFirst().dropLast())] = index <= queryArguments.count && "\(query):" != prefix
               ? queryArguments[index] : nil
           }
           completion(route: route, arguments: arguments)