Apply changes to support -Playback mode

src/Resources/Policy.Autorest/custom/Helpers.ps1

@@ -514,20 +514,66 @@ function LocationCompleter(
     $fakeBoundParameter
 )
 {
-    $currentLocations | Where-Object { $_ -like "$wordToComplete*" }
+    if ($global:currentLocations.Count -le 0) {
+        $response = Invoke-AzRestMethod -Uri "https://management.azure.com/subscriptions/$subscriptionId/locations?api-version=2022-12-01" -Method GET
+        $global:currentLocations = ($response.Content | ConvertFrom-Json -Depth 100).value | Sort-Object -Property name | Select-Object -ExpandProperty name
+    }
+
+    # If you see the following error, it means your context access has expired
+    #   The given key 'AzureAttestationServiceEndpointSuffix' was not present in the dictionary.
+    $global:currentLocations | Where-Object { $_ -like "$wordToComplete*" }
 }
 
 function Get-SubscriptionId {
     (Utils\Get-SubscriptionIdTestSafe)
 }
 
+function Get-ExtraParameters
+(
+    $DefaultProfile,
+    $Break,
+    $HttpPipelineAppend,
+    $HttpPipelinePrepend,
+    $Proxy,
+    $ProxyCredential,
+    $ProxyUseDefaultCredentials
+) {
+    $parms = @{}
+    if ($PSBoundParameters['DefaultProfile']) {
+        $parms['DefaultProfile'] = $PSBoundParameters['DefaultProfile']
+    }
+
+    if ($PSBoundParameters['Break']) {
+        $parms['Break'] = $PSBoundParameters['Break']
+    }
+
+    if ($PSBoundParameters['HttpPipelineAppend']) {
+        $parms['HttpPipelineAppend'] = $PSBoundParameters['HttpPipelineAppend']
+    }
+
+    if ($PSBoundParameters['HttpPipelinePrepend']) {
+        $parms['HttpPipelinePrepend'] = $PSBoundParameters['HttpPipelinePrepend']
+    }
+
+    if ($PSBoundParameters['Proxy']) {
+        $parms['Proxy'] = $PSBoundParameters['Proxy']
+    }
+
+    if ($PSBoundParameters['ProxyCredential']) {
+        $parms['ProxyCredential'] = $PSBoundParameters['ProxyCredential']
+    }
+
+    if ($PSBoundParameters['ProxyUseDefaultCredentials']) {
+        $parms['ProxyUseDefaultCredentials'] = $PSBoundParameters['ProxyUseDefaultCredentials']
+    }
+
+    return $parms
+}
+
 # register the location completer for New-AzPolicyAssignment
 Register-ArgumentCompleter -CommandName New-AzPolicyAssignment -ParameterName Location -ScriptBlock ${function:LocationCompleter}
 
 # cache Azure locations to be used by the location completer (Get-AzLocation is not available in this context, need to use REST)
-$subscriptionId = (Get-SubscriptionId)
+$global:currentLocations = @()
 
-$response = Invoke-AzRestMethod -Uri "https://management.azure.com/subscriptions/$subscriptionId/locations?api-version=2022-12-01" -Method GET
-$currentLocations = ($response.Content | ConvertFrom-Json -Depth 100).value | Sort-Object -Property name | Select-Object -ExpandProperty name
-# If you see the following error, it means your context access has expired
-# The given key 'AzureAttestationServiceEndpointSuffix' was not present in the dictionary.
+$subscriptionId = (Get-SubscriptionId)

src/Resources/Policy.Autorest/custom/Update-AzPolicyAssignment.ps1

@@ -244,7 +244,9 @@ process {
 
     # construct id for assignment to update
     $resolved = ResolvePolicyAssignment $Name $Scope $thisId
-    $getParameters = @{ Id = $resolved.ResourceId }
+
+    $getParameters = Get-ExtraParameters @PSBoundParameters
+    $getParameters['Id'] = $resolved.ResourceId
 
     if ($writeln) {
         Write-Host -ForegroundColor Blue -> Get-AzPolicyAssignment'(' $getParameters ')'

src/Resources/Policy.Autorest/custom/Update-AzPolicyDefinition.ps1

@@ -188,7 +188,9 @@ process {
 
     # construct id for definition to update
     $resolved = ResolvePolicyDefinition $Name $SubscriptionId $ManagementGroupName $thisId
-    $getParameters = @{ Id = $resolved.ResourceId }
+
+    $getParameters = Get-ExtraParameters @PSBoundParameters
+    $getParameters['Id'] = $resolved.ResourceId
 
     if ($writeln) {
         Write-Host -ForegroundColor Blue -> Get-AzPolicyDefinition'(' $getParameters ')'

src/Resources/Policy.Autorest/custom/Update-AzPolicyExemption.ps1

@@ -187,7 +187,9 @@ process {
 
     # construct id for exemption to update
     $resolved = ResolvePolicyExemption $Name $Scope $thisId
-    $getParameters = @{ Id = $resolved.ResourceId }
+
+    $getParameters = Get-ExtraParameters @PSBoundParameters
+    $getParameters['Id'] = $resolved.ResourceId
 
     if ($writeln) {
         Write-Host -ForegroundColor Blue -> Get-AzPolicyExemption'(' $getParameters ')'

src/Resources/Policy.Autorest/custom/Update-AzPolicySetDefinition.ps1

@@ -189,7 +189,9 @@ process {
 
     # construct id for definition to update
     $resolved = ResolvePolicySetDefinition $Name $SubscriptionId $ManagementGroupName $thisId
-    $getParameters = @{ Id = $resolved.ResourceId }
+
+    $getParameters = Get-ExtraParameters @PSBoundParameters
+    $getParameters['Id'] = $resolved.ResourceId
 
     if ($writeln) {
         Write-Host -ForegroundColor Blue -> Get-AzPolicySetDefinition'(' $getParameters ')'