Add listDocuments to collection

Author: op

src/firestore-collection.js

@@ -65,6 +65,23 @@ MockFirestoreCollection.prototype.doc = function (path) {
   return child;
 };
 
+MockFirestoreCollection.prototype.listDocuments = function () {
+  var err = this._nextErr('listDocuments');
+  var self = this;
+  return new Promise(function (resolve, reject) {
+    self._defer('listDocuments', _.toArray(arguments), function () {
+      if (err === null) {
+        var docs = _.map(self.data, function (value, key) {
+          return self.doc(key);
+        });
+        resolve(docs);
+      } else {
+        reject(err);
+      }
+    });
+  });
+};
+
 MockFirestoreCollection.prototype._hasChild = function (key) {
   return _.isObject(this.data) && _.has(this.data, key);
 };

test/unit/firestore-collection.js

@@ -356,4 +356,31 @@ describe('MockFirestoreCollection', function () {
       ]);
     });
   });
+  describe('#listDocuments', function () {
+    it('retrieves all data for existing collection', function(done) {
+      db.autoFlush();
+      var keys = Object.keys(require('./data.json').collections);
+      collection.listDocuments().then(function(refs) {
+        expect(refs.length).to.equal(6);
+        refs.forEach(function(ref) {
+          expect(keys).to.contain(ref.id);
+        });
+        done();
+      }).catch(done);
+    });
+
+    it('retrieves data added to collection', function(done) {
+      db.autoFlush();
+      db.collection('group').add({
+        name: 'test'
+      });
+      db.collection('group').listDocuments().then(function(refs) {
+        expect(refs.length).to.equal(1);
+        refs[0].get().then(function(doc) {
+          expect(doc.data().name).to.equal('test');
+          done();
+        }).catch(done);
+      }).catch(done);
+    });
+  });
 });