Reference newly generated Track 1 SDK in Tags Resources cmdlets (#22348)

* Initial commit to use Track 1 SDK for Tags and support a 202 response code for Tags PUT/PATCH/DELETE

* Remove commented code from test case

* Fix tests to check for async behavior

* Add Purview project to tests

* Update test cases

* Mark Tags tests using Az.Purview as LiveOnly

* Mark Tags tests using Az.Purview as LiveOnly pt 2

Author: meghanagupta95

src/Resources/ResourceManager/Implementation/Policy/PsPolicyIdentity.cs

@@ -18,7 +18,7 @@ public PsPolicyIdentity(JToken input)
             PrincipalId = identity.PrincipalId;
             TenantId = identity.TenantId;
 
-            var userAssignedIdentityId = identity.UserAssignedIdentities?.Take(1).Select(d => d.Key).First();
+            var userAssignedIdentityId = identity.UserAssignedIdentities?.Take(1).Select(d => d.Key).FirstOrDefault();
 
             UserAssignedIdentities = userAssignedIdentityId != null ? 
                 new Dictionary<string, PsUserAssignedIdentity>

src/Resources/Resources.Test/Resources.Test.csproj

@@ -24,6 +24,9 @@
     <ProjectReference Include="..\Resources\Resources.csproj" />
     <ProjectReference Include="..\..\ManagedServiceIdentity\Az.ManagedServiceIdentity.csproj" />
 	<ProjectReference Include="..\Resources.Sdk\Resources.Sdk.csproj" />
+    <ProjectReference Include="..\..\Purview\Purview\Purview.csproj" />
+    <ProjectReference Include="..\..\Purview\Purview.Autorest\Az.Purview.csproj" />
+    <ProjectReference Include="..\..\Purview\Purviewdata.Autorest\Az.Purviewdata.csproj" />
   </ItemGroup>
 
   <ItemGroup>

src/Resources/Resources.Test/ScenarioTests/ResourcesTestRunner.cs

@@ -43,7 +43,8 @@ protected ResourcesTestRunner(ITestOutputHelper output)
                 {
                     helper.RMResourceModule,
                     helper.GetRMModulePath("Az.Monitor.psd1"),
-                    helper.GetRMModulePath("Az.ManagedServiceIdentity.psd1")
+                    helper.GetRMModulePath("Az.ManagedServiceIdentity.psd1"),
+                    helper.GetRMModulePath("Az.Purview.psd1")
                 })
                 .WithRecordMatcher(
                     (ignoreResourcesClient, resourceProviders, userAgentsToIgnore) => new ResourcesRecordMatcher(ignoreResourcesClient, resourceProviders, userAgentsToIgnore)

src/Resources/Resources.Test/ScenarioTests/TagTests.cs

@@ -79,5 +79,26 @@ public void TagDeleteWithResourceIdParamsForResource()
         {
             TestRunner.RunTestScript("Test-TagDeleteWithResourceIdParamsForResource");
         }
+
+        [Fact]
+        [Trait(Category.RunType, Category.LiveOnly)]
+        public void TagCreateOrUpdateAsyncWithResourceIdParamsForResource()
+        {
+            TestRunner.RunTestScript("Test-TagCreateOrUpdateAsyncWithResourceIdParamsForResource");
+        }
+
+        [Fact]
+        [Trait(Category.RunType, Category.LiveOnly)]
+        public void TagUpdateAsyncWithResourceIdParamsForResource()
+        {
+            TestRunner.RunTestScript("Test-TagUpdateAsyncWithResourceIdParamsForResource");
+        }
+
+        [Fact]
+        [Trait(Category.RunType, Category.LiveOnly)]
+        public void TagDeleteAsyncWithResourceIdParamsForResource()
+        {
+            TestRunner.RunTestScript("Test-TagDeleteAsyncWithResourceIdParamsForResource");
+        }
     }
 }

src/Resources/Resources.Test/ScenarioTests/TagTests.ps1

