Use provided axios instance for credentials refresh operations (#193)

openfga · Jan 8, 2025 · 24f8188 · 24f8188
2 parents 006b5c4 + 691afea
commit 24f8188
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/base.ts b/base.ts
@@ -48,7 +48,7 @@ export class BaseAPI {
     }
     this.configuration.isValid();
 
-    this.credentials = Credentials.init(this.configuration);
+    this.credentials = Credentials.init(this.configuration, this.axios);
 
     if (!this.axios) {
       const httpAgent = new http.Agent({ keepAlive: true });

diff --git a/credentials/credentials.ts b/credentials/credentials.ts
@@ -26,9 +26,9 @@ export class Credentials {
   private accessToken?: string;
   private accessTokenExpiryDate?: Date;
 
-  public static init(configuration: { credentials: AuthCredentialsConfig, telemetry: TelemetryConfiguration, baseOptions?: any }): Credentials {
-    return new Credentials(configuration.credentials, globalAxios, configuration.telemetry, configuration.baseOptions);
-  }
+  public static init(configuration: { credentials: AuthCredentialsConfig, telemetry: TelemetryConfiguration, baseOptions?: any }, axios: AxiosInstance = globalAxios): Credentials {
+    return new Credentials(configuration.credentials, axios, configuration.telemetry, configuration.baseOptions);
+}
 
   public constructor(private authConfig: AuthCredentialsConfig, private axios: AxiosInstance = globalAxios, private telemetryConfig: TelemetryConfiguration, private baseOptions?: any) {
     this.initConfig();
@@ -155,7 +155,7 @@ export class Credentials {
       }, {
         maxRetry: 3,
         minWaitInMs: 100,
-      }, globalAxios);
+      }, this.axios);
 
       const response = wrappedResponse?.response;
       if (response) {
@@ -180,7 +180,7 @@ export class Credentials {
 
         attributes = TelemetryAttributes.fromResponse({
           response,
-          attributes,  
+          attributes,
         });
 
         attributes = TelemetryAttributes.prepare(attributes, this.telemetryConfig.metrics?.counterCredentialsRequest?.attributes);