#4 Fix Issue When Writing to Empty Dictionary

rogerluan · Jun 2, 2022 · dd73b21 · dd73b21
2 parents 3df9171 + 81e905a
commit dd73b21
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/Sources/KeyPath.swift b/Sources/KeyPath.swift
@@ -86,7 +86,7 @@ public extension Dictionary where Key == String {
                 // Reached the end of the key path.
                 self[head] = newValue as? Value
             case let (head, remainingKeyPath)?:
-                let value = self[head]
+                let value: Any = self[head] ?? [:]
                 if var nestedDictionary = value as? [Key:Any] {
                     // Key path has a tail we need to traverse
                     nestedDictionary[keyPath: remainingKeyPath] = newValue

diff --git a/Tests/KeyPathTests.swift b/Tests/KeyPathTests.swift
@@ -92,4 +92,11 @@ final class KeyPathTests : XCTestCase {
         let rhs = "this.is.path.end"
         XCTAssertEqual(KeyPath(lhs) + KeyPath(rhs), "this.is.path.start.this.is.path.end")
     }
+
+    func test_writingToEmptyDictionary() throws {
+        let data = "this is my data"
+        var dictionary: [String:Any] = [:]
+        dictionary[keyPath: "nested.data"] = data
+        XCTAssertEqual(dictionary[keyPath: "nested.data"] as? String, data)
+    }
 }