Runs az and az-devops during warmup

images/win/scripts/ImageHelpers/ImageHelpers.psm1

@@ -12,18 +12,22 @@ Export-ModuleMember -Function @(
     'Disconnect-Hive'
     'Test-MachinePath'
     'Get-MachinePath'
+    'Get-UserPath'
     'Get-DefaultPath'
     'Set-MachinePath'
     'Set-DefaultPath'
+    'Update-Path'
     'Add-MachinePathItem'
     'Add-DefaultPathItem'
     'Add-DefaultItem'
     'Get-SystemVariable'
+    'Get-UserVariable'
     'Get-DefaultVariable'
     'Set-SystemVariable'
     'Set-DefaultVariable'
     'Install-Binary'
     'Install-VisualStudio'
+    'Invoke-ValidateCommand'
     'Get-ToolsetContent'
     'Get-ToolsetToolFullPath'
     'Stop-SvcWithErrHandling'

images/win/scripts/ImageHelpers/InstallHelpers.ps1

@@ -644,6 +644,43 @@ function Get-GitHubPackageDownloadUrl {
     return $downloadUrl
 }
 
+function Invoke-ValidateCommand {
+    param(
+        [Parameter(Mandatory)]
+        [string]$Command,
+        [int] $Timeout = 0
+    )
+
+    if ($Timeout -eq 0)
+    {
+        $output = Invoke-Expression -Command $Command
+        if ($LASTEXITCODE -ne 0) {
+            throw "Command '$Command' has finished with exit code $LASTEXITCODE"
+        }
+        return $output
+    }
+    else
+    {
+        $job = $command | Start-Job -ScriptBlock {
+            $output = Invoke-Expression -Command $input
+            if ($LASTEXITCODE -ne 0) {
+                  throw 'Command failed'
+            }
+            return $output
+        }
+        $waitObject = $job | Wait-Job -Timeout $Timeout
+        if(-not $waitObject)
+        {
+             throw "Command '$Command' has timed out"
+        }
+        if($waitObject.State -eq 'Failed')
+        {
+             throw "Command '$Command' has failed"
+        }
+        Receive-Job -Job $job
+    }
+}
+
 function Use-ChecksumComparison {
     param (
         [Parameter(Mandatory=$true)]
@@ -691,7 +728,7 @@ function Get-HashFromGitHubReleaseBody {
         }
     }
     $body = (Invoke-RestMethod -Uri $releaseUrl).body -replace('`', "") -join "`n"
-    $matchingLine = $body.Split("`n") | Where-Object { $_ -like "*$FileName*" }    
+    $matchingLine = $body.Split("`n") | Where-Object { $_ -like "*$FileName*" }
     if ([string]::IsNullOrEmpty($matchingLine)) {
         throw "File name '$FileName' not found in release body."
     }
@@ -708,9 +745,9 @@ function Test-FileSignature {
         [Parameter(Mandatory=$true)]
         [string]$ExpectedThumbprint
     )
- 
+
     $signature = Get-AuthenticodeSignature $FilePath
- 
+
     if ($signature.Status -ne "Valid") {
         throw "Signature status is not valid. Status: $($signature.Status)"
     }
@@ -720,4 +757,41 @@ function Test-FileSignature {
     }
 
     Write-Output "Signature for $FilePath is valid"
-}
+}
+
+function Invoke-ValidateCommand {
+    param(
+        [Parameter(Mandatory)]
+        [string]$Command,
+        [int] $Timeout = 0
+    )
+
+    if ($Timeout -eq 0)
+    {
+        $output = Invoke-Expression -Command $Command
+        if ($LASTEXITCODE -ne 0) {
+            throw "Command '$Command' has finished with exit code $LASTEXITCODE"
+        }
+        return $output
+    }
+    else
+    {
+        $job = $command | Start-Job -ScriptBlock {
+            $output = Invoke-Expression -Command $input
+            if ($LASTEXITCODE -ne 0) {
+                  throw 'Command failed'
+            }
+            return $output
+        }
+        $waitObject = $job | Wait-Job -Timeout $Timeout
+        if(-not $waitObject)
+        {
+             throw "Command '$Command' has timed out"
+        }
+        if($waitObject.State -eq 'Failed')
+        {
+             throw "Command '$Command' has failed"
+        }
+        Receive-Job -Job $job
+    }
+}

images/win/scripts/ImageHelpers/PathHelpers.ps1

@@ -33,6 +33,14 @@ function Disconnect-Hive {
     }
 }
 
