fix: allow to override endpoint for OperationService

yandex-cloud · Mar 9, 2023 · 20091c9 · 20091c9
1 parent c8c3af8
commit 20091c9
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/src/utils/operation/wait-for.ts b/src/utils/operation/wait-for.ts
@@ -9,8 +9,13 @@ const { operation: { operation_service: { GetOperationRequest } } } = cloudApi;
 
 const DEFAULT_TIMEOUT_MS = 10 * 60 * 1000; // 10 minutes
 
-export const waitForOperation = (op: Operation, session: Session, timeoutMs: number = DEFAULT_TIMEOUT_MS): Promise<Operation> => {
-    const client = session.client(serviceClients.OperationServiceClient);
+export const waitForOperation = (
+    op: Operation,
+    session: Session,
+    timeoutMs: number = DEFAULT_TIMEOUT_MS,
+    operationServiceEndpoint?: string,
+): Promise<Operation> => {
+    const client = session.client(serviceClients.OperationServiceClient, operationServiceEndpoint);
     const maxChecksCount = Math.ceil(timeoutMs / session.pollInterval);
 
     let checksCount = 0;