Adding script cmdlets for managing storage accounts. This will replace use of production Storage management cmdlets in test setup code.

Author: markcowl

src/ResourceManager/Common/Commands.ScenarioTests.ResourceManager.Common/AzureRM.Storage.ps1

@@ -0,0 +1,93 @@
+
+function Get-AzureRmStorageAccount
+{
+
+  [CmdletBinding()]
+  param(
+    [string] [Parameter(Position=0, ValueFromPipelineByPropertyName=$true)] $ResourceGroupName,
+    [string] [Parameter(Position=1, ValueFromPipelineByPropertyName=$true)] $Name)
+  BEGIN { 
+    $context = Get-Context
+	$client = Get-StorageClient $context
+  }
+  PROCESS {
+    $getTask = $client.StorageAccounts.GetPropertiesAsync($ResourceGroupName, $name, [System.Threading.CancellationToken]::None)
+	$sa = $getTask.Result
+	$account = Get-StorageAccount $sa $ResourceGroupName
+	Write-Output $account
+  }
+  END {}
+
+}
+
+function New-AzureRmStorageAccount
+{
+  [CmdletBinding()]
+  param(
+    [string] [Parameter(Position=0, ValueFromPipelineByPropertyName=$true)] $ResourceGroupName,
+    [string] [Parameter(Position=1, ValueFromPipelineByPropertyName=$true)] $Name,
+	[string] [Parameter(Position=2, ValueFromPipelineByPropertyName=$true)] $Location,
+	[string] [Parameter(Position=3, ValueFromPipelineByPropertyName=$true)] $Type)
+  BEGIN { 
+    $context = Get-Context
+	$client = Get-StorageClient $context
+  }
+  PROCESS {
+    $createParms = New-Object -Type Microsoft.Azure.Management.Storage.Models.StorageAccountCreateParameters
+	$createParms.AccountType = [Microsoft.Azure.Management.Storage.Models.AccountType]::StandardLRS
+	$createParms.Location = $Location
+    $getTask = $client.StorageAccounts.CreateAsync($ResourceGroupName, $name, $createParms, [System.Threading.CancellationToken]::None)
+	$sa = $getTask.Result
+	$account = Get-StorageAccount $ResourceGroupName $Name
+	Write-Output $account
+  }
+  END {}
+
+}
+
+function Get-AzureRmStorageAccountKey
+{
+  [CmdletBinding()]
+  param(
+    [string] [Parameter(Position=0, ValueFromPipelineByPropertyName=$true)] $ResourceGroupName,
+    [string] [Parameter(Position=1, ValueFromPipelineByPropertyName=$true)] $Name)
+  BEGIN {  
+    $context = Get-Context
+	$client = Get-StorageClient $context
+  }
+  PROCESS { 
+    $getTask = $client.StorageAccounts.ListKeysAsync($ResourceGroupName, $name, [System.Threading.CancellationToken]::None)
+	Write-Output $getTask.Result.StorageAccountKeys
+  }
+  END {}
+}
+
+function Get-Context
+{
+    [Microsoft.Azure.Commands.Common.Authentication.Models.AzureContext]$context = $null
+    $profile = [Microsoft.WindowsAzure.Commands.Common.AzureRmProfileProvider]::Instance.Profile
+	if ($profile -ne $null)
+	{
+	  $context = $profile.Context
+	}
+
+	return $context
+}
+
+function Get-StorageClient
+{
+  param([Microsoft.Azure.Commands.Common.Authentication.Models.AzureContext] $context)
+  $factory = [Microsoft.Azure.Commands.Common.Authentication.AzureSession]::ClientFactory
+  [System.Type[]]$types = [Microsoft.Azure.Commands.Common.Authentication.Models.AzureContext], [Microsoft.Azure.Commands.Common.Authentication.Models.AzureEnvironment+Endpoint]
+  $method = [Microsoft.Azure.Commands.Common.Authentication.IClientFactory].GetMethod("CreateClient", $types)
+  $closedMethod = $method.MakeGenericMethod([Microsoft.Azure.Management.Storage.StorageManagementClient])
+  $arguments = $context, [Microsoft.Azure.Commands.Common.Authentication.Models.AzureEnvironment+Endpoint]::ResourceManager
+  $client = $closedMethod.Invoke($factory, $arguments)
+  return $client
+}
+
+function Get-StorageAccount {
+  param([string] $resourceGroupName, [string] $name)
+  $sa = New-Object PSObject -Property @{"Name" = $name; "ResourceGroupName" = $resourceGroupName}
+  return $sa
+}

src/ResourceManager/Common/Commands.ScenarioTests.ResourceManager.Common/Commands.ScenarioTests.ResourceManager.Common.csproj

@@ -161,6 +161,9 @@
     <None Include="Assert.ps1">
       <CopyToOutputDirectory>PreserveNewest</CopyToOutputDirectory>
     </None>
+    <None Include="AzureRM.Storage.ps1">
+      <CopyToOutputDirectory>PreserveNewest</CopyToOutputDirectory>
+    </None>
     <None Include="Common.ps1">
       <CopyToOutputDirectory>PreserveNewest</CopyToOutputDirectory>
     </None>

src/ResourceManager/Compute/Commands.Compute.Test/Commands.Compute.Test.csproj

@@ -218,6 +218,9 @@
       <Link>ScenarioTests\Assert.ps1</Link>
       <CopyToOutputDirectory>Always</CopyToOutputDirectory>
     </None>
+    <None Include="..\..\Common\Commands.ScenarioTests.ResourceManager.Common\AzureRM.Storage.ps1">
+      <Link>ScenarioTests\AzureRM.Storage.ps1</Link>
+    </None>
     <None Include="..\..\Resources\Commands.Resources.Test\ScenarioTests\Common.ps1">
       <Link>ScenarioTests\Common.ps1</Link>
       <CopyToOutputDirectory>Always</CopyToOutputDirectory>

src/ResourceManager/Compute/Commands.Compute.Test/Common/ComputeTestController.cs

@@ -134,7 +134,8 @@ public void RunPsTestWorkflow(
                     helper.RMStorageDataPlaneModule,
                     helper.RMStorageModule,
                     helper.GetRMModulePath("AzureRM.Compute.psd1"),
-                    helper.GetRMModulePath("AzureRM.Network.psd1"));
+                    helper.GetRMModulePath("AzureRM.Network.psd1"),
+                    "AzureRM.Storage.ps1");
 
                 try
                 {