+function Get-UserVariable {
+    param(
+        [string]$UserVariable
+    )
+
+    [System.Environment]::GetEnvironmentVariable($UserVariable, "User")
+}
+
 function Get-SystemVariable {
     param(
         [string]$SystemVariable
@@ -80,10 +88,18 @@ function Set-DefaultVariable {
     [System.GC]::Collect()
 }
 
+function Update-Path {
+    $env:Path = (Get-MachinePath) + ";" + (Get-UserPath)
+}
+
 function Get-MachinePath {
     Get-SystemVariable PATH
 }
 
+function Get-UserPath {
+    Get-UserVariable PATH
+}
+
 function Get-DefaultPath {
     Get-DefaultVariable Path
 }

images/win/scripts/Installers/Install-AzureCli.ps1

@@ -6,10 +6,34 @@
 Write-Host "Install the latest Azure CLI release"
 $azCliUrl = "https://aka.ms/installazurecliwindowsx64"
 Install-Binary -Url $azCliUrl -Name "azure-cli.msi"
+Update-Path
 
+# Store cli extensions outside of the provisioning user's profile
 $azureCliExtensionPath = Join-Path $Env:CommonProgramFiles 'AzureCliExtensionDirectory'
 $null = New-Item -ItemType "Directory" -Path $azureCliExtensionPath
+$env:AZURE_EXTENSION_DIR = Set-SystemVariable -SystemVariable "AZURE_EXTENSION_DIR" -value $azureCliExtensionPath
 
-[Environment]::SetEnvironmentVariable("AZURE_EXTENSION_DIR", $azureCliExtensionPath, [System.EnvironmentVariableTarget]::Machine)
+# Store cli config outside of the provisioning user's profile
+$azureCliConfigPath = Join-Path $Env:CommonProgramFiles 'AzureCliDirectory'
+$null = New-Item -ItemType "Directory" -Path $azureCliConfigPath
+$env:AZURE_CONFIG_DIR = Set-SystemVariable -SystemVariable "AZURE_CONFIG_DIR" -value $azureCliConfigPath
 
-Invoke-PesterTests -TestFile "CLI.Tools" -TestName "Azure CLI"
+# Turn use automation recommended settings to speed up first-call on the hosted runner images.
+# See: https://techcommunity.microsoft.com/t5/azure-tools-blog/streamline-configuring-azure-cli-with-az-init/ba-p/3051810
+
+Write-Host "Setting automation defaults for 'az'"
+Invoke-ValidateCommand -Command "az config set auto-upgrade.enable=false --only-show-errors"
+Invoke-ValidateCommand -Command "az config set core.error_recommendation=off --only-show-errors"
+Invoke-ValidateCommand -Command "az config set core.disable_progress_bar=true --only-show-errors"
+Invoke-ValidateCommand -Command "az config set core.no_color=true --only-show-errors"
+Invoke-ValidateCommand -Command "az config set core.survey_message=false --only-show-errors"
+
+# Though recommended by az init, this may be a breaking change.
+# Invoke-ValidateCommand -Command "az config set logging.enable_log_file=no --only-show-errors"
+
+
+# Warm-up Azure CLI
+Write-Host "Warmup 'az'"
+Invoke-ValidateCommand -Command "az --help"
+
+Invoke-PesterTests -TestFile "CLI.Tools" -TestName "Azure CLI"

images/win/scripts/Installers/Install-AzureDevOpsCli.ps1

@@ -3,6 +3,27 @@
 ##  Desc:  Install Azure DevOps CLI
 ################################################################################
 
-az extension add -n azure-devops
+# Store azure-devops-cli cache outside of the provisioning user's profile
+$azureDevOpsCliConfigPath = Join-Path $Env:CommonProgramFiles 'AzureDevOpsCliConfigDirectory'
+$null = New-Item -ItemType "Directory" -Path $azureDevOpsCliConfigPath
+$env:AZURE_DEVOPS_EXT_CONFIG_DIR = Set-SystemVariable -SystemVariable "AZURE_DEVOPS_EXT_CONFIG_DIR" -value $azureDevOpsCliConfigPath
 
-Invoke-PesterTests -TestFile "CLI.Tools" -TestName "Azure DevOps CLI"
+$azureDevOpsCliCachePath = Join-Path $azureDevOpsCliConfigPath "cache"
+$null = New-Item -ItemType "Directory" -Path $azureDevOpsCliCachePath
+$env:AZURE_DEVOPS_CACHE_DIR = Set-SystemVariable -SystemVariable "AZURE_DEVOPS_CACHE_DIR" -value $azureDevOpsCliCachePath
+
+Invoke-ValidateCommand -Command "az extension add -n azure-devops"
+
+# Warm-up Azure DevOps CLI
+
+Write-Host "Warmup 'az-devops'"
+@("devops", "pipelines", "boards", "repos", "artifacts") | ForEach-Object {
+    Invoke-ValidateCommand -Command "az $_ --help"
+}
+
+# calling az devops login to force it to install `keyring`. Login will actually fail, redirecting error to null
+echo "dummy" | az devops login | out-null
+# calling az devops logout to be sure no credentials remain.
+az devops logout | out-null
+
+Invoke-PesterTests -TestFile "CLI.Tools" -TestName "Azure DevOps CLI"

images/win/scripts/Installers/Warmup-User.ps1

@@ -21,5 +21,4 @@ reg.exe load HKLM\DEFAULT c:\users\default\ntuser.dat
 reg.exe copy HKCU\Software\Microsoft\VisualStudio HKLM\DEFAULT\Software\Microsoft\VisualStudio /s
 reg.exe unload HKLM\DEFAULT
 
-
-Write-Host "Warmup-User.ps1 - completed"
+Write-Host "Warmup-User.ps1 - completed"