fix: lb mode error

mongodb · Apr 5, 2023 · 871d208 · 871d208
1 parent 70da1d6
commit 871d208
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 3 deletions.
diff --git a/test/integration/mongodb-handshake/mongodb-handshake.test.ts b/test/integration/mongodb-handshake/mongodb-handshake.test.ts
@@ -1,7 +1,12 @@
 import { expect } from 'chai';
 import * as sinon from 'sinon';
 
-import { Connection, LEGACY_HELLO_COMMAND, MongoServerSelectionError } from '../../mongodb';
+import {
+  Connection,
+  LEGACY_HELLO_COMMAND,
+  MongoServerError,
+  MongoServerSelectionError
+} from '../../mongodb';
 
 describe('MongoDB Handshake Node tests', () => {
   let client;
@@ -29,7 +34,11 @@ describe('MongoDB Handshake Node tests', () => {
     it('client fails to connect with an error relating to size', async function () {
       client = this.configuration.newClient({ serverSelectionTimeoutMS: 2000 });
       const error = await client.connect().catch(error => error);
-      expect(error).to.be.instanceOf(MongoServerSelectionError);
+      if (this.configuration.isLoadBalanced) {
+        expect(error).to.be.instanceOf(MongoServerError);
+      } else {
+        expect(error).to.be.instanceOf(MongoServerSelectionError);
+      }
       expect(error).to.match(/client metadata document must be less/);
     });
   });

diff --git a/test/unit/cmap/handshake/client_metadata.test.ts b/test/unit/cmap/handshake/client_metadata.test.ts
@@ -9,7 +9,7 @@ import { getFAASEnv, Int32, makeClientMetadata } from '../../../mongodb';
 const NODE_DRIVER_VERSION = require('../../../../package.json').version;
 
 describe('client metadata module', () => {
-  describe('determineCloudProvider()', function () {
+  describe('getFAASEnv()', function () {
     const tests: Array<[string, string]> = [
       ['AWS_EXECUTION_ENV', 'aws.lambda'],
       ['AWS_LAMBDA_RUNTIME_API', 'aws.lambda'],