SNOW-2044853: Fix XML reader test in stored proc (#3345)

Author: sfc-gh-jdu

tests/integ/scala/test_dataframe_reader_suite.py

@@ -136,6 +136,9 @@ def get_df_from_reader_and_file_format(reader, file_format):
 
 @pytest.fixture(scope="module", autouse=True)
 def setup(session, resources_path, local_testing_mode):
+    # TODO SNOW-2098847: remove this workaround after fixing the issue
+    session._use_scoped_temp_objects = False
+
     test_files = TestFiles(resources_path)
     if not local_testing_mode:
         Utils.create_stage(session, tmp_stage_name1, is_temporary=True)
@@ -1971,10 +1974,6 @@ def test_read_multiple_csvs(session):
     "config.getoption('local_testing_mode', default=False)",
     reason="xml not supported in local testing mode",
 )
-@pytest.mark.skipif(
-    IS_IN_STORED_PROC,
-    reason="SNOW-2044853: Flaky in stored procedure test",
-)
 @pytest.mark.parametrize(
     "file,row_tag,expected_row_count,expected_column_count",
     [
@@ -1997,10 +1996,6 @@ def test_read_xml_row_tag(
     "config.getoption('local_testing_mode', default=False)",
     reason="xml not supported in local testing mode",
 )
-@pytest.mark.skipif(
-    IS_IN_STORED_PROC,
-    reason="SNOW-2044853: Flaky in stored procedure test",
-)
 def test_read_xml_no_xxe(session):
     row_tag = "bar"
     stage_file_path = f"@{tmp_stage_name1}/{test_file_xxe_xml}"
@@ -2012,10 +2007,6 @@ def test_read_xml_no_xxe(session):
     "config.getoption('local_testing_mode', default=False)",
     reason="xml not supported in local testing mode",
 )
-@pytest.mark.skipif(
-    IS_IN_STORED_PROC,
-    reason="SNOW-2044853: Flaky in stored procedure test",
-)
 def test_read_xml_query_nested_data(session):
     row_tag = "tag"
     df = session.read.option("rowTag", row_tag).xml(
@@ -2046,10 +2037,6 @@ def test_read_xml_non_existing_file(session):
     "config.getoption('local_testing_mode', default=False)",
     reason="xml not supported in local testing mode",
 )
-@pytest.mark.skipif(
-    IS_IN_STORED_PROC,
-    reason="SNOW-2044853: Flaky in stored procedure test",
-)
 @pytest.mark.parametrize(
     "file",
     (
@@ -2060,12 +2047,13 @@ def test_read_xml_non_existing_file(session):
 )
 def test_read_malformed_xml(session, file):
     row_tag = "record"
+    file_path = f"@{tmp_stage_name1}/{file}"
 
     # permissive mode
     df = (
         session.read.option("rowTag", row_tag)
         .option("mode", "permissive")
-        .xml(f"@{tmp_stage_name1}/{file}")
+        .xml(file_path)
     )
     result = df.collect()
     assert len(result) == 2
@@ -2079,17 +2067,15 @@ def test_read_malformed_xml(session, file):
     df = (
         session.read.option("rowTag", row_tag)
         .option("mode", "dropmalformed")
-        .xml(f"@{tmp_stage_name1}/{test_file_malformed_no_closing_tag_xml}")
+        .xml(file_path)
     )
     result = df.collect()
     assert len(result) == 1
     assert len(result[0]) == 3
 
     # failfast mode
     df = (
-        session.read.option("rowTag", row_tag)
-        .option("mode", "failfast")
-        .xml(f"@{tmp_stage_name1}/{test_file_malformed_no_closing_tag_xml}")
+        session.read.option("rowTag", row_tag).option("mode", "failfast").xml(file_path)
     )
     with pytest.raises(SnowparkSQLException, match="Malformed XML record at bytes"):
         df.collect()