From 7cafff85f44571d250d8b41c06c66edc5913da1c Mon Sep 17 00:00:00 2001
From: Jeremi Piotrowski <jpiotrowski@microsoft.com>
Date: Fri, 7 Jan 2022 13:56:14 +0100
Subject: [PATCH] jenkins/kola/azure: make use of --azure-use-gallery parameter

Signed-off-by: Jeremi Piotrowski <jpiotrowski@microsoft.com>
---
 jenkins/kola/azure.sh | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/jenkins/kola/azure.sh b/jenkins/kola/azure.sh
index d97475047e..110c0e3dfe 100755
--- a/jenkins/kola/azure.sh
+++ b/jenkins/kola/azure.sh
@@ -6,8 +6,11 @@ rm -rf *.tap _kola_temp*
 NAME="jenkins-${JOB_NAME##*/}-${BUILD_NUMBER}"
 
 if [[ "${BOARD}" == "arm64-usr" ]]; then
-  echo "Unsupported board"
-  exit 1
+  if [[ "${AZURE_HYPER_V_GENERATION}" != "V2" ]]; then
+    echo "Unsupported combination"
+    exit 1
+  fi
+  AZURE_USE_GALLERY="--azure-use-gallery"
 fi
 
 if [[ "${KOLA_TESTS}" == "" ]]; then
@@ -42,6 +45,7 @@ timeout --signal=SIGQUIT 20h bin/kola run \
     --azure-auth="${AZURE_AUTH_CREDENTIALS}" \
     --tapfile="${JOB_NAME##*/}.tap" \
     --torcx-manifest=torcx_manifest.json \
+    ${AZURE_USE_GALLERY} \
     ${AZURE_MACHINE_SIZE_OPT} \
     ${AZURE_HYPER_V_GENERATION:+--azure-hyper-v-generation=${AZURE_HYPER_V_GENERATION}} \
     ${KOLA_TESTS}