feat(NODE-6390): Add timeoutMS support to auto encryption (#4265)

Co-authored-by: Warren James <warren.james@mongodb.com> Co-authored-by: Neal Beeken <neal.beeken@mongodb.com> Co-authored-by: Bailey Pearson <bailey.pearson@mongodb.com>
mongodb · Nov 6, 2024 · 55e08e7 · 55e08e7
1 parent b48b7af
commit 55e08e7
Show file tree

Hide file tree

Showing 6 changed files with 491 additions and 20 deletions.
diff --git a/src/client-side-encryption/auto_encrypter.ts b/src/client-side-encryption/auto_encrypter.ts
@@ -395,7 +395,7 @@ export class AutoEncrypter {
       socketOptions: autoSelectSocketOptions(this._client.options)
     });
 
-    return deserialize(await stateMachine.execute(this, context), {
+    return deserialize(await stateMachine.execute(this, context, options.timeoutContext), {
       promoteValues: false,
       promoteLongs: false
     });
@@ -416,7 +416,11 @@ export class AutoEncrypter {
       socketOptions: autoSelectSocketOptions(this._client.options)
     });
 
-    return await stateMachine.execute(this, context);
+    return await stateMachine.execute(
+      this,
+      context,
+      options.timeoutContext?.csotEnabled() ? options.timeoutContext : undefined
+    );
   }
 
   /**