test

sonic-net · Dec 13, 2024 · 6c092bc · 6c092bc
1 parent 5465226
commit 6c092bc
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/.azure-pipelines/impacted_area_testing/calculate_instance_number.py b/.azure-pipelines/impacted_area_testing/calculate_instance_number.py
@@ -61,7 +61,8 @@ def get_access_token():
         raise Exception(f"Failed to az login after {MAX_GET_TOKEN_RETRY_TIMES} attempts.")
 
     # 2. Get access token with re-try
-    get_token_cmd = "az account get-access-token --resource https://api.kusto.windows.net --query accessToken -o tsv"
+    get_token_cmd = f"az login --identity --username {managed_identity_id} " \
+                    f"& az account get-access-token --resource https://api.kusto.windows.net --query accessToken -o tsv"
     print(get_token_cmd)
     get_token_attempts = 0
     while get_token_attempts < MAX_GET_TOKEN_RETRY_TIMES: