Modify for extra space outside of file path

snowflakedb · Dec 6, 2024 · e51ca63 · e51ca63
1 parent 21dd042
commit e51ca63
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/Snowflake.Data.Tests/UnitTests/SFFileTransferAgentTests.cs b/Snowflake.Data.Tests/UnitTests/SFFileTransferAgentTests.cs
@@ -648,7 +648,7 @@ public void TestGetFilePathWithoutSpacesFromPutCommand()
         [Test]
         public void TestGetFilePathWithSpacesFromPutCommand()
         {
-            TestGetFilePathFromPutCommand("PUT file://" + s_filePathWithSpaces + " @TestStage", s_filePathWithSpaces);
+            TestGetFilePathFromPutCommand("PUT file://" + s_filePathWithSpaces + "  @TestStage", s_filePathWithSpaces);
         }
 
         [Test]
@@ -660,7 +660,7 @@ public void TestGetFilePathWithoutSpacesAndWithSingleQuotesFromPutCommand()
         [Test]
         public void TestGetFilePathWithSpacesAndWithSingleQuotesFromPutCommand()
         {
-            TestGetFilePathFromPutCommand("PUT 'file://" + s_filePathWithSpaces + "' @TestStage", s_filePathWithSpaces);
+            TestGetFilePathFromPutCommand("PUT 'file://" + s_filePathWithSpaces + "'  @TestStage", s_filePathWithSpaces);
         }
 
         public void TestGetFilePathFromPutCommand(string query, string expectedFilePath)

diff --git a/Snowflake.Data/Core/FileTransfer/SFFileTransferAgent.cs b/Snowflake.Data/Core/FileTransfer/SFFileTransferAgent.cs
@@ -533,7 +533,7 @@ internal static string getFilePathFromPutCommand(string query)
             // E.g. "PUT file://C:<path-to-file> @DB.SCHEMA.%TABLE;"
             int startIndex = query.IndexOf("file://") + "file://".Length;
             int endIndex = query.Substring(startIndex).IndexOf('@') - 1;
-            string filePath = query.Substring(startIndex, endIndex);
+            string filePath = query.Substring(startIndex, endIndex).Trim();
 
             // Check if file path contains an enclosing (') char
             if (filePath[filePath.Length - 1] == '\'')