Merge pull request soumak77#101 from Storr-co/firestore-batch-chaining

Enable chaining of firestore batch methods

Author: soumak77

src/firestore.js

@@ -71,7 +71,7 @@ MockFirestore.prototype.runTransaction = function(transFunc) {
 
 MockFirestore.prototype.batch = function () {
   var self = this;
-  return {
+  var batch = {
     set: function(doc, data, opts) {
       var _opts = _.assign({}, { merge: false }, opts);
       if (_opts.merge) {
@@ -80,15 +80,18 @@ MockFirestore.prototype.batch = function () {
       else {
         doc.set(data);
       }
+      return batch;
     },
     create: function(doc, data) {
       doc.create(data);
     },
     update: function(doc, data) {
       doc.update(data);
+      return batch;
     },
     delete: function(doc) {
       doc.delete();
+      return batch;
     },
     commit: function() {
       if (self.queue.events.length > 0) {
@@ -97,6 +100,7 @@ MockFirestore.prototype.batch = function () {
       return Promise.resolve();
     }
   };
+  return batch;
 };
 
 MockFirestore.prototype.collection = function (path) {

test/unit/firestore.js

@@ -189,5 +189,35 @@ describe('MockFirestore', function () {
 
       db.flush();
     });
+
+    it('supports method chaining', function () {
+      var doc1 = db.doc('doc1');
+      var doc2 = db.doc('doc2');
+      var doc3 = db.doc('doc3');
+      var doc4 = db.doc('doc4');
+
+      doc3.set({value: -1});
+      doc4.set({value: 4});
+
+      db.batch()
+        .set(doc1, {value: 1})
+        .set(doc2, {value: 2})
+        .update(doc3, {value: 3})
+        .delete(doc4)
+        .commit();
+
+      var awaitChecks = Promise
+        .all([doc1.get(), doc2.get(), doc3.get(), doc4.get()])
+        .then(function(snaps) {
+          expect(snaps[0].data()).to.deep.equal({value: 1});
+          expect(snaps[1].data()).to.deep.equal({value: 2});
+          expect(snaps[2].data()).to.deep.equal({value: 3});
+          expect(snaps[3].exists).to.equal(false);
+        });
+
+      db.flush();
+
+      return awaitChecks;
+    });
   });
 });