Added very rough support for collectionGroup

Only supports finding 1 subcollection at the moment
siteline · Nov 14, 2019 · cfd816c · cfd816c
1 parent eac85a7
commit cfd816c
Show file tree

Hide file tree

Showing 2 changed files with 44 additions and 0 deletions.
diff --git a/browser/firebasemock.js b/browser/firebasemock.js
@@ -51778,6 +51778,28 @@ MockFirestore.prototype.batch = function () {
  return batch;
 };
 
+MockFirestore.prototype.collectionGroup = function (subcollection) {
+ // need to find all subcollections with the name `subcollection`
+ const children = []
+ this._findAllSubcollections (subcollection, this, children)
+ if (children.length) {
+ // TODO: support more than just 1 result
+ return children[0]
+ } else {
+ return this.collection(subcollection)
+ }
+}
+
+MockFirestore.prototype._findAllSubcollections = function (name, obj, foundObjs) {
+ _.forEach(obj.children, (value, key) => {
+ if (key === name) {
+ foundObjs.push(value)
+ } else {
+ return this._findAllSubcollections (name, value, foundObjs)
+ }
+ })
+}
+
 MockFirestore.prototype.collection = function (path) {
  return this._child(path, false);
 };

diff --git a/src/firestore.js b/src/firestore.js
@@ -137,6 +137,28 @@ MockFirestore.prototype.batch = function () {
  return batch;
 };
 
+MockFirestore.prototype.collectionGroup = function (subcollection) {
+ // need to find all subcollections with the name `subcollection`
+ const children = []
+ this._findAllSubcollections (subcollection, this, children)
+ if (children.length) {
+ // TODO: support more than just 1 result
+ return children[0]
+ } else {
+ return this.collection(subcollection)
+ }
+}
+
+MockFirestore.prototype._findAllSubcollections = function (name, obj, foundObjs) {
+ _.forEach(obj.children, (value, key) => {
+ if (key === name) {
+ foundObjs.push(value)
+ } else {
+ return this._findAllSubcollections (name, value, foundObjs)
+ }
+ })
+}
+
 MockFirestore.prototype.collection = function (path) {
  return this._child(path, false);
 };