test: fix kms test

mongodb · May 6, 2024 · 8da6c4a · 8da6c4a
1 parent 920a61e
commit 8da6c4a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/...tion/client-side-encryption/client_side_encryption.prose.18.azure_kms_mock_server.test.ts b/...tion/client-side-encryption/client_side_encryption.prose.18.azure_kms_mock_server.test.ts
@@ -7,7 +7,7 @@ import {
   type AzureKMSRequestOptions,
   fetchAzureKMSToken
 } from '../../../src/client-side-encryption/providers/azure';
-import { type Document } from '../../mongodb';
+import { type Document, MongoNetworkTimeoutError } from '../../mongodb';
 
 const BASE_URL = new URL(`http://127.0.0.1:8080/metadata/identity/oauth2/token`);
 class KMSRequestOptions implements AzureKMSRequestOptions {
@@ -119,7 +119,7 @@ context('Azure KMS Mock Server Tests', function () {
     it('returns an error after the request times out', async () => {
       const error = await fetchAzureKMSToken(new KMSRequestOptions('slow')).catch(e => e);
 
-      expect(error).to.be.instanceof(MongoCryptAzureKMSRequestError);
+      expect(error).to.be.instanceof(MongoNetworkTimeoutError);
     });
   });
 });