add .set(obj) and .inc(obj) support

Author: jonathanong

lib/utils.js

@@ -84,8 +84,12 @@ exports.updateify = function (constructor, fname) {
     constructor.prototype[$op] = function (key, value) {
       if (fname) this[fname]()
       var document = this.document
-      var changes = document[$op] = document[$op] || {}
-      changes[key] = value
+      if (typeof key === 'object') {
+        document[$op] = key
+      } else {
+        var changes = document[$op] = document[$op] || {}
+        changes[key] = value
+      }
       return this
     }
     // only set without a $ if it doesn't overwrite another property
@@ -98,8 +102,12 @@ exports.updateify = function (constructor, fname) {
   constructor.prototype.inc = function (key, value) {
     if (fname) this[fname]()
     var document = this.document
-    var changes = document.$inc = document.$inc || {}
-    changes[key] = value == null ? 1 : value
+    if (typeof key === 'object') {
+      document.$inc = key
+    } else {
+      var changes = document.$inc = document.$inc || {}
+      changes[key] = value == null ? 1 : value
+    }
     return this
   }
 

test/update.js

@@ -30,6 +30,19 @@ describe('.find().update()', function () {
     })
   })
 
+  it('.set(obj)', function () {
+    return collection.find().set({
+      obj: true
+    }).then(function () {
+      return collection.find()
+    }).then(function (docs) {
+      assert(docs.length > 1)
+      assert(docs.every(function (doc) {
+        return doc.obj = true
+      }))
+    })
+  })
+
   it('.set(key, value)', function () {
     return collection.find().set('asdf', 3).then(function () {
       return collection.find()
@@ -190,6 +203,16 @@ describe('.find().inc()', function () {
       assert.equal(7, doc.qqqq)
     })
   })
+
+  it('(obj)', function () {
+    return collection.find('name', 'taylor').inc({
+      qqqq: 1
+    }).then(function () {
+      return collection.findOne('name', 'taylor')
+    }).then(function (doc) {
+      assert.equal(8, doc.qqqq)
+    })
+  })
 })
 
 describe('.find().unset()', function () {