test(NODE-7169): Migrate node-mongodb-native/test/unit tests (#4676)

Co-authored-by: Bailey Pearson <bailey.pearson@gmail.com>

Author: PavelSafronov

test/unit/assorted/polling_srv_records_for_mongos_discovery.prose.test.ts

@@ -98,12 +98,9 @@ describe('Polling Srv Records for Mongos Discovery', () => {
       return mock.cleanup();
     });
 
-    afterEach(function (done) {
+    afterEach(function () {
       if (context.topology) {
         context.topology.close();
-        done();
-      } else {
-        done();
       }
     });
 

test/unit/bson.test.ts

@@ -1,7 +1,6 @@
+import { BSON } from 'bson';
 import { expect } from 'chai';
 
-import { BSON } from '../mongodb';
-
 describe('When importing BSON', function () {
   const types = [
     ['Long', 23],

test/unit/change_stream.test.ts

@@ -2,7 +2,9 @@ import { Long, Timestamp } from 'bson';
 import { expect } from 'chai';
 import * as sinon from 'sinon';
 
-import { ChangeStreamCursor, MongoClient, MongoDBNamespace } from '../mongodb';
+import { ChangeStreamCursor } from '../../src/cursor/change_stream_cursor';
+import { MongoClient } from '../../src/mongo_client';
+import { MongoDBNamespace } from '../../src/utils';
 
 describe('ChangeStreamCursor', function () {
   afterEach(function () {

test/unit/client-side-encryption/auto_encrypter.test.ts

@@ -1,13 +1,14 @@
+import { BSON } from 'bson';
 import { expect } from 'chai';
 import * as fs from 'fs';
 import * as net from 'net';
 import * as sinon from 'sinon';
 
 import { AutoEncrypter } from '../../../src/client-side-encryption/auto_encrypter';
+import { type DataKey } from '../../../src/client-side-encryption/client_encryption';
 import { MongocryptdManager } from '../../../src/client-side-encryption/mongocryptd_manager';
 import { StateMachine } from '../../../src/client-side-encryption/state_machine';
 import { MongoClient } from '../../../src/mongo_client';
-import { BSON, type DataKey } from '../../mongodb';
 import * as requirements from './requirements.helper';
 
 const bson = BSON;

test/unit/client-side-encryption/client_encryption.test.ts

@@ -1,3 +1,4 @@
+import { Binary, BSON, deserialize } from 'bson';
 import { expect } from 'chai';
 import * as fs from 'fs';
 import { resolve } from 'path';
@@ -10,7 +11,7 @@ import {
   MongoCryptCreateEncryptedCollectionError
 } from '../../../src/client-side-encryption/errors';
 import { StateMachine } from '../../../src/client-side-encryption/state_machine';
-import { Binary, BSON, deserialize, MongoClient } from '../../mongodb';
+import { MongoClient } from '../../../src/mongo_client';
 
 const { EJSON } = BSON;
 

test/unit/client-side-encryption/errors.test.ts

@@ -7,7 +7,7 @@ import {
   MongoCryptError,
   MongoCryptInvalidArgumentError
 } from '../../../src/client-side-encryption/errors';
-import { MongoError } from '../../mongodb';
+import { MongoError } from '../../../src/error';
 
 describe('MongoCryptError', function () {
   const errors = [

test/unit/client-side-encryption/providers/credentialsProvider.test.ts

@@ -13,8 +13,8 @@ import {
   tokenCache
 } from '../../../../src/client-side-encryption/providers/azure';
 import { AWSSDKCredentialProvider } from '../../../../src/cmap/auth/aws_temporary_credentials';
+import { MongoNetworkTimeoutError } from '../../../../src/error';
 import * as utils from '../../../../src/utils';
-import { MongoNetworkTimeoutError } from '../../../mongodb';
 import * as requirements from '../requirements.helper';
 
 const originalAccessKeyId = process.env.AWS_ACCESS_KEY_ID;

test/unit/client-side-encryption/state_machine.test.ts

@@ -1,27 +1,21 @@
+import { BSON, Int32, Long, serialize } from 'bson';
 import { expect } from 'chai';
 import { EventEmitter, once } from 'events';
 import * as fs from 'fs/promises';
 import { type MongoCryptKMSRequest } from 'mongodb-client-encryption';
 import * as net from 'net';
 import * as sinon from 'sinon';
-import { setTimeout } from 'timers';
 import { setTimeout as setTimeoutAsync } from 'timers/promises';
 import * as tls from 'tls';
 
 import { StateMachine } from '../../../src/client-side-encryption/state_machine';
+import { Collection } from '../../../src/collection';
+import { CursorTimeoutContext } from '../../../src/cursor/abstract_cursor';
 import { Db } from '../../../src/db';
-import {
-  BSON,
-  Collection,
-  CSOTTimeoutContext,
-  CursorTimeoutContext,
-  type FindOptions,
-  Int32,
-  Long,
-  MongoClient,
-  serialize,
-  squashError
-} from '../../mongodb';
+import { MongoClient } from '../../../src/mongo_client';
+import { type FindOptions } from '../../../src/operations/find';
+import { CSOTTimeoutContext } from '../../../src/timeout';
+import { squashError } from '../../../src/utils';
 import { sleep } from '../../tools/utils';
 
 describe('StateMachine', function () {
@@ -129,7 +123,7 @@ describe('StateMachine', function () {
         });
       });
 
-      it('should only resolve once bytesNeeded drops to zero', function (done) {
+      it('should only resolve once bytesNeeded drops to zero', async function () {
         const stateMachine = new StateMachine({} as any);
         const request = new MockRequest(Buffer.from('foobar'), 500);
         let status = 'pending';
@@ -143,22 +137,21 @@ describe('StateMachine', function () {
           .catch(() => {});
 
         this.fakeSocket.emit('connect');
-        setTimeout(() => {
-          expect(status).to.equal('pending');
-          expect(request.bytesNeeded).to.equal(500);
-          expect(request.kmsProvider).to.equal('aws');
-          this.fakeSocket.emit('data', Buffer.alloc(300));
-          setTimeout(() => {
-            expect(status).to.equal('pending');
-            expect(request.bytesNeeded).to.equal(200);
-            this.fakeSocket.emit('data', Buffer.alloc(200));
-            setTimeout(() => {
-              expect(status).to.equal('resolved');
-              expect(request.bytesNeeded).to.equal(0);
-              done();
-            });
-          });
-        });
+        await sleep();
+
+        expect(status).to.equal('pending');
+        expect(request.bytesNeeded).to.equal(500);
+        expect(request.kmsProvider).to.equal('aws');
+        this.fakeSocket.emit('data', Buffer.alloc(300));
+        await sleep();
+
+        expect(status).to.equal('pending');
+        expect(request.bytesNeeded).to.equal(200);
+        this.fakeSocket.emit('data', Buffer.alloc(200));
+        await sleep();
+
+        expect(status).to.equal('resolved');
+        expect(request.bytesNeeded).to.equal(0);
       });
     });
 
@@ -196,13 +189,9 @@ describe('StateMachine', function () {
               } as any);
               const request = new MockRequest(Buffer.from('foobar'), 500);
 
-              it('rejects with the validation error', function (done) {
-                stateMachine.kmsRequest(request).catch(err => {
-                  expect(err.message).to.equal(
-                    `Insecure TLS options prohibited for aws: ${option}`
-                  );
-                  done();
-                });
+              it('rejects with the validation error', async function () {
+                const err = await stateMachine.kmsRequest(request).catch(e => e);
+                expect(err.message).to.equal(`Insecure TLS options prohibited for aws: ${option}`);
               });
             });
           }

test/unit/cmap/auth/auth_provider.test.ts

@@ -1,6 +1,7 @@
 import { expect } from 'chai';
 
-import { type AuthContext, AuthProvider, MongoRuntimeError } from '../../../mongodb';
+import { type AuthContext, AuthProvider } from '../../../../src/cmap/auth/auth_provider';
+import { MongoRuntimeError } from '../../../../src/error';
 
 describe('AuthProvider', function () {
   describe('#reauth', function () {

test/unit/cmap/auth/gssapi.test.ts

@@ -6,7 +6,7 @@ import {
   GSSAPICanonicalizationValue,
   performGSSAPICanonicalizeHostName,
   resolveCname
-} from '../../../mongodb';
+} from '../../../../src/cmap/auth/gssapi';
 
 describe('GSSAPI', () => {
   let lookupSpy;