fix(aggregate): add .pipeline() helper to get the current pipeline

Fix #5825
Automattic · Nov 17, 2017 · 22d4657 · 22d4657
1 parent 001f98b
commit 22d4657
Show file tree

Hide file tree

Showing 3 changed files with 29 additions and 1 deletion.
diff --git a/lib/aggregate.js b/lib/aggregate.js
@@ -638,6 +638,22 @@ Aggregate.prototype.facet = function(options) {
   return this.append({$facet: options});
 };
 
+/**
+ * Returns the current pipeline
+ *
+ * ####Example:
+ *
+ *     MyModel.aggregate().match({ test: 1 }).pipeline(); // [{ $match: { test: 1 } }]
+ *
+ * @return {Array}
+ * @api public
+ */
+
+
+Aggregate.prototype.pipeline = function() {
+  return this._pipeline;
+};
+
 /**
  * Executes the aggregate pipeline on the currently bound Model.
  *

diff --git a/package-lock.json b/package-lock.json
diff --git a/test/aggregate.test.js b/test/aggregate.test.js
@@ -773,6 +773,18 @@ describe('aggregate: ', function() {
         });
     });
 
+    it('pipeline() (gh-5825)', function(done) {
+      var aggregate = new Aggregate();
+
+      var pipeline = aggregate.
+        model(db.model('Employee')).
+        match({ sal: { $lt: 16000 } }).
+        pipeline();
+
+      assert.deepEqual(pipeline, [{ $match: { sal: { $lt: 16000 } } }]);
+      done();
+    });
+
     it('explain()', function(done) {
       var aggregate = new Aggregate();
       start.mongodVersion(function(err, version) {