Framework/Helpers/AzHelper.ps1

class AzHelper
{
static [void] UploadStorageBlobContent([string] $fileName, [string] $blobName, [string] $containerName, [object] $stgCtx)
    {
        if([FeatureFlightingManager]::GetFeatureStatus("IsSetAzStorageBlobAvailable","*") -eq $true)
        {
            Set-AzStorageBlobContent -Blob $blobName -Container $containerName -File $fileName -Context $stgCtx -Force | Out-Null
        }
        else {
            $blob = $stgCtx.StorageAccount.CreateCloudBlobClient().GetContainerReference($containerName).GetBlockBlobReference($blobName)
            $task = $blob.UploadFromFileAsync($fileName)
            $task.Wait()
            if (-not ($task.IsCompleted -and !$task.IsFaulted))
            {
                #Need to change write method
                Write-Debug "Transferring file" + $fileName + "to storage has failed!!"
            }
        }
    }

    static [object] GetStorageBlobContent([string] $folderName, [string] $fileName, [string] $blobName, [string] $containerName, [object] $stgCtx)
    {
        $folderName = $folderName.TrimEnd("\")
        $folderName = $folderName.TrimEnd("/")
        $fileName = Join-Path $folderName $fileName
        return [AzHelper]::GetStorageBlobContent($fileName, $blobName, $containerName, $stgCtx)
    }

    static [object] GetStorageBlobContent([string] $fileName, [string] $blobName, [string] $containerName, [object] $stgCtx)
    {
        if([FeatureFlightingManager]::GetFeatureStatus("IsSetAzStorageBlobAvailable","*") -eq $true)
        {
            $result = Get-AzStorageBlobContent -Blob $blobName -Container $containerName -Destination $fileName -Context $stgCtx -Force 
        }
        else {
            $blob = Get-AzStorageBlob -Container $containerName -Blob $blobName -Context $stgCtx
            $task = $blob.ICloudBlob.DownloadToFileAsync($fileName,[System.IO.FileMode]::Create)
            $task.Wait()
            if (-not ($task.IsCompleted -and !$task.IsFaulted))
            {
                #Need to change write method
                Write-Debug "Downloading file from" + $blobName + " has failed!!"
            }
            $result = $blob 
        }
        return $result
    }
}