fix: test

Author: durran

test/mongodb.ts

@@ -119,6 +119,7 @@ export * from '../src/cmap/auth/gssapi';
 export * from '../src/cmap/auth/mongo_credentials';
 export * from '../src/cmap/auth/mongodb_aws';
 export * from '../src/cmap/auth/mongodb_oidc';
+export * from '../src/cmap/auth/mongodb_oidc/automated_callback_workflow';
 export * from '../src/cmap/auth/mongodb_oidc/azure_machine_workflow';
 export * from '../src/cmap/auth/mongodb_oidc/callback_workflow';
 export * from '../src/cmap/auth/plain';

test/unit/cmap/auth/mongodb_oidc/automated_callback_workflow.test.ts

@@ -1,31 +1,38 @@
 import { expect } from 'chai';
 import * as sinon from 'sinon';
 
-// eslint-disable-next-line @typescript-eslint/no-restricted-imports
-import { AutomatedCallbackWorkflow } from '../../../../../src/cmap/auth/mongodb_oidc/automated_callback_workflow';
 // eslint-disable-next-line @typescript-eslint/no-restricted-imports
 import { callback } from '../../../../../src/cmap/auth/mongodb_oidc/gcp_machine_workflow';
 // eslint-disable-next-line @typescript-eslint/no-restricted-imports
 import { TokenCache } from '../../../../../src/cmap/auth/mongodb_oidc/token_cache';
-import { Connection, MongoCredentials } from '../../../../mongodb';
+import {
+  AutomatedCallbackWorkflow,
+  CallbackWorkflow,
+  Connection,
+  MongoCredentials
+} from '../../../../mongodb';
 
 describe('AutomatedCallbackWorkflow', function () {
   describe('#execute', function () {
     context('when the cache has a token', function () {
+      const sandbox = sinon.createSandbox();
+
+      // See NODE-6801 and corresponding PR: https://github.com/mongodb/node-mongodb-native/pull/4438
+      // This is a test to ensure that we do not regress on the above issue. Do NOT remove this test.
       context('when the connection has no token', function () {
         const cache = new TokenCache();
-        const connection = sinon.createStubInstance(Connection);
-        const credentials = sinon.createStubInstance(MongoCredentials);
+        const connection = sandbox.createStubInstance(Connection);
+        const credentials = sandbox.createStubInstance(MongoCredentials);
+        sandbox.stub(CallbackWorkflow.prototype, 'finishAuthentication').resolves();
         const workflow = new AutomatedCallbackWorkflow(cache, callback);
-        sinon.stub(workflow, 'finishAuthentication').returns(Promise.resolve());
 
         beforeEach(function () {
           cache.put({ accessToken: 'test', expiresInSeconds: 7200 });
           workflow.execute(connection, credentials);
         });
 
         afterEach(function () {
-          sinon.restore();
+          sandbox.restore();
         });
 
         it('sets the token on the connection', async function () {