fix tests

Automattic · May 13, 2024 · 3c74f46 · 3c74f46
1 parent f498cc5
commit 3c74f46
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/lib/aggregate.js b/lib/aggregate.js
@@ -1022,7 +1022,7 @@ Aggregate.prototype.exec = async function exec() {
   applyGlobalMaxTimeMS(this.options, model.db.options, model.base.options);
   applyGlobalDiskUse(this.options, model.db.options, model.base.options);
 
-  const asyncLocalStorage = this.model?.db?.base.transactionAsyncLocalStorage?.getStore();
+  const asyncLocalStorage = this.model()?.db?.base.transactionAsyncLocalStorage?.getStore();
   if (!this.options.hasOwnProperty('session') && asyncLocalStorage?.session != null) {
     this.options.session = asyncLocalStorage.session;
   }

diff --git a/test/docs/transactions.test.js b/test/docs/transactions.test.js
@@ -370,7 +370,7 @@ describe('transactions', function() {
       await Test.createCollection();
       await Test.deleteMany({});
 
-      const doc = new Test({ name: 'test' });
+      const doc = new Test({ name: 'test_transactionAsyncLocalStorage' });
       await assert.rejects(
         () => m.connection.transaction(async() => {
           await doc.save();
@@ -383,10 +383,10 @@ describe('transactions', function() {
           docs = await Test.find({ _id: doc._id });
           assert.equal(docs.length, 1);
 
-          docs = await Promise.all([async() => {
+          docs = await async function test() {
             return await Test.findOne({ _id: doc._id });
-          }]).then(res => res[0]);
-          assert.equal(docs.length, 1);
+          }();
+          assert.equal(doc.name, 'test_transactionAsyncLocalStorage');
 
           throw new Error('Oops!');
         }),