SNOW-1782852: Testing arrow too many rows

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

@@ -3,7 +3,14 @@
  */
 package net.snowflake.client.jdbc;
 
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertTrue;
+
+import java.sql.ResultSet;
+import java.sql.SQLException;
+import java.sql.Statement;
 import net.snowflake.client.category.TestCategoryArrow;
+import org.junit.Test;
 import org.junit.experimental.categories.Category;
 
 /**
@@ -15,4 +22,27 @@ public class ResultSetArrowLatestIT extends ResultSetLatestIT {
   public ResultSetArrowLatestIT() {
     super("arrow");
   }
+
+  // TODO move to parent - for now we only want to test ARROW
+  @Test
+  public void testChunkedResultSetHasExpectedNumberOfRows() throws SQLException {
+    int rowsToGet = 8295305;
+    int count = 0;
+    long seq8 = -1;
+    try (Statement statement = connection.createStatement();
+        ResultSet resultSet =
+            statement.executeQuery(
+                "select randstr(20,random()), seq8() as s from table(generator(rowcount=>"
+                    + rowsToGet
+                    + ")) where 1=1 and s >= 0 "
+                    + "order by s")) {
+      while (resultSet.next()) {
+        count++;
+        long newSeq8 = resultSet.getLong(2);
+        assertTrue(seq8 < newSeq8);
+        seq8 = newSeq8;
+      }
+    }
+    assertEquals(rowsToGet, count);
+  }
 }