Merge pull request #8133 from wastoresh/pipeline

[Storage] Add test for account pipeline

Author: markcowl

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

@@ -128,5 +128,14 @@ public void TestGetAzureStorageLocationUsage()
         {
             TestController.NewInstance.RunPsTest(_logger, "Test-GetAzureStorageLocationUsage");
         }
+
+        [Fact]
+        [Trait(Category.AcceptanceType, Category.CheckIn)]
+        public void TestPipingNewUpdateAccount()
+        {
+            TestController.NewInstance.RunPsTest(_logger, "Test-PipingNewUpdateAccount");
+        }
+
+        
     }
 }

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

@@ -538,7 +538,7 @@ function Test-NetworkRule
 
         New-AzureRmResourceGroup -Name $rgname -Location $loc;
 
-        New-AzureRmStorageAccount -ResourceGroupName $rgname -Name $stoname -Location $loc -Type $stotype -NetworkRuleSet (@{bypass="Logging,Metrics,AzureServices";
+        $global:sto = New-AzureRmStorageAccount -ResourceGroupName $rgname -Name $stoname -Location $loc -Type $stotype -NetworkRuleSet (@{bypass="Logging,Metrics,AzureServices";
             ipRules=(@{IPAddressOrRange="$ip1";Action="allow"},
             @{IPAddressOrRange="$ip2";Action="allow"});
             defaultAction="Deny"}) 
@@ -551,8 +551,8 @@ function Test-NetworkRule
         Assert-AreEqual $ip2 $stoacl.IpRules[1].IPAddressOrRange;
         Assert-AreEqual 0 $stoacl.VirtualNetworkRules.Count
 
-        Update-AzureRmStorageAccountNetworkRuleSet -verbose -ResourceGroupName $rgname -Name $stoname -Bypass AzureServices,Metrics -DefaultAction Allow -IpRule (@{IPAddressOrRange="$ip3";Action="allow"},@{IPAddressOrRange="$ip4";Action="allow"})
-        $stoacl = Get-AzureRmStorageAccountNetworkRuleSet -ResourceGroupName $rgname -Name $stoname
+        $sto | Update-AzureRmStorageAccountNetworkRuleSet -verbose -Bypass AzureServices,Metrics -DefaultAction Allow -IpRule (@{IPAddressOrRange="$ip3";Action="allow"},@{IPAddressOrRange="$ip4";Action="allow"})
+        $stoacl = $sto | Get-AzureRmStorageAccountNetworkRuleSet
         $stoacliprule = $stoacl.IpRules
         Assert-AreEqual 6 $stoacl.Bypass;
         Assert-AreEqual Allow $stoacl.DefaultAction;
@@ -677,3 +677,44 @@ function Test-GetAzureStorageLocationUsage
         Assert-AreNotEqual 0 $usage.Limit;
         Assert-AreNotEqual 0 $usage.CurrentValue;      
 }
+
+<#
+.SYNOPSIS
+Test Get-AzureRmStorageAccount | New/Set-AzureRmStorageAccount
+#>
+function Test-PipingNewUpdateAccount
+{
+    # Setup
+    $rgname = Get-StorageManagementTestResourceName;
+
+    try
+    {
+        # Test
+        $stoname = 'sto' + $rgname;
+        $stoname2 = 'sto' + $rgname + '2';
+        $stotype = 'Standard_GRS';
+        $loc = Get-ProviderLocation ResourceManagement;
+
+        New-AzureRmResourceGroup -Name $rgname -Location $loc;
+
+        $global:sto = New-AzureRmStorageAccount -ResourceGroupName $rgname -Name $stoname -Location $loc -Type $stotype;
+
+        Retry-IfException { $global:sto2 = Get-AzureRmStorageAccount -ResourceGroupName $rgname -Name $stoname | New-AzureRmStorageAccount -Name $stoname2 -skuName $stotype; }
+        Assert-AreEqual $sto.ResourceGroupName $sto2.ResourceGroupName;
+        Assert-AreEqual $sto.Location $sto2.Location;
+        Assert-AreNotEqual $sto.StorageAccountName $sto2.StorageAccountName;
+		
+		Get-AzureRmStorageAccount -ResourceGroupName $rgname -Name $stoname | set-AzureRmStorageAccount -UpgradeToStorageV2
+		$global:sto = $sto | set-AzureRmStorageAccount -EnableHttpsTrafficOnly $true
+        Assert-AreEqual 'StorageV2' $sto.Kind;
+        Assert-AreEqual $true $sto.EnableHttpsTrafficOnly;
+
+        Get-AzureRmStorageAccount -ResourceGroupName $rgname -Name $stoname | Remove-AzureRmStorageAccount -Force;
+        $sto2 | Remove-AzureRmStorageAccount -Force;
+    }
+    finally
+    {
+        # Cleanup
+        Clean-ResourceGroup $rgname
+    }
+}