Moved generator based aggregation test to generator_based_tests.js

Author: christkv

test/functional/aggregation_tests.js

@@ -826,45 +826,4 @@ exports['Should correctly handle ISODate date matches in aggregation framework']
     });
     // DOC_END
   }
-}
-
-exports['should maintain batch size between calls to receive new batches'] = {
-  metadata: {
-    requires: { generators: true, topology: 'single' , node: ">6.0.0" }
-  },
-
-  // The actual test we wish to run
-  test: function(configure, test) {
-    var co = require('co');
-
-    co(function*() {
-      var instance = configure.newDbInstance({ w: 1 }, { poolSize: 1 });
-      var db = yield instance.open();
-
-      var docs = [ { a: 1 }, { a: 1 }, { a: 1 }, { a: 1 }, { a: 1 }, { a: 1 } ];
-      var collection = db.collection('batchSizeContinue');
-      yield collection.insertMany(docs, { w: 1 });
-      var cursor = collection.aggregate([
-          { $match: { a: 1 } }, { $limit: 6 }
-        ], {
-          cursor: { batchSize: 2 }
-        });
-
-      var count = 0;
-      while (yield cursor.hasNext()) {
-        var data = yield cursor.next();
-        test.equal(data.a, 1);
-
-        // ensure batch size is as specified
-        test.equal(cursor.cursorState.documents.length, 2);
-        count++;
-      }
-
-      test.equal(count, 6);
-      db.close();
-      test.done();
-    }).catch(err => {
-      console.log(err)
-    });
-  }
-}
+}

test/functional/generator_based_tests.js

@@ -0,0 +1,40 @@
+exports['should maintain batch size between calls to receive new batches'] = {
+  metadata: {
+    requires: { generators: true, topology: 'single' , node: ">6.0.0" }
+  },
+
+  // The actual test we wish to run
+  test: function(configure, test) {
+    var co = require('co');
+
+    co(function*() {
+      var instance = configure.newDbInstance({ w: 1 }, { poolSize: 1 });
+      var db = yield instance.open();
+
+      var docs = [ { a: 1 }, { a: 1 }, { a: 1 }, { a: 1 }, { a: 1 }, { a: 1 } ];
+      var collection = db.collection('batchSizeContinue');
+      yield collection.insertMany(docs, { w: 1 });
+      var cursor = collection.aggregate([
+          { $match: { a: 1 } }, { $limit: 6 }
+        ], {
+          cursor: { batchSize: 2 }
+        });
+
+      var count = 0;
+      while (yield cursor.hasNext()) {
+        var data = yield cursor.next();
+        test.equal(data.a, 1);
+
+        // ensure batch size is as specified
+        test.equal(cursor.cursorState.documents.length, 2);
+        count++;
+      }
+
+      test.equal(count, 6);
+      db.close();
+      test.done();
+    }).catch(err => {
+      console.log(err)
+    });
+  }
+}

test/runner.js

@@ -364,6 +364,7 @@ try {
   testFiles.push('/test/functional/buffering_proxy_tests.js');
   testFiles.push('/test/functional/view_tests.js');
   testFiles.push('/test/functional/crud_spec_tests.js');
+  testFiles.push('/test/functional/generator_based_tests.js');
 } catch(err) {}
 
 // Add all the tests to run