Merge pull request soumak77#100 from Storr-co/firestore-stream-query

Implement Firestore Query#stream mock

Author: soumak77

src/firestore-query.js

@@ -2,6 +2,7 @@
 
 var _ = require('./lodash');
 var assert = require('assert');
+var Stream = require('stream');
 var Promise = require('rsvp').Promise;
 var autoId = require('firebase-auto-ids');
 var QuerySnapshot = require('./firestore-query-snapshot');
@@ -108,6 +109,19 @@ MockFirestoreQuery.prototype.get = function () {
   });
 };
 
+MockFirestoreQuery.prototype.stream = function () {
+  var stream = new Stream();
+
+  this.get().then(function (snapshots) {
+    snapshots.forEach(function (snapshot) {
+      stream.emit('data', snapshot);
+    });
+    stream.emit('end');
+  });
+
+  return stream;
+};
+
 MockFirestoreQuery.prototype.where = function (property, operator, value) {
   var query;
 

test/unit/firestore-collection.js

@@ -268,6 +268,37 @@ describe('MockFirestoreCollection', function () {
     });
   });
 
+  describe('#stream', function () {
+    function makeSnapComparable(snap) {
+      return {
+        id: snap.id,
+        data: snap.data(),
+      };
+    }
+
+    it('returns a stream that emits all results', function (done) {
+      collection.get().then(function (snaps) {
+        var streamDocs = [];
+
+        collection.stream()
+          .on('data', function(snap) {
+            streamDocs.push(makeSnapComparable(snap));
+          })
+          .on('end', function () {
+            try {
+              expect(streamDocs).to.eql(snaps.docs.map(makeSnapComparable));
+              done();
+            } catch (err) {
+              done(err);
+            }
+          });
+
+        db.flush();
+      }).catch(done);
+      db.flush();
+    });
+  });
+
   describe('#orderBy', function () {
     it('allow calling orderBy() on collection', function() {
       expect(function() {