From deb78bee2b59ab5581a65d19711006cf541b4b07 Mon Sep 17 00:00:00 2001
From: Shiying Chen <shiyingchen@microsoft.com>
Date: Wed, 19 Jun 2024 16:27:55 +0800
Subject: [PATCH] add await to setazurestack

---
 lib/main/index.js        | 20 ++++++++++----------
 src/Cli/AzureCliLogin.ts | 20 ++++++++++----------
 2 files changed, 20 insertions(+), 20 deletions(-)

diff --git a/lib/main/index.js b/lib/main/index.js
index 74a534806..ec2c6a997 100644
--- a/lib/main/index.js
+++ b/lib/main/index.js
@@ -4003,7 +4003,7 @@ class AzureCliLogin {
             };
             yield this.executeAzCliCommand(["version"], true, execOptions);
             core.debug(`Azure CLI version used:\n${output}`);
-            this.setAzurestackEnvIfNecessary();
+            yield this.registerAzurestackEnvIfNecessary();
             yield this.executeAzCliCommand(["cloud", "set", "-n", this.loginConfig.environment], false);
             core.info(`Done setting cloud: "${this.loginConfig.environment}"`);
             if (this.loginConfig.authType === LoginConfig_1.LoginConfig.AUTH_TYPE_SERVICE_PRINCIPAL) {
@@ -4029,7 +4029,7 @@ class AzureCliLogin {
             }
         });
     }
-    setAzurestackEnvIfNecessary() {
+    registerAzurestackEnvIfNecessary() {
         return __awaiter(this, void 0, void 0, function* () {
             if (this.loginConfig.environment != "azurestack") {
                 return;
@@ -4037,14 +4037,14 @@ class AzureCliLogin {
             if (!this.loginConfig.resourceManagerEndpointUrl) {
                 throw new Error("resourceManagerEndpointUrl is a required parameter when environment is defined.");
             }
-            // core.info(`Unregistering cloud: "${this.loginConfig.environment}" first if it exists`);
-            // try {
-            //     await this.executeAzCliCommand(["cloud", "set", "-n", "AzureCloud"], true);
-            //     await this.executeAzCliCommand(["cloud", "unregister", "-n", this.loginConfig.environment], false);
-            // }
-            // catch (error) {
-            //     core.info(`Ignore cloud not registered error: "${error}"`);
-            // }
+            core.info(`Unregistering cloud: "${this.loginConfig.environment}" first if it exists`);
+            try {
+                yield this.executeAzCliCommand(["cloud", "set", "-n", "AzureCloud"], true);
+                yield this.executeAzCliCommand(["cloud", "unregister", "-n", this.loginConfig.environment], false);
+            }
+            catch (error) {
+                core.info(`Ignore cloud not registered error: "${error}"`);
+            }
             core.info(`Registering cloud: "${this.loginConfig.environment}" with ARM endpoint: "${this.loginConfig.resourceManagerEndpointUrl}"`);
             try {
                 let baseUri = this.loginConfig.resourceManagerEndpointUrl;
diff --git a/src/Cli/AzureCliLogin.ts b/src/Cli/AzureCliLogin.ts
index cf89b62d7..2c4c5f8f0 100644
--- a/src/Cli/AzureCliLogin.ts
+++ b/src/Cli/AzureCliLogin.ts
@@ -31,7 +31,7 @@ export class AzureCliLogin {
         await this.executeAzCliCommand(["version"], true, execOptions);
         core.debug(`Azure CLI version used:\n${output}`);
 
-        this.setAzurestackEnvIfNecessary();
+        await this.registerAzurestackEnvIfNecessary();
 
         await this.executeAzCliCommand(["cloud", "set", "-n", this.loginConfig.environment], false);
         core.info(`Done setting cloud: "${this.loginConfig.environment}"`);
@@ -59,7 +59,7 @@ export class AzureCliLogin {
         }
     }
 
-    async setAzurestackEnvIfNecessary() {
+    async registerAzurestackEnvIfNecessary() {
         if (this.loginConfig.environment != "azurestack") {
             return;
         }
@@ -67,14 +67,14 @@ export class AzureCliLogin {
             throw new Error("resourceManagerEndpointUrl is a required parameter when environment is defined.");
         }
 
-        // core.info(`Unregistering cloud: "${this.loginConfig.environment}" first if it exists`);
-        // try {
-        //     await this.executeAzCliCommand(["cloud", "set", "-n", "AzureCloud"], true);
-        //     await this.executeAzCliCommand(["cloud", "unregister", "-n", this.loginConfig.environment], false);
-        // }
-        // catch (error) {
-        //     core.info(`Ignore cloud not registered error: "${error}"`);
-        // }
+        core.info(`Unregistering cloud: "${this.loginConfig.environment}" first if it exists`);
+        try {
+            await this.executeAzCliCommand(["cloud", "set", "-n", "AzureCloud"], true);
+            await this.executeAzCliCommand(["cloud", "unregister", "-n", this.loginConfig.environment], false);
+        }
+        catch (error) {
+            core.info(`Ignore cloud not registered error: "${error}"`);
+        }
 
         core.info(`Registering cloud: "${this.loginConfig.environment}" with ARM endpoint: "${this.loginConfig.resourceManagerEndpointUrl}"`);
         try {