1 files changed, 116 insertions, 0 deletions
diff --git a/jstests/core/write/batch_write_collation_estsize.js b/jstests/core/write/batch_write_collation_estsize.js
new file mode 100644
index 00000000000..bfaf1e54bd3
--- /dev/null
+++ b/jstests/core/write/batch_write_collation_estsize.js
@@ -0,0 +1,116 @@
+// Cannot implicitly shard accessed collections because of following errmsg: A single
+// update/delete on a sharded collection must contain an exact match on _id or contain the shard
+// key.
+// @tags: [assumes_unsharded_collection, requires_multi_updates, requires_non_retryable_writes]
+
+// Tests that the update and delete batch write operations account for the size of the collation
+// specification in the write operation document.
+
+(function() {
+"use strict";
+
+// Setup the test collection.
+db.batch_write_collation_estsize.drop();
+assert.commandWorked(db.batch_write_collation_estsize.insert({str: "FOO"}));
+
+// Test updateOne bulk write operation with collation specification.
+let res = db.batch_write_collation_estsize.bulkWrite([{
+    updateOne: {
+        filter: {str: "FOO"},
+        update: {$set: {str: "BAR"}},
+        collation: {
+            locale: "en_US",
+            caseLevel: false,
+            caseFirst: "off",
+            strength: 3,
+            numericOrdering: false,
+            alternate: "non-ignorable",
+            maxVariable: "punct",
+            normalization: false,
+            backwards: false
+        }
+    }
+}]);
+assert.eq(1, res.matchedCount);
+
+// Test updateMany bulk write operation with collation specification.
+res = db.batch_write_collation_estsize.bulkWrite([{
+    updateMany: {
+        filter: {str: "BAR"},
+        update: {$set: {str: "FOO"}},
+        collation: {
+            locale: "en_US",
+            caseLevel: false,
+            caseFirst: "off",
+            strength: 3,
+            numericOrdering: false,
+            alternate: "non-ignorable",
+            maxVariable: "punct",
+            normalization: false,
+            backwards: false
+        }
+    }
+}]);
+assert.eq(1, res.matchedCount);
+
+// Test replaceOne bulk write operation with collation specification.
+res = db.batch_write_collation_estsize.bulkWrite([{
+    replaceOne: {
+        filter: {str: "FOO"},
+        replacement: {str: "BAR"},
+        collation: {
+            locale: "en_US",
+            caseLevel: false,
+            caseFirst: "off",
+            strength: 3,
+            numericOrdering: false,
+            alternate: "non-ignorable",
+            maxVariable: "punct",
+            normalization: false,
+            backwards: false
+        }
+    }
+}]);
+assert.eq(1, res.matchedCount);
+
+// Test deleteMany bulk write operation with collation specification.
+res = db.batch_write_collation_estsize.bulkWrite([{
+    deleteOne: {
+        filter: {str: "BAR"},
+        collation: {
+            locale: "en_US",
+            caseLevel: false,
+            caseFirst: "off",
+            strength: 3,
+            numericOrdering: false,
+            alternate: "non-ignorable",
+            maxVariable: "punct",
+            normalization: false,
+            backwards: false
+        }
+    }
+}]);
+assert.eq(1, res.deletedCount);
+
+// Reinsert a document to test deleteMany bulk write operation.
+assert.commandWorked(db.batch_write_collation_estsize.insert({str: "FOO"}));
+
+// Test deleteMany bulk write operation with collation specification.
+res = db.batch_write_collation_estsize.bulkWrite([{
+    deleteMany: {
+        filter: {str: "FOO"},
+        collation: {
+            locale: "en_US",
+            caseLevel: false,
+            caseFirst: "off",
+            strength: 3,
+            numericOrdering: false,
+            alternate: "non-ignorable",
+            maxVariable: "punct",
+            normalization: false,
+            backwards: false
+        }
+    }
+}]);
+assert.eq(1, res.deletedCount);
+})();