Add arrow tests (#1291)

* SNOW-748294
snowflakedb · Mar 14, 2023 · 6a7bfef · 6a7bfef
1 parent 7cd122d
commit 6a7bfef
Show file tree

Hide file tree

Showing 4 changed files with 407 additions and 33 deletions.
diff --git a/src/main/java/net/snowflake/client/jdbc/ArrowResultChunk.java b/src/main/java/net/snowflake/client/jdbc/ArrowResultChunk.java
@@ -443,7 +443,7 @@ public int getCurrentRowInRecordBatch() {
    * merge arrow result chunk with more than one batches into one record batch (Only used for the
    * first chunk when client side sorting is required)
    */
-  private void mergeBatchesIntoOne() throws SnowflakeSQLException {
+  public void mergeBatchesIntoOne() throws SnowflakeSQLException {
     try {
       List<ValueVector> first = batchOfVectors.get(0);
       for (int i = 1; i < batchOfVectors.size(); i++) {

diff --git a/src/main/java/net/snowflake/client/jdbc/SnowflakeResultSetSerializableV1.java b/src/main/java/net/snowflake/client/jdbc/SnowflakeResultSetSerializableV1.java
@@ -251,7 +251,7 @@ public void setChunkFileCount(int chunkFileCount) {
     this.chunkFileCount = chunkFileCount;
   }
 
-  public void setFristChunkStringData(String firstChunkStringData) {
+  public void setFirstChunkStringData(String firstChunkStringData) {
     this.firstChunkStringData = firstChunkStringData;
   }