From 56bc2dd4a9e86548689d3c8481e393c03e3c78a8 Mon Sep 17 00:00:00 2001
From: Vibhatha Abeykoon <vibhatha@gmail.com>
Date: Wed, 5 Jun 2024 04:49:39 +0530
Subject: [PATCH] fix: adding multi buffer test

---
 .../org/apache/arrow/c/RoundtripTest.java     | 30 ++++++++++++++++---
 1 file changed, 26 insertions(+), 4 deletions(-)

diff --git a/java/c/src/test/java/org/apache/arrow/c/RoundtripTest.java b/java/c/src/test/java/org/apache/arrow/c/RoundtripTest.java
index cf226e68e1f22..9a24fd6c9e3fa 100644
--- a/java/c/src/test/java/org/apache/arrow/c/RoundtripTest.java
+++ b/java/c/src/test/java/org/apache/arrow/c/RoundtripTest.java
@@ -520,31 +520,53 @@ public void testVarBinaryVector() {
 
   @Test
   public void testViewVector() {
-    // VarCharViewVector with short strings
+    // ViewVarCharVector with short strings
     try (final ViewVarCharVector vector = new ViewVarCharVector("v1", allocator)) {
       setVector(vector, "abc".getBytes(StandardCharsets.UTF_8), "def".getBytes(StandardCharsets.UTF_8), null);
       assertTrue(roundtrip(vector, ViewVarCharVector.class));
     }
 
-    // VarCharViewVector with long strings
+    // ViewVarCharVector with long strings
     try (final ViewVarCharVector vector = new ViewVarCharVector("v2", allocator)) {
       setVector(vector, "01234567890123".getBytes(StandardCharsets.UTF_8),
           "01234567890123567".getBytes(StandardCharsets.UTF_8), null);
       assertTrue(roundtrip(vector, ViewVarCharVector.class));
     }
 
-    // VarBinaryViewVector with short values
+    // ViewVarBinaryVector with short values
     try (final ViewVarBinaryVector vector = new ViewVarBinaryVector("v3", allocator)) {
       setVector(vector, "abc".getBytes(StandardCharsets.UTF_8), "def".getBytes(StandardCharsets.UTF_8), null);
       assertTrue(roundtrip(vector, ViewVarBinaryVector.class));
     }
 
-    // VarBinaryViewVector with long values
+    // ViewVarBinaryVector with long values
     try (final ViewVarBinaryVector vector = new ViewVarBinaryVector("v4", allocator)) {
       setVector(vector, "01234567890123".getBytes(StandardCharsets.UTF_8),
           "01234567890123567".getBytes(StandardCharsets.UTF_8), null);
       assertTrue(roundtrip(vector, ViewVarBinaryVector.class));
     }
+
+    List<byte[]> byteArrayList = new ArrayList<>();
+    for (int i = 1; i <= 500; i++) {
+      StringBuilder sb = new StringBuilder(i);
+      for (int j = 0; j < i; j++) {
+        sb.append(j); // or any other character
+      }
+      byte[] bytes = sb.toString().getBytes(StandardCharsets.UTF_8);
+      byteArrayList.add(bytes);
+    }
+
+    // ViewVarCharVector with short long strings with multiple data buffers
+    try (final ViewVarCharVector vector = new ViewVarCharVector("v5", allocator)) {
+      setVector(vector, byteArrayList.toArray(new byte[0][]));
+      assertTrue(roundtrip(vector, ViewVarCharVector.class));
+    }
+
+    // ViewVarBinaryVector with short long strings with multiple data buffers
+    try (final ViewVarBinaryVector vector = new ViewVarBinaryVector("v6", allocator)) {
+      setVector(vector, byteArrayList.toArray(new byte[0][]));
+      assertTrue(roundtrip(vector, ViewVarBinaryVector.class));
+    }
   }
 
   @Test