Add large query integration test

Author: madhav-db

src/test/java/com/databricks/jdbc/integration/e2e/LargeQueryExecutionTests.java

@@ -10,41 +10,41 @@
 import org.junit.jupiter.api.Test;
 
 public class LargeQueryExecutionTests {
-    @Test
-    void testQueryYieldingLargeNarrowResultSet() throws SQLException {
-        String largeQuerySQL =
-                "SELECT * FROM range(37500000)"; // ~300MB of data
-        ResultSet rs = executeQuery(largeQuerySQL);
-        int rows = 0;
-        while (rs != null && rs.next()) {
-            rows++;
-        }
-        assertEquals(37500000, rows, "Expected 37500000 rows, got " + rows);
+  @Test
+  void testQueryYieldingLargeNarrowResultSet() throws SQLException {
+    String largeQuerySQL = "SELECT * FROM range(37500000)"; // ~300MB of data
+    ResultSet rs = executeQuery(largeQuerySQL);
+    int rows = 0;
+    while (rs != null && rs.next()) {
+      rows++;
     }
+    assertEquals(37500000, rows, "Expected 37500000 rows, got " + rows);
+  }
 
-    @Test
-    void testQueryYieldingLargeWideResultSet() throws SQLException {
-        int resultSize = 300 * 1000 * 100; // 30 MB
-        int width = 8192; // B
-        int rows = resultSize / width;
-        int cols = width / 36;
+  @Test
+  void testQueryYieldingLargeWideResultSet() throws SQLException {
+    int resultSize = 300 * 1000 * 100; // 30 MB
+    int width = 8192; // B
+    int rows = resultSize / width;
+    int cols = width / 36;
 
-        // Generate the UUID columns
-        String uuids = IntStream.rangeClosed(0, cols)
-                .mapToObj(i -> "uuid() uuid" + i)
-                .collect(Collectors.joining(", "));
+    // Generate the UUID columns
+    String uuids =
+        IntStream.rangeClosed(0, cols)
+            .mapToObj(i -> "uuid() uuid" + i)
+            .collect(Collectors.joining(", "));
 
-        // Create the SQL query
-        String query = String.format("SELECT id, %s FROM RANGE(%d)", uuids, rows);
-        System.out.println("query: " + query);
-        ResultSet rs = executeQuery(query);
-        int rowCount = 0;
-        while (rs != null && rs.next()) {
-            System.out.println("check");
-            assertEquals(rs.getInt("id"), rowCount, "Expected id to be equal to row number");
-            assertEquals(rs.getString("uuid0").length(), 36, "Expected UUID length of 36");
-            rowCount++;
-        }
-        assertEquals(rows, rowCount, "Expected " + rows + " rows, got " + rowCount);
+    // Create the SQL query
+    String query = String.format("SELECT id, %s FROM RANGE(%d)", uuids, rows);
+    System.out.println("query: " + query);
+    ResultSet rs = executeQuery(query);
+    int rowCount = 0;
+    while (rs != null && rs.next()) {
+      System.out.println("check");
+      assertEquals(rs.getInt("id"), rowCount, "Expected id to be equal to row number");
+      assertEquals(rs.getString("uuid0").length(), 36, "Expected UUID length of 36");
+      rowCount++;
     }
+    assertEquals(rows, rowCount, "Expected " + rows + " rows, got " + rowCount);
+  }
 }