From f0ba25cd41fc068ec85b55783aef98cf8635e95d Mon Sep 17 00:00:00 2001
From: Yifan Zhang <yifanzhang@microsoft.com>
Date: Tue, 29 Oct 2024 11:49:38 +0800
Subject: [PATCH] Update managed disk test; update utils

---
 src/Storage/RegressionTests/dataplane.ps1 | 2 +-
 src/Storage/RegressionTests/utils.ps1     | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/Storage/RegressionTests/dataplane.ps1 b/src/Storage/RegressionTests/dataplane.ps1
index 1f576bab7744..ed6697e39f15 100644
--- a/src/Storage/RegressionTests/dataplane.ps1
+++ b/src/Storage/RegressionTests/dataplane.ps1
@@ -1071,7 +1071,7 @@ Describe "dataplane test" {
         $Error.Clear()        
         
         # Disk is https://ms.portal.azure.com/#@microsoft.onmicrosoft.com/resource/subscriptions/{SubscriptionId}/resourceGroups/weitry/providers/Microsoft.Compute/disks/weioauthsas_download1G/overview
-        $DiskName = $testNode.downloadManagedDisk.diskName
+        $DiskName = "yifantestdisksasoauth" #$testNode.downloadManagedDisk.diskName
 
         #Generate the SAS Uri to download
         New-AzDiskUpdateConfig -DataAccessAuthMode "AzureActiveDirectory" | Update-AzDisk -ResourceGroupName $resourceGroupName -DiskName $DiskName
diff --git a/src/Storage/RegressionTests/utils.ps1 b/src/Storage/RegressionTests/utils.ps1
index f541d77262d1..2bcea940d4df 100644
--- a/src/Storage/RegressionTests/utils.ps1
+++ b/src/Storage/RegressionTests/utils.ps1
@@ -132,7 +132,7 @@ function CompareFileFileList
 
 function Upload_Download_FileTree
 {
-    param([Microsoft.WindowsAzure.Commands.Storage.AzureStorageContext]$context, [string]$rootFolder)
+    param([Microsoft.WindowsAzure.Commands.Storage.Common.AzureStorageContext]$context, [string]$rootFolder)
 
     $containerName = GetRandomContainerName