fix: ManagementTokenProviders should also respect the keepAlive confi…

…g option
auth0 · Sep 7, 2023 · a63f8d8 · a63f8d8
1 parent d10c97a
commit a63f8d8
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/src/management/ManagementTokenProvider.js b/src/management/ManagementTokenProvider.js
@@ -20,6 +20,7 @@ class ManagementTokenProvider {
    * @param {number}  [options.cacheTTLInSeconds]     By default the `expires_in` value will be used to determine the cached time of the token, this can be overridden.
    * @param {object}  [options.headers]               Additional headers that will be added to the outgoing requests.
    * @param {string}  [options.proxy]                 Add the `superagent-proxy` dependency and specify a proxy url eg 'https://myproxy.com:1234'
+   * @param {boolean}  [options.keepAlive]          Keep the http connections alive.
    */
   constructor(options) {
     if (!options || typeof options !== 'object') {
@@ -76,6 +77,7 @@ class ManagementTokenProvider {
       clientInfo: this.options.clientInfo,
       headers: this.options.headers,
       proxy: this.options.proxy,
+      keepAlive: this.options.keepAlive,
     };
     this.authenticationClient = new AuthenticationClient(authenticationClientOptions);
 

diff --git a/test/management/management-token-provider.tests.js b/test/management/management-token-provider.tests.js
@@ -154,6 +154,20 @@ describe('ManagementTokenProvider', () => {
     expect(provider.options.headers).to.be.equal(options.headers);
   });
 
+  it('should send keepAlive true value to authentication client when passed into options', () => {
+    const options = Object.assign({}, defaultOptions);
+    options.keepAlive = true;
+    const provider = new ManagementTokenProvider(options);
+    expect(provider.authenticationClient.oauth.oauth.options.keepAlive).to.be.true;
+  });
+
+  it('should send keepAlive false value to authentication client when passed into options', () => {
+    const options = Object.assign({}, defaultOptions);
+    options.keepAlive = false;
+    const provider = new ManagementTokenProvider(options);
+    expect(provider.authenticationClient.oauth.oauth.options.keepAlive).to.be.false;
+  });
+
   it('should handle network errors correctly', async () => {
     const options = Object.assign({}, defaultOptions);
     options.domain = 'domain';