From b7b4e8b640622dbe9f258cb9e4153d4fd570b000 Mon Sep 17 00:00:00 2001
From: Warren James <warren.james@mongodb.com>
Date: Wed, 25 Sep 2024 16:46:03 -0400
Subject: [PATCH] feat: allow clientBulkWrite to use TimeoutContext

---
 .../client_bulk_write/client_bulk_write.ts           | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/src/operations/client_bulk_write/client_bulk_write.ts b/src/operations/client_bulk_write/client_bulk_write.ts
index cb020bde40c..b7e6529c3fd 100644
--- a/src/operations/client_bulk_write/client_bulk_write.ts
+++ b/src/operations/client_bulk_write/client_bulk_write.ts
@@ -3,6 +3,7 @@ import { type Document } from 'bson';
 import { ClientBulkWriteCursorResponse } from '../../cmap/wire_protocol/responses';
 import type { Server } from '../../sdam/server';
 import type { ClientSession } from '../../sessions';
+import { type TimeoutContext } from '../../timeout';
 import { MongoDBNamespace } from '../../utils';
 import { CommandOperation } from '../command';
 import { Aspect, defineAspects } from '../operation';
@@ -35,9 +36,16 @@ export class ClientBulkWriteOperation extends CommandOperation<ClientBulkWriteCu
    */
   override async execute(
     server: Server,
-    session: ClientSession | undefined
+    session: ClientSession | undefined,
+    timeoutContext: TimeoutContext
   ): Promise<ClientBulkWriteCursorResponse> {
-    return await super.executeCommand(server, session, this.command, ClientBulkWriteCursorResponse);
+    return await super.executeCommand(
+      server,
+      session,
+      this.command,
+      timeoutContext,
+      ClientBulkWriteCursorResponse
+    );
   }
 }