Support Set-AzureRMstorageAccount with Kind StorageV2

Author: blueww

src/ResourceManager/Storage/ChangeLog.md

@@ -21,6 +21,7 @@
 * Upgrade SRP SDK to 7.1.0
 * Add StorageV2 account kind to resource mode storage account cmdlets
     - New-AzureRmStorageAccount
+    - Set-AzureRmStorageAccount
 
 ## Version 4.0.1
 * Fixed assembly loading issue that caused some cmdlets to fail when executing

src/ResourceManager/Storage/Commands.Management.Storage.Test/ScenarioTests/StorageAccountTests.cs

@@ -112,5 +112,11 @@ public void TestNetworkRule()
             TestController.NewInstance.RunPsTest("Test-NetworkRule");
         }
 
+        [Fact]
+        [Trait(Category.AcceptanceType, Category.CheckIn)]
+        public void TestSetAzureStorageAccountStorageV2()
+        {
+            TestController.NewInstance.RunPsTest("Test-SetAzureStorageAccountStorageV2");
+        }
     }
 }

src/ResourceManager/Storage/Commands.Management.Storage.Test/ScenarioTests/StorageAccountTests.ps1

@@ -520,7 +520,6 @@ function Test-SetAzureRmCurrentStorageAccount
     }
 }
 
-
 <#
 .SYNOPSIS
 Test NetworkRule
@@ -611,3 +610,50 @@ function Test-NetworkRule
         Clean-ResourceGroup $rgname
     }
 }
+
+<#
+.SYNOPSIS
+Test SetAzureStorageAccount with Kind as StorageV2
+.Description
+AzureAutomationTest
+#>
+function Test-SetAzureStorageAccountStorageV2
+{
+    # Setup
+    $rgname = Get-StorageManagementTestResourceName;
+
+    try
+    {
+        # Test
+        $stoname = 'sto' + $rgname;
+        $stotype = 'Standard_GRS';
+        $loc = Get-ProviderLocation ResourceManagement;
+		$kind = 'Storage'
+
+        New-AzureRmResourceGroup -Name $rgname -Location $loc;
+        $loc = Get-ProviderLocation_Stage ResourceManagement;
+        New-AzureRmStorageAccount -ResourceGroupName $rgname -Name $stoname -Location $loc -Type $stotype -Kind $kind;
+
+        Retry-IfException { $global:sto = Get-AzureRmStorageAccount -ResourceGroupName $rgname -Name $stoname; }
+        $stotype = 'StandardGRS';
+        Assert-AreEqual $sto.StorageAccountName $stoname;
+        Assert-AreEqual $sto.Sku.Name $stotype;
+        Assert-AreEqual $sto.Location $loc;
+        Assert-AreEqual $sto.Kind $kind;        
+      				
+		$kind = 'StorageV2'
+        Set-AzureRmStorageAccount -ResourceGroupName $rgname -Name $stoname -Kind $kind;
+        $sto = Get-AzureRmStorageAccount -ResourceGroupName $rgname  -Name $stoname;
+        Assert-AreEqual $sto.StorageAccountName $stoname;
+        Assert-AreEqual $sto.Sku.Name $stotype;
+        Assert-AreEqual $sto.Location $loc;
+        Assert-AreEqual $sto.Kind $kind;
+
+        Remove-AzureRmStorageAccount -Force -ResourceGroupName $rgname -Name $stoname;
+    }
+    finally
+    {
+        # Cleanup
+        Clean-ResourceGroup $rgname
+    }
+}