Update tests for GetTables -- start refactor to use proper schemas

rafael-telles · Apr 20, 2022 · 7e9498f · 7e9498f
1 parent ab8f7b6
commit 7e9498f
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/java/flight/flight-sql/src/test/java/org/apache/arrow/flight/TestFlightSql.java b/java/flight/flight-sql/src/test/java/org/apache/arrow/flight/TestFlightSql.java
@@ -785,8 +785,12 @@ List<List<String>> getResults(FlightStream stream) {
               }
             } else if (fieldVector instanceof IntVector) {
               for (int rowIndex = 0; rowIndex < rowCount; rowIndex++) {
-                Object data = fieldVector.getObject(rowIndex);
-                results.get(rowIndex).add(isNull(data) ? null : Objects.toString(data));
+                try {
+                  results.get(rowIndex).add(String.valueOf(((IntVector) fieldVector).get(rowIndex)));
+                } catch (IllegalStateException e) {
+                  System.out.println(("Failed at index " + rowIndex));
+                  throw e;
+                }
               }
             } else if (fieldVector instanceof VarBinaryVector) {
               final VarBinaryVector varbinaryVector = (VarBinaryVector) fieldVector;