chore: fix sinon

Author: nbbeeken

test/integration/client-side-encryption/driver.test.ts

@@ -3,16 +3,15 @@ import { expect } from 'chai';
 import * as crypto from 'crypto';
 import * as sinon from 'sinon';
 
-// eslint-disable-next-line @typescript-eslint/no-restricted-imports
-import { StateMachine } from '../../../lib/client-side-encryption/state_machine';
 // eslint-disable-next-line @typescript-eslint/no-restricted-imports
 import { ClientEncryption } from '../../../src/client-side-encryption/client_encryption';
 import {
   BSON,
   type Collection,
   type CommandStartedEvent,
   type MongoClient,
-  MongoOperationTimeoutError
+  MongoOperationTimeoutError,
+  StateMachine
 } from '../../mongodb';
 import { type FailPoint, getEncryptExtraOptions } from '../../tools/utils';
 
@@ -479,16 +478,19 @@ describe('Client Side Encryption Functional', function () {
     });
 
     afterEach(async function () {
-      await client.close();
+      await client?.close();
     });
 
     describe('rewrapManyDataKey', function () {
       makeBlockingFailFor('update', 2000);
 
       beforeEach(async function () {
-        sinon.stub(StateMachine.prototype, 'execute').callsFake(async function () {
-          return BSON.serialize({ v: [{ _id: new UUID() }] });
-        });
+        sinon
+          .stub(StateMachine.prototype, 'execute')
+          .callsFake(async function (executor, context, timeoutContext) {
+            expect(timeoutContext).to.have.property('timeoutMS', 500);
+            return BSON.serialize({ v: [{ _id: new UUID() }] });
+          });
       });
 
       afterEach(async function () {

test/mongodb.ts

@@ -103,6 +103,16 @@ export * from '../src/bulk/common';
 export * from '../src/bulk/ordered';
 export * from '../src/bulk/unordered';
 export * from '../src/change_stream';
+export * from '../src/client-side-encryption/auto_encrypter';
+export * from '../src/client-side-encryption/client_encryption';
+export * from '../src/client-side-encryption/crypto_callbacks';
+export * from '../src/client-side-encryption/errors';
+export * from '../src/client-side-encryption/mongocryptd_manager';
+export * from '../src/client-side-encryption/providers/aws';
+export * from '../src/client-side-encryption/providers/azure';
+export * from '../src/client-side-encryption/providers/gcp';
+export * from '../src/client-side-encryption/providers/index';
+export * from '../src/client-side-encryption/state_machine';
 export * from '../src/cmap/auth/auth_provider';
 export * from '../src/cmap/auth/aws_temporary_credentials';
 export * from '../src/cmap/auth/gssapi';