chore: move to assertion errors

Author: durran

test/tools/unified-spec-runner/entity_event_registry.ts

@@ -1,3 +1,5 @@
+import { AssertionError } from 'chai';
+
 import {
   COMMAND_FAILED,
   COMMAND_STARTED,
@@ -61,7 +63,7 @@ export class EntityEventRegistry {
     if (this.clientEntity.storeEventsAsEntities) {
       for (const { id, events } of this.clientEntity.storeEventsAsEntities) {
         if (this.entitiesMap.has(id) || this.clientEntity.id === id) {
-          throw new Error(`Duplicate id ${id} found while storing events as entities`);
+          throw new AssertionError(`Duplicate id ${id} found while storing events as entities`);
         }
         this.entitiesMap.set(id, []);
         for (const eventName of events) {

test/tools/unified-spec-runner/operations.ts

@@ -44,7 +44,7 @@ type RunOperationFn = (
 ) => Promise<Document | boolean | number | null | void | string>;
 export const operations = new Map<string, RunOperationFn>();
 
-export class MalformedOperationError extends Error {}
+export class MalformedOperationError extends AssertionError {}
 
 operations.set('createEntities', async ({ entities, operation, testConfig }) => {
   if (!operation.arguments?.entities) {

test/tools/unified-spec-runner/runner.ts

@@ -1,9 +1,15 @@
 /* eslint-disable @typescript-eslint/no-non-null-assertion */
-import { expect } from 'chai';
+import { AssertionError, expect } from 'chai';
 import { gte as semverGte, satisfies as semverSatisfies } from 'semver';
 
 import type { MongoClient } from '../../mongodb';
-import { MONGODB_ERROR_CODES, ns, ReadPreference, TopologyType } from '../../mongodb';
+import {
+  MONGODB_ERROR_CODES,
+  MongoServerError,
+  ns,
+  ReadPreference,
+  TopologyType
+} from '../../mongodb';
 import { ejson } from '../utils';
 import { AstrolabeResultsWriter } from './astrolabe_results_writer';
 import { EntitiesMap, type UnifiedMongoClient } from './entities';
@@ -317,7 +323,9 @@ export function runUnifiedSuite(
             const error = await runUnifiedTest(this, unifiedSuite, test, skipFilter).catch(
               error => error
             );
-            expect(error).to.exist;
+            expect(error).to.satisfy(value => {
+              return value instanceof AssertionError || value instanceof MongoServerError;
+            });
           } else {
             await runUnifiedTest(this, unifiedSuite, test, skipFilter);
           }

test/tools/unified-spec-runner/unified-utils.ts

@@ -1,5 +1,5 @@
 import { EJSON } from 'bson';
-import { expect } from 'chai';
+import { AssertionError, expect } from 'chai';
 import ConnectionString from 'mongodb-connection-string-url';
 import { gte as semverGte, lte as semverLte } from 'semver';
 import { isDeepStrictEqual } from 'util';
@@ -351,7 +351,9 @@ export function mergeKMSProviders(
         : test['sessionToken']);
 
     if (!awsProviders['accessKeyId'] || !awsProviders['secretAccessKey']) {
-      throw new Error('AWS KMS providers must constain "accessKeyId" and "secretAccessKey"');
+      throw new AssertionError(
+        'AWS KMS providers must constain "accessKeyId" and "secretAccessKey"'
+      );
     }
 
     return awsProviders;
@@ -385,7 +387,7 @@ export function mergeKMSProviders(
       !azureProviders['clientId'] ||
       !azureProviders['clientSecret']
     ) {
-      throw new Error(
+      throw new AssertionError(
         'Azure KMS providers must contain "tenantId", "clientId", and "clientSecret"'
       );
     }
@@ -409,7 +411,7 @@ export function mergeKMSProviders(
         : test['endPoint']);
 
     if (!gcpProviders['email'] || !gcpProviders['privateKey']) {
-      throw new Error('GCP KMS providers must contain "email" and "privateKey"');
+      throw new AssertionError('GCP KMS providers must contain "email" and "privateKey"');
     }
 
     return gcpProviders;
@@ -514,7 +516,7 @@ export function mergeKMSProviders(
   }
 
   if (Object.keys(providers).length === 0) {
-    throw new Error('Found empty KMS providers in test');
+    throw new AssertionError('Found empty KMS providers in test');
   }
 
   return providers;