refactor: support query in EstimatedDocumentCountOperation

Author: mbroadst

lib/operations/estimated_document_count.js

@@ -5,14 +5,27 @@ const defineAspects = require('./operation').defineAspects;
 const CommandOperationV2 = require('./command_v2');
 
 class EstimatedDocumentCountOperation extends CommandOperationV2 {
-  constructor(collection, options) {
+  constructor(collection, query, options) {
+    if (typeof options === 'undefined') {
+      options = query;
+      query = undefined;
+    }
+
     super(collection, options);
     this.collectionName = collection.s.namespace.collection;
+    if (query) {
+      this.query = query;
+    }
   }
 
   execute(server, callback) {
     const options = this.options;
     const cmd = { count: this.collectionName };
+
+    if (this.query) {
+      cmd.query = this.query;
+    }
+
     if (typeof options.skip === 'number') {
       cmd.skip = options.skip;
     }