potential test improvement and more debug assert

Author: sfc-gh-zyao

src/snowflake/connector/direct_file_operation_utils.py

@@ -78,7 +78,9 @@ def parse_file_operation(
             ret = cursor._execute_helper(
                 sql, binding_params=processed_params, is_internal=True
             )
-            assert ret.get("success", False), f"parse failed, ret is {ret}"
+            assert ret.get(
+                "success", False
+            ), f"parse failed, ret is {ret}, local_file_name is {local_file_name}, stage_location is {stage_location}"
             return ret
 
 

test/integ/test_direct_file_operation_utils.py

@@ -23,7 +23,7 @@
 
 
 def _normalize_windows_local_path(path):
-    return path.replace("\\", "\\\\")
+    return path.replace("\\", "\\\\").replace("'", "\\'")
 
 
 def _validate_upload_content(
@@ -96,7 +96,7 @@ def test_upload(
 ):
     def upload_task(cursor, stage_name, temp_dir, base_file_name):
         cursor._upload(
-            local_file_name=f"file://{_normalize_windows_local_path(os.path.join(temp_dir, base_file_name))}",
+            local_file_name=f"'file://{_normalize_windows_local_path(os.path.join(temp_dir, base_file_name))}'",
             stage_location=f"@{stage_name}",
             options={"auto_compress": is_compressed},
         )