debug long test time

Author: sfc-gh-akolodziejczyk

src/test/java/net/snowflake/client/jdbc/SnowflakeResultSetSerializableIT.java

@@ -10,20 +10,27 @@
 import java.io.File;
 import java.io.FileInputStream;
 import java.io.FileOutputStream;
+import java.io.FileWriter;
 import java.io.ObjectInputStream;
 import java.io.ObjectOutputStream;
 import java.sql.Connection;
 import java.sql.ResultSet;
 import java.sql.ResultSetMetaData;
 import java.sql.SQLException;
 import java.sql.Statement;
+import java.text.SimpleDateFormat;
 import java.util.ArrayList;
+import java.util.Date;
 import java.util.List;
 import java.util.Properties;
+import java.util.UUID;
+import java.util.concurrent.TimeUnit;
+import java.util.logging.Logger;
 import javax.annotation.Nullable;
 import net.snowflake.client.annotations.DontRunOnGithubActions;
 import net.snowflake.client.category.TestTags;
 import net.snowflake.client.providers.SimpleResultFormatProvider;
+import org.apache.commons.lang3.time.StopWatch;
 import org.junit.jupiter.api.Disabled;
 import org.junit.jupiter.api.Tag;
 import org.junit.jupiter.api.Test;
@@ -34,6 +41,8 @@
 /** SnowflakeResultSetSerializable tests */
 @Tag(TestTags.RESULT_SET)
 public class SnowflakeResultSetSerializableIT extends BaseJDBCTest {
+  private static Logger logger = Logger.getLogger(HeartbeatAsyncLatestIT.class.getName());
+
   @TempDir private File tmpFolder;
 
   private static boolean developPrint = false;
@@ -242,6 +251,9 @@ private void testBasicTableHarness(
     String originalResultCSVString = null;
     try (Connection connection = init(queryResultFormat)) {
       Statement statement = connection.createStatement();
+      StopWatch stopwatch = new StopWatch();
+      stopwatch.start();
+      logger.severe(stopwatch.getTime(TimeUnit.SECONDS) + " ms: testBasicTableHarness start");
 
       if (developPrint) {
         System.out.println(
@@ -261,14 +273,44 @@ private void testBasicTableHarness(
         statement.execute(
             "create or replace table table_basic " + " (int_c int, string_c string(128))");
 
+        logger.severe(
+            stopwatch.getTime(TimeUnit.SECONDS)
+                + " ms: testBasicTableHarness before insert into table_basic");
         if (rowCount > 0) {
-          statement.execute(
-              "insert into table_basic select "
-                  + "seq4(), 'arrow_1234567890arrow_1234567890arrow_1234567890arrow_1234567890'"
-                  + " from table(generator(rowcount=>"
-                  + rowCount
-                  + "))");
+          File csvFile = File.createTempFile("snowflake_data_", ".csv");
+          String randomSuffix = UUID.randomUUID().toString().replace("-", "").substring(0, 8);
+          String tmpStageName = "temp_csv_stage_" + randomSuffix;
+          String fileFormatName = "my_csv_format_" + randomSuffix;
+          try (FileWriter writer = new FileWriter(csvFile)) {
+            writer.write("id,value_column\n");
+            for (int i = 1; i <= rowCount; i++) {
+              writer.write(
+                  i + ",arrow_1234567890arrow_1234567890arrow_1234567890arrow_1234567890\n");
+            }
+          }
+          try (Statement stmt = connection.createStatement()) {
+            stmt.execute("CREATE TEMPORARY STAGE " + tmpStageName);
+            stmt.execute(
+                "CREATE OR REPLACE FILE FORMAT "
+                    + fileFormatName
+                    + " TYPE = 'CSV' FIELD_DELIMITER = ',' SKIP_HEADER = 1");
+            String putCmd =
+                "PUT file://"
+                    + csvFile.getAbsolutePath().replace("\\", "/")
+                    + " @"
+                    + tmpStageName
+                    + " AUTO_COMPRESS=TRUE OVERWRITE=TRUE";
+            stmt.execute(putCmd);
+            stmt.execute(
+                "COPY INTO table_basic FROM @" + tmpStageName + " FILE_FORMAT = " + fileFormatName);
+            stmt.execute("DROP STAGE IF EXISTS " + tmpStageName);
+            stmt.execute("DROP FILE FORMAT IF EXISTS " + fileFormatName);
+          }
+          csvFile.delete();
         }
+        logger.severe(
+            stopwatch.getTime(TimeUnit.SECONDS)
+                + " ms: testBasicTableHarness after insert into table_basic");
       }
 
       String sqlSelect = "select * from table_basic " + whereClause;
@@ -277,16 +319,29 @@ private void testBasicTableHarness(
               ? statement.unwrap(SnowflakeStatement.class).executeAsyncQuery(sqlSelect)
               : statement.executeQuery(sqlSelect)) {
 
+        logger.severe(
+            stopwatch.getTime(TimeUnit.SECONDS) + " ms: testBasicTableHarness after select query");
         fileNameList = serializeResultSet((SnowflakeResultSet) rs, maxSizeInBytes, "txt");
+        logger.severe(
+            stopwatch.getTime(TimeUnit.SECONDS)
+                + " ms: testBasicTableHarness after serializeResultSet");
 
         originalResultCSVString = generateCSVResult(rs);
+        logger.severe(
+            stopwatch.getTime(TimeUnit.SECONDS)
+                + " ms: testBasicTableHarness after whole testcase");
       }
     }
 
     String chunkResultString = deserializeResultSet(fileNameList);
     assertEquals(chunkResultString, originalResultCSVString);
   }
 
+  private String generateRandomSuffix() {
+    SimpleDateFormat sdf = new SimpleDateFormat("yyyyMMddHHmmssSSS");
+    return sdf.format(new Date());
+  }
+
   @ParameterizedTest
   @ArgumentsSource(SimpleResultFormatProvider.class)
   @DontRunOnGithubActions