Added Automation to Return Storage String

Author: mwevill

Infrastructure/Resources/Get-ConfigStorageKey.ps1

@@ -0,0 +1,54 @@
+<#
+
+.SYNOPSIS
+Return either the Primary or Seconday connection String to the Consig Storage account and Write to a VSTS variable.
+
+.DESCRIPTION
+Return either the Primary or Seconday connection String to the Consig Storage account and Write to a VSTS variable.
+
+.PARAMETER Name
+The name of the Storage Account
+
+.PARAMETER useSecondary
+Boolean Switch to Return Secondary String
+
+.EXAMPLE
+.\Get-ConfigStorageKey.ps1  -Name stracc - -useSecondary $false
+
+.EXAMPLE
+.\Get-ConfigStorageKey.ps1  -Name stracc -useSecondary $true
+
+#>
+
+Param(
+    [Parameter(Mandatory = $true)]
+    [String]$Name,
+    [Parameter(Mandatory = $false)]
+    [bool]$useSecondary = $false)
+
+
+# --- Import Azure H
+Import-Module (Resolve-Path -Path $PSScriptRoot\..\Modules\Azure.psm1).Path
+Import-Module (Resolve-Path -Path $PSScriptRoot\..\Modules\Helpers.psm1).Path
+
+Write-Log -LogLevel Information -Message "Checking for existing Storage Account"
+# --- Check if storage account exists in our subscription
+$StorageAccount = Get-AzureRmResource -ResourceName $Name -ErrorAction SilentlyContinue
+
+# --- If the Storage Account doesn't exist, erorr
+if (!$StorageAccount) {
+        Write-Log -LogLevel Information -Message "StorageAccount $Name Does not exist"
+    }
+
+# --- If the storage account exists in this subscription get the key and set the env variable
+if ($StorageAccount -and !$useSecondary ) {
+    #$ResourceGroup = $($StorageAccount.ResourceGroupName)
+    $Key = (Invoke-AzureRmResourceAction -Action listKeys -ResourceType "Microsoft.ClassicStorage/storageAccounts" -ApiVersion "2016-11-01" -ResourceGroupName $($StorageAccount.ResourceGroupName) -ResourceName $($StorageAccount.Name) -force).primaryKey
+    $ConnectionString = "DefaultEndpointsProtocol=https;AccountName=$($Name);AccountKey=$($Key)"
+    Write-Output ("##vso[task.setvariable variable=ConfigurationStorageConnectionString;]$($ConnectionString)")
+} else {
+    #$ResourceGroup = $StorageAccount.ResourceGroupName
+    $Key = (Invoke-AzureRmResourceAction -Action listKeys -ResourceType "Microsoft.ClassicStorage/storageAccounts" -ApiVersion "2016-11-01" -ResourceGroupName $($StorageAccount.ResourceGroupName)  -ResourceName $($StorageAccount.Name) -force).secondaryKey
+    $ConnectionString = "DefaultEndpointsProtocol=https;AccountName=$($Name);AccountKey=$($Key)"
+    Write-Output ("##vso[task.setvariable variable=ConfigurationStorageConnectionString;]$($ConnectionString)")
+}

Tests/AT015.Get-ConfigStorageKeyTests.ps1

@@ -0,0 +1,21 @@
+$Config = Get-Content $PSScriptRoot\..\Tests\Acceptance.Config.json -Raw | ConvertFrom-Json
+Push-Location -Path $PSScriptRoot\..\Infrastructure\Resources\
+
+# requires AT003.New-ClassicStorage.Acceptance.Tests.ps1 to have ran first!
+Describe "GetConfigStorageKey Tests" -Tag "Acceptance-ARM" {
+
+    $StorageAccountName = "$($Config.classicStorageAccountName)$($Config.suffix)"
+
+    It "Should  return one output with one parameter" {
+        $Result = .\Get-ConfigStorageKey.ps1 -name $StorageAccountName
+        $Result.Count | Should Be 1
+    }
+
+    It "Should  return one output with two parameter" {
+
+        $Result = .\Get-ConfigStorageKey.ps1 -name $StorageAccountName -useSecondary $true
+        $Result.Count | Should Be 1
+    }
+}
+
+Pop-Location