From 871d20853f02e560d1a5d4410220d018ccf94a67 Mon Sep 17 00:00:00 2001
From: Neal Beeken <neal.beeken@mongodb.com>
Date: Wed, 5 Apr 2023 13:21:56 -0400
Subject: [PATCH] fix: lb mode error

---
 .../mongodb-handshake/mongodb-handshake.test.ts     | 13 +++++++++++--
 test/unit/cmap/handshake/client_metadata.test.ts    |  2 +-
 2 files changed, 12 insertions(+), 3 deletions(-)

diff --git a/test/integration/mongodb-handshake/mongodb-handshake.test.ts b/test/integration/mongodb-handshake/mongodb-handshake.test.ts
index 366a0eb0d4..c6636a2298 100644
--- 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
index d337744cd0..286af2f182 100644
--- 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'],