From 1506c59a33d3e8b2afc8149307ffded734dfa0bd Mon Sep 17 00:00:00 2001
From: Fuming Zhang <fumingzhang@microsoft.com>
Date: Wed, 19 Oct 2022 13:13:11 +0800
Subject: [PATCH] update pipeline settings

---
 src/aks-preview/azcli_aks_live_test/scripts/setup_venv.sh       | 2 +-
 .../azcli_aks_live_test/vsts-azcli-aks-live-test.yaml           | 2 +-
 .../azcli_aks_live_test/vsts-azcli-aks-unit-test.yaml           | 2 +-
 3 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/aks-preview/azcli_aks_live_test/scripts/setup_venv.sh b/src/aks-preview/azcli_aks_live_test/scripts/setup_venv.sh
index eb2210a39e1..d01c4ca5aff 100755
--- a/src/aks-preview/azcli_aks_live_test/scripts/setup_venv.sh
+++ b/src/aks-preview/azcli_aks_live_test/scripts/setup_venv.sh
@@ -145,7 +145,7 @@ if [[ -n ${setup_option} ]]; then
         echo "Start to setup az-aks-tool!"
         local_setup=${3:-"n"}
         if [[ ${local_setup} == "y" ]]; then
-            wheel_file=${4}
+            wheel_file=${4:-"/az_aks_tool-latest-py3-none-any.whl"}
             installAZAKSTOOLFromLocal "${wheel_file}"
         else
             installAZAKSTOOL
diff --git a/src/aks-preview/azcli_aks_live_test/vsts-azcli-aks-live-test.yaml b/src/aks-preview/azcli_aks_live_test/vsts-azcli-aks-live-test.yaml
index 8b693854341..e9fa239e865 100644
--- a/src/aks-preview/azcli_aks_live_test/vsts-azcli-aks-live-test.yaml
+++ b/src/aks-preview/azcli_aks_live_test/vsts-azcli-aks-live-test.yaml
@@ -54,7 +54,7 @@ jobs:
       condition: succeeded()
       displayName: "Start Container"
     - bash: |
-        docker exec $(ContainerName) /opt/scripts/setup_venv.sh setup-tool y
+        docker exec $(ContainerName) /opt/scripts/setup_venv.sh setup-tool y y /az_aks_tool-latest-py3-none-any.whl
         docker exec $(ContainerName) /opt/scripts/setup_venv.sh setup-az n azure-cli/ azure-cli-extensions/
       condition: succeeded()
       displayName: "Set up Virtual Environment"
diff --git a/src/aks-preview/azcli_aks_live_test/vsts-azcli-aks-unit-test.yaml b/src/aks-preview/azcli_aks_live_test/vsts-azcli-aks-unit-test.yaml
index 9a4f7577748..d957049239d 100644
--- a/src/aks-preview/azcli_aks_live_test/vsts-azcli-aks-unit-test.yaml
+++ b/src/aks-preview/azcli_aks_live_test/vsts-azcli-aks-unit-test.yaml
@@ -54,7 +54,7 @@ jobs:
       condition: succeeded()
       displayName: "Start Container"
     - bash: |
-        docker exec $(ContainerName) /opt/scripts/setup_venv.sh setup-tool y
+        docker exec $(ContainerName) /opt/scripts/setup_venv.sh setup-tool y y /az_aks_tool-latest-py3-none-any.whl
         docker exec $(ContainerName) /opt/scripts/setup_venv.sh setup-az n azure-cli/ azure-cli-extensions/
       condition: succeeded()
       displayName: "Set up Virtual Environment"