SNOW-723810: Clean up and refactor

snowflakedb · Sep 22, 2023 · e6bcd21 · e6bcd21
1 parent d302f60
commit e6bcd21
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/Snowflake.Data.Tests/UnitTests/SFFileTransferAgentTests.cs b/Snowflake.Data.Tests/UnitTests/SFFileTransferAgentTests.cs
@@ -8,7 +8,6 @@ namespace Snowflake.Data.Tests.UnitTests
     using Snowflake.Data.Core;
     using Snowflake.Data.Core.FileTransfer;
     using System.Collections.Generic;
-    using Snowflake.Data.Tests.Mock;
     using System.Threading.Tasks;
     using System.Threading;
     using System.IO;

diff --git a/Snowflake.Data/Core/FileTransfer/SFFileTransferAgent.cs b/Snowflake.Data/Core/FileTransfer/SFFileTransferAgent.cs
@@ -596,7 +596,7 @@ private void initFileMetadata(
 
                     /// The storage client used to upload data from files or streams
                     /// This is only needed for remote storage types
-                    if (TransferMetadata.stageInfo.locationType != LOCAL_FS)
+                    if (StorageClientType.REMOTE == GetStorageClientType(TransferMetadata.stageInfo))
                     {
                         fileMetadata.client = SFRemoteStorageUtil.GetRemoteStorage(TransferMetadata);
                     }
@@ -661,7 +661,7 @@ private void initFileMetadata(
 
                     /// The storage client used to download data from files or streams
                     /// This is only needed for remote storage types
-                    if (TransferMetadata.stageInfo.locationType != LOCAL_FS)
+                    if (StorageClientType.REMOTE == GetStorageClientType(TransferMetadata.stageInfo))
                     {
                         fileMetadata.client = SFRemoteStorageUtil.GetRemoteStorage(TransferMetadata);