Adds update functionality to Map entries (#100)

Resolves issue #89 by supporting the Map set syntax in the update function.
kolodny · Mar 22, 2018 · e5e6252 · e5e6252
1 parent d9c800c
commit e5e6252
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 1 deletion.
diff --git a/index.js b/index.js
@@ -83,7 +83,11 @@ function newContext() {
           if (nextObject === object) {
             nextObject = copy(object);
           }
-          nextObject[key] = nextValueForKey;
+          if (type(nextObject) === 'Map') {
+            nextObject.set(key, nextValueForKey);
+          } else {
+            nextObject[key] = nextValueForKey;
+          }
         }
       }
     })

diff --git a/test.js b/test.js
@@ -529,4 +529,20 @@ describe('update', function() {
     expect(state2.items.top).toBe(0)
   });
 
+  it('supports Maps', function() {
+    var state = new Map([
+      ['mapKey', 'mapValue']
+    ]);
+
+    var updatedState = update(state, {
+      ['mapKey']: {$set: 'updatedMapValue' }
+    });
+
+    expect(updatedState).toEqual(
+      new Map([
+        ['mapKey', 'updatedMapValue']
+      ])
+    );
+  });
+
 });