Continue fix issue in migrate to file track2 (#19658)

Author: blueww

src/Storage/Storage/File/Cmdlet/RemoveAzureStorageFile.cs

@@ -98,7 +98,7 @@ public override void ExecuteCmdlet()
                     // when only track1 object input, will miss storage context, so need to build storage context for prepare the output object.
                     if (this.Context == null)
                     {
-                        this.Context = GetStorageContextFromTrack1FileServiceClient(this.Directory.ServiceClient, DefaultContext);
+                        this.Context = GetStorageContextFromTrack1FileServiceClient(this.File.ServiceClient, DefaultContext);
                     }
                     break;
 
@@ -124,7 +124,7 @@ public override void ExecuteCmdlet()
                     // when only track1 object input, will miss storage context, so need to build storage context for prepare the output object.
                     if (this.Context == null)
                     {
-                        this.Context = GetStorageContextFromTrack1FileServiceClient(this.Share.ServiceClient, DefaultContext);
+                        this.Context = GetStorageContextFromTrack1FileServiceClient(this.Directory.ServiceClient, DefaultContext);
                     }
                     break;
 

src/Storage/Storage/File/Cmdlet/RemoveAzureStorageShare.cs

@@ -103,12 +103,12 @@ public override void ExecuteCmdlet()
                     else
                     {
                         share = AzureStorageFileShare.GetTrack2FileShareClient(this.Share, (AzureStorageContext)this.Context, this.ClientOptions);
+                    }
 
-                        // when only track1 object input, will miss storage context, so need to build storage context for prepare the output object.
-                        if (this.Context == null)
-                        {
-                            this.Context = GetStorageContextFromTrack1FileServiceClient(this.Share.ServiceClient, DefaultContext);
-                        }
+                    // when only track1 object input, will miss storage context, so need to build storage context for prepare the output object.
+                    if (this.Context == null)
+                    {
+                        this.Context = GetStorageContextFromTrack1FileServiceClient(this.Share.ServiceClient, DefaultContext);
                     }
                     break;
 

src/Storage/Storage/File/Cmdlet/StartAzureStorageFileCopy.cs

@@ -228,6 +228,12 @@ internal IStorageFileManagement GetDestinationChannel()
                         destChannel = new StorageFileManagement(this.GetCmdletStorageContext(DestContext));
                     }
                 }
+                else if (BlobFileParameterSet == this.ParameterSetName ||
+                    FileFileParameterSet == this.ParameterSetName ||
+                    UriFileParameterSet == this.ParameterSetName)
+                {
+                    destChannel = new StorageFileManagement(this.GetCmdletStorageContext(DestContext));
+                }
                 else
                 {
                     destChannel = base.CreateChannel();
@@ -243,6 +249,15 @@ internal IStorageFileManagement GetDestinationChannel()
         [PermissionSet(SecurityAction.Demand, Name = "FullTrust")]
         public override void ExecuteCmdlet()
         {
+            if(this.DestFile != null)
+            {
+                // when only track1 object input, will miss storage context, so need to build storage context for prepare the output object.
+                if (this.DestContext == null)
+                {
+                    this.DestContext = GetStorageContextFromTrack1FileServiceClient(this.DestFile.ServiceClient, DefaultContext);
+                }
+            }
+
             blobChannel = this.GetBlobChannel();
             destChannel = GetDestinationChannel();
             IStorageFileManagement srcChannel = Channel;