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

Co-authored-by: Warren James <[email protected]> Co-authored-by: Neal Beeken <[email protected]> Co-authored-by: Bailey Pearson <[email protected]>
mongodb · Nov 5, 2024 · 3054eaf · 3054eaf
1 parent 18b3d31
commit 3054eaf
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
+    );
   }
 
   /**