Apply changes to support -Playback mode (#24555)

* Update to autorest 4.x
Update to Azure Policy 2023-04-01
Move to non-versioned interface types
Complete full test suite for existing cmdlets
Few bug fixes
Add completer for Location parameter
Fix record/playback, remove -LiveOnly tags
Make test names repeatable to work with record/playback
Implement resource group functions that became inaccessible
Fix test failures due to default parameter injection

* Clean up some transforms in README
Remove unsupported (for now) parameters
Streamline common test code
Move common test code to utils.ps1
Add serialization of test variables to env.json

* Remove extra parameters from Get-AzPolicyAssignment
Rerecord the tests

* Fix $testFilesFolder calculation to work on macOS/linux.

* Hide generated parameter sets for New-*
Correct definitions, validation, and tests for New-AzPolicyAssignment parameters
Re-record all tests

* Replace Get-AzContext with Utils\Get-SubscriptionIdTestSafe
Move to autorest 4.0.690
Update managed identity properties to match generation changes
Remove several parameter transforms no longer needed
Rerecord all tests
91 tests still failing on -Playback with the same signature

* Remove location completer from exports/docs/examples/help

* Improve location completer tests

* Remove transforms for Scope and Id parameters no longer needed
Update cmdlets to use Scope and Id directly
Rerecord all tests
All tests pass in -Record mode
Failures in -Playback mode unchanged

* Change location completer to lazy load (vs. module load)
Preserve "extra" input parameters in Update-* calls to Get-* in order to support -Playback test mode
Tag test files that don't support -Playback mode as LiveOnly
Rerecord all recording files

Author: mentat9

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 ')'