Merge pull request #14743 from Automattic/vkarpov15/gh-14738

fix(model): handle `transactionAsyncLocalStorage` option with `insertMany()`
Automattic · Jul 12, 2024 · 70e8e19 · 70e8e19
2 parents 090297e + 3646952
commit 70e8e19
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/lib/model.js b/lib/model.js
@@ -2838,6 +2838,11 @@ Model.$__insertMany = function(arr, options, callback) {
   const throwOnValidationError = typeof options.throwOnValidationError === 'boolean' ? options.throwOnValidationError : false;
   const lean = !!options.lean;
 
+  const asyncLocalStorage = this.db.base.transactionAsyncLocalStorage?.getStore();
+  if ((!options || !options.hasOwnProperty('session')) && asyncLocalStorage?.session != null) {
+    options = { ...options, session: asyncLocalStorage.session };
+  }
+
   if (!Array.isArray(arr)) {
     arr = [arr];
   }

diff --git a/test/docs/transactions.test.js b/test/docs/transactions.test.js
@@ -388,6 +388,8 @@ describe('transactions', function() {
           }();
           assert.equal(doc.name, 'test_transactionAsyncLocalStorage');
 
+          await Test.insertMany([{ name: 'bar' }]);
+
           throw new Error('Oops!');
         }),
         /Oops!/
@@ -397,6 +399,9 @@ describe('transactions', function() {
 
       exists = await Test.exists({ name: 'foo' });
       assert.ok(!exists);
+
+      exists = await Test.exists({ name: 'bar' });
+      assert.ok(!exists);
     });
   });