more test

Author: ViggoC

flight/flight-sql-jdbc-core/src/test/java/org/apache/arrow/driver/jdbc/accessor/ArrowFlightJdbcAccessorFactoryTest.java

@@ -46,6 +46,8 @@
 import org.apache.arrow.vector.LargeVarCharVector;
 import org.apache.arrow.vector.ValueVector;
 import org.apache.arrow.vector.VarCharVector;
+import org.apache.arrow.vector.ViewVarBinaryVector;
+import org.apache.arrow.vector.ViewVarCharVector;
 import org.apache.arrow.vector.complex.DenseUnionVector;
 import org.apache.arrow.vector.complex.MapVector;
 import org.apache.arrow.vector.complex.StructVector;
@@ -239,6 +241,18 @@ public void createAccessorForFixedSizeBinaryVector() {
     }
   }
 
+  @Test
+  public void createAccessorForViewVarBinaryVector() {
+    try (ValueVector valueVector =
+        new ViewVarBinaryVector("", rootAllocatorTestExtension.getRootAllocator())) {
+      ArrowFlightJdbcAccessor accessor =
+          ArrowFlightJdbcAccessorFactory.createAccessor(
+              valueVector, GET_CURRENT_ROW, (boolean wasNull) -> {});
+
+      assertTrue(accessor instanceof ArrowFlightJdbcBinaryVectorAccessor);
+    }
+  }
+
   @Test
   public void createAccessorForTimeStampVector() {
     try (ValueVector valueVector = rootAllocatorTestExtension.createTimeStampMilliVector()) {
@@ -340,6 +354,18 @@ public void createAccessorForLargeVarCharVector() {
     }
   }
 
+  @Test
+  public void createAccessorForViewVarCharVector() {
+    try (ValueVector valueVector =
+        new ViewVarCharVector("", rootAllocatorTestExtension.getRootAllocator())) {
+      ArrowFlightJdbcAccessor accessor =
+          ArrowFlightJdbcAccessorFactory.createAccessor(
+              valueVector, GET_CURRENT_ROW, (boolean wasNull) -> {});
+
+      assertTrue(accessor instanceof ArrowFlightJdbcVarCharVectorAccessor);
+    }
+  }
+
   @Test
   public void createAccessorForDurationVector() {
     try (ValueVector valueVector =

flight/flight-sql-jdbc-core/src/test/java/org/apache/arrow/driver/jdbc/accessor/impl/text/ArrowFlightJdbcVarCharVectorAccessorTest.java

@@ -24,6 +24,8 @@
 import static org.hamcrest.CoreMatchers.instanceOf;
 import static org.hamcrest.MatcherAssert.assertThat;
 import static org.junit.jupiter.api.Assertions.assertArrayEquals;
+import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertNull;
 import static org.junit.jupiter.api.Assertions.assertThrows;
 import static org.mockito.Mockito.when;
 
@@ -46,6 +48,8 @@
 import org.apache.arrow.vector.DateMilliVector;
 import org.apache.arrow.vector.TimeMilliVector;
 import org.apache.arrow.vector.TimeStampVector;
+import org.apache.arrow.vector.VarCharVector;
+import org.apache.arrow.vector.ViewVarCharVector;
 import org.apache.arrow.vector.util.Text;
 import org.junit.jupiter.api.BeforeEach;
 import org.junit.jupiter.api.Test;
@@ -695,4 +699,26 @@ public void testShouldGetObjectClassReturnString() {
     final Class<?> clazz = accessor.getObjectClass();
     assertThat(clazz, equalTo(String.class));
   }
+
+  @Test
+  public void testViewVarcharVector() throws Exception {
+    try (VarCharVector varCharVector =
+            new VarCharVector("", rootAllocatorTestExtension.getRootAllocator());
+        ViewVarCharVector viewVarCharVector =
+            new ViewVarCharVector("", rootAllocatorTestExtension.getRootAllocator())) {
+      varCharVector.allocateNew(1);
+      viewVarCharVector.allocateNew(1);
+
+      ArrowFlightJdbcVarCharVectorAccessor varCharVectorAccessor =
+          new ArrowFlightJdbcVarCharVectorAccessor(varCharVector, () -> 0, (boolean wasNull) -> {});
+      ArrowFlightJdbcVarCharVectorAccessor viewVarcharVectorAccessor =
+          new ArrowFlightJdbcVarCharVectorAccessor(
+              viewVarCharVector, () -> 0, (boolean wasNull) -> {});
+      assertNull(viewVarcharVectorAccessor.getString());
+
+      varCharVector.set(0, new Text("looooong_string"));
+      viewVarCharVector.set(0, new Text("looooong_string"));
+      assertEquals(varCharVectorAccessor.getString(), viewVarcharVectorAccessor.getString());
+    }
+  }
 }