Merge 50d98ae into dc50df3

flintinatux · Aug 7, 2019 · c89461c · c89461c
2 parents dc50df3 + 50d98ae
commit c89461c
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/src.js b/src.js
@@ -81,7 +81,7 @@ const assoc = curry((prop, val, obj) => {
 })
 
 // assocPath :: [k] -> v -> { k: v } -> { k: v }
-const assocPath = curry(([ head, ...tail ], x, obj) =>
+const assocPath = curryN(3, ([ head, ...tail ], x, obj={}) =>
   assoc(head, length(tail) ? assocPath(tail, x, obj[head]) : x, obj)
 )
 

diff --git a/test/assocPath.js b/test/assocPath.js
@@ -4,11 +4,14 @@ const { assocPath } = require('..')
 
 describe('assocPath', () => {
   it('sets a value on a path of a nested object', () =>
-    expect(assocPath(['foo', 'bar'], 'baz', {})).to.eql({ foo: { bar: 'baz' } })
+    expect(assocPath(['foo', 'bar', 'bat'], 'baz', {}))
+      .to.eql({ foo: { bar: { bat: 'baz' } } })
   )
 
   it('is curried', () => {
-    expect(assocPath(['foo', 'bar'])('baz', {})).to.eql({ foo: { bar: 'baz' } })
-    expect(assocPath(['foo', 'bar'], 'baz')({})).to.eql({ foo: { bar: 'baz' } })
+    expect(assocPath(['foo', 'bar', 'bat'])('baz', {}))
+      .to.eql({ foo: { bar: { bat: 'baz' } } })
+    expect(assocPath(['foo', 'bar', 'bat'], 'baz')({}))
+      .to.eql({ foo: { bar: { bat: 'baz' } } })
   })
 })