HELP-21340: Adding Tests For Max BSON Size Batching (#2693)

test: Adding tests for max byte size batching

Author: durran

test/functional/bulk.test.js

@@ -9,10 +9,13 @@ const {
 const test = require('./shared').assert;
 const { MongoError } = require('../../src/error');
 const { Long } = require('../../src');
+const crypto = require('crypto');
 const chai = require('chai');
 const expect = chai.expect;
 chai.use(require('chai-subset'));
 
+const MAX_BSON_SIZE = 16777216;
+
 describe('Bulk', function () {
   before(function () {
     return setupDatabase(this.configuration);
@@ -1627,6 +1630,54 @@ describe('Bulk', function () {
       });
   });
 
+  it('properly accounts for bson size in bytes in bulk ordered inserts', function () {
+    const client = this.configuration.newClient();
+    const size = MAX_BSON_SIZE / 2;
+    const largeString = crypto.randomBytes(size - 100).toString('hex');
+    const documents = [{ s: largeString }, { s: largeString }];
+
+    let db;
+
+    return client
+      .connect()
+      .then(() => {
+        db = client.db(this.configuration.db);
+        return db.dropCollection('doesnt_matter').catch(() => {});
+      })
+      .then(() => {
+        return db.createCollection('doesnt_matter');
+      })
+      .then(() => {
+        const coll = db.collection('doesnt_matter');
+        coll.insertMany(documents, { ordered: true });
+      })
+      .finally(() => client.close());
+  });
+
+  it('properly accounts for bson size in bytes in bulk unordered inserts', function () {
+    const client = this.configuration.newClient();
+    const size = MAX_BSON_SIZE / 2;
+    const largeString = crypto.randomBytes(size - 100).toString('hex');
+    const documents = [{ s: largeString }, { s: largeString }];
+
+    let db;
+
+    return client
+      .connect()
+      .then(() => {
+        db = client.db(this.configuration.db);
+        return db.dropCollection('doesnt_matter').catch(() => {});
+      })
+      .then(() => {
+        return db.createCollection('doesnt_matter');
+      })
+      .then(() => {
+        const coll = db.collection('doesnt_matter');
+        coll.insertMany(documents, { ordered: false });
+      })
+      .finally(() => client.close());
+  });
+
   function testPropagationOfBulkWriteError(bulk) {
     return bulk.execute().then(
       err => {