@@ -61,6 +61,44 @@ function Test-TagCreateOrUpdateWithResourceIdParamsForResource
     Test-TagCreateOrUpdateWithResourceIdParams $resourceId
 }
 
+<#
+.SYNOPSIS
+Utility method to test async CreateOrUpdate for Tags within tracked resources and subscription.
+#>
+function Test-TagCreateOrUpdateAsyncWithResourceIdParams($resourceId)
+{
+    # Setup
+    $expected = @{"key1"="value1"; "key2"="value2";}
+
+    try
+    {
+        # Test
+        $newTagres = New-AzTag -ResourceId $resourceId -Tag $expected
+        Start-TestSleep -Seconds 120
+        $res = Get-AzTag -ResourceId $resourceId
+        
+        [hashtable]$actual = $res.Properties.TagsProperty
+        Assert-True { AreHashtableEqual $expected $actual }
+    }
+    finally
+    {
+        # Cleanup
+        Remove-AzTag -ResourceId $resourceId
+    } 
+}
+
+<#
+.SYNOPSIS
+Tests creating or updating tags on tracked resource.
+#>
+function Test-TagCreateOrUpdateAsyncWithResourceIdParamsForResource
+{
+    # Setup
+    $resourceId = NewTestResourcePurviewAccount
+
+    Test-TagCreateOrUpdateAsyncWithResourceIdParams $resourceId
+}
+
 
 <#
 .SYNOPSIS
@@ -143,6 +181,77 @@ function Test-TagUpdateWithResourceIdParamsForResource
     Test-UpdateWithResourceIdParams $resourceId
 }
 
+<#
+.SYNOPSIS
+Utility method to test async updating tags on subscription and tracked resource, including Merge, Replace, and Delete Operation.
+#>
+function Test-UpdateAsyncWithResourceIdParams($resourceId)
+{
+    # Setup
+    $original = @{"key1"="value1"; "key2"="value2";}
+    New-AzTag -ResourceId $resourceId -Tag $original
+    Start-TestSleep -Seconds 120
+
+    try
+    {
+        # Test
+        {
+            # merge operation
+            $merged = @{"key1"="value1"; "key3"="value3";}
+            $res = Update-AzTag -ResourceId $resourceId -Tag $merged -Operation Merge
+            Start-TestSleep -Seconds 120
+    
+            $expected = @{"key1"="value1"; "key2"="value2"; "key3"="value3";}
+            [hashtable]$actual = $res.Properties.TagsProperty
+
+            # Assert
+            Assert-True { AreHashtableEqual $expected $actual }
+        }
+
+        {
+            # replace operation
+            $replaced = @{"key1"="value1"; "key3"="value3";}
+            $res = Update-AzTag -ResourceId $resourceId -Tag $replaced -Operation Replace
+            Start-TestSleep -Seconds 120
+
+            $expected = $replaced
+            [hashtable]$actual = $res.Properties.TagsProperty
+
+            # Assert
+            Assert-True { AreHashtableEqual $expected $actual }
+        }
+
+        {
+            # delete operation
+            $deleted = @{"key1"="value1"; "key3"="value3";}
+            $res = Update-AzTag -ResourceId $resourceId -Tag $deleted -Operation Delete
+            Start-TestSleep -Seconds 120
+
+            $expected = null
+            [hashtable]$actual = $res.Properties.TagsProperty
+
+            # Assert
+            Assert-True { AreHashtableEqual $expected $actual }
+        }
+    }
+    finally
+    {
+        # Cleanup
+        Remove-AzTag -ResourceId $resourceId
+    } 
+}
+
+<#
+.SYNOPSIS
+Tests async updating tags on tracked resource.
+#>
+function Test-TagUpdateAsyncWithResourceIdParamsForResource
+{
+    # Setup
+    $resourceId = NewTestResourcePurviewAccount
+
+    Test-UpdateAsyncWithResourceIdParams $resourceId
+}
 
 <#
 .SYNOPSIS
@@ -247,6 +356,46 @@ function Test-TagDeleteWithResourceIdParamsForResource
     Test-TagDeleteWithResourceIdParams $resourceId
 }
 
+<#
+.SYNOPSIS
+Utility method to test Delete for Tags within tracked resources and subscription.
+#>
+function Test-TagDeleteAsyncWithResourceIdParams($resourceId)
+{
+    # Setup
+    $original = @{"key1"="value1"; "key2"="value2";}
+    New-AzTag -ResourceId $resourceId -Tag $original
+    Start-TestSleep -Seconds 120
+
+    try 
+    {
+        # Test
+        Remove-AzTag -ResourceId $resourceId  
+        Start-TestSleep -Seconds 120
+        $actual = Get-AzTag -ResourceId $resourceId
+
+        # Assert
+        Assert-AreEqual $actual.Properties.TagsProperty.Count 0
+    }
+    finally
+    {
+        # Cleanup
+        Remove-AzTag -ResourceId $resourceId
+    }        
+}
+
+<#
+.SYNOPSIS
+Tests getting tags on tracked resource.
+#>
+function Test-TagDeleteAsyncWithResourceIdParamsForResource
+{
+    # Setup
+    $resourceId = NewTestResourcePurviewAccount
+
+    Test-TagDeleteAsyncWithResourceIdParams $resourceId
+}
+
 <#
 .SYNOPSIS
 utility method to get default subscriptionId
@@ -265,13 +414,13 @@ utility method to create resource group
 #>
 function NewTestResourceGroup
 {
-    $rgName = "RG-Test05"
+    $rgName = Get-ResourceGroupName
     $location = "Central US"
 
-    $existed = Get-AzureRmResourceGroup -Name $rgName -ErrorVariable notPresent -ErrorAction SilentlyContinue
+    $existed = Get-AzResourceGroup -Name $rgName -ErrorVariable notPresent -ErrorAction SilentlyContinue
 
     if($notPresent) {
-        $existed = New-AzureRmResourceGroup -Name $rgName -Location $location
+        $existed = New-AzResourceGroup -Name $rgName -Location $location
 	}
 
     return $existed
@@ -285,18 +434,40 @@ function NewTestResource
 {
     $rg = NewTestResourceGroup
 
-    $resourceName = "RS-Test05"
+    $resourceName = Get-ResourceName
     $resourceId = $rg.ResourceId + "/providers/microsoft.web/sites/" + $resourceName
 
-    $location = "Central US"
+    $location = "West Central US"
     $property = @{test="test-tag"}
     $resourceType = "microsoft.web/sites"
 
-    $existed = Get-AzureRmResource -ResourceId $resourceId -ErrorVariable notPresent -ErrorAction SilentlyContinue
+    $existed = Get-AzResource -ResourceId $resourceId -ErrorVariable notPresent -ErrorAction SilentlyContinue
 
     if($notPresent) {
-        $existed = New-AzureRmResource -Location $location -Properties $property -ResourceName $resourceName -ResourceType $resourceType -ResourceGroupName $rg.ResourceGroupName -Force
+        $existed = New-AzResource -Location $location -Properties $property -ResourceName $resourceName -ResourceType $resourceType -ResourceGroupName $rg.ResourceGroupName -Force
 	}
 
     return $resourceId
 }
+
+<#
+.SYNOPSIS
+utility method to creare resource
+#>
+function NewTestResourcePurviewAccount
+{
+    $rg = NewTestResourceGroup
+    $resourceName = Get-ResourceName
+    $resourceId = $rg.ResourceId + "/providers/Microsoft.Purview/accounts/" + $resourceName
+
+    $location = "westus"
+    $property = @{test="test-tag"}
+    $resourceType = "microsoft.purview/accounts"
+
+    $existed = Get-AzResource -ResourceId $resourceId -ErrorVariable notPresent -ErrorAction SilentlyContinue
+    if ($notPresent) {
+        $existed = New-AzPurviewAccount -Name $resourceName -ResourceGroupName $rg.ResourceGroupName -IdentityType SystemAssigned -Location $location -Tag $property -SkuCapacity 4 -SkuName Standard
+    }
+
+    return $resourceId
+}