types(connection): make transaction() return type match the executor …

…function Fix #14656
Automattic · Jun 10, 2024 · 7bfd2bc · 7bfd2bc
1 parent 7bc5395
commit 7bfd2bc
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 5 deletions.
diff --git a/test/docs/transactions.test.js b/test/docs/transactions.test.js
@@ -479,7 +479,7 @@ describe('transactions', function() {
     const doc = await Test.findById(_id).orFail();
     let attempt = 0;
 
-    await db.transaction(async(session) => {
+    const res = await db.transaction(async(session) => {
       await doc.save({ session });
 
       if (attempt === 0) {
@@ -489,7 +489,10 @@ describe('transactions', function() {
           errorLabels: [mongoose.mongo.MongoErrorLabel.TransientTransactionError]
         });
       }
+
+      return { answer: 42 };
     });
+    assert.deepStrictEqual(res, { answer: 42 });
 
     const { items } = await Test.findById(_id).orFail();
     assert.ok(Array.isArray(items));

diff --git a/test/types/connection.test.ts b/test/types/connection.test.ts
@@ -45,11 +45,11 @@ expectType<mongodb.Db>(conn.db);
 expectType<mongodb.MongoClient>(conn.getClient());
 expectType<Connection>(conn.setClient(new mongodb.MongoClient('mongodb://127.0.0.1:27017/test')));
 
-expectType<Promise<void>>(conn.transaction(async(res) => {
+expectType<Promise<string>>(conn.transaction(async(res) => {
   expectType<mongodb.ClientSession>(res);
   return 'a';
 }));
-expectType<Promise<void>>(conn.transaction(async(res) => {
+expectType<Promise<string>>(conn.transaction(async(res) => {
   expectType<mongodb.ClientSession>(res);
   return 'a';
 }, { readConcern: 'majority' }));
@@ -148,4 +148,4 @@ function schemaInstanceMethodsAndQueryHelpersOnConnection() {
   });
 
   const TestModel = connection.model<User, UserModel, UserQueryHelpers>('User', userSchema);
-}
+}
diff --git a/types/connection.d.ts b/types/connection.d.ts
@@ -236,7 +236,7 @@ declare module 'mongoose' {
      * async function executes successfully and attempt to retry if
      * there was a retryable error.
      */
-    transaction(fn: (session: mongodb.ClientSession) => Promise<any>, options?: mongodb.TransactionOptions): Promise<void>;
+    transaction<ReturnType = unknown>(fn: (session: mongodb.ClientSession) => Promise<ReturnType>, options?: mongodb.TransactionOptions): Promise<ReturnType>;
 
     /** Switches to a different database using the same connection pool. */
     useDb(name: string, options?: { useCache?: boolean, noListener?: boolean }): Connection;