Skip to content

Commit 67d1871

Browse files
test: add test for null field cases with BigIntVector
1 parent 129c65d commit 67d1871

1 file changed

Lines changed: 7 additions & 1 deletion

File tree

src/test/java/com/influxdb/v3/client/TestUtils.java

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -35,6 +35,7 @@
3535
import org.apache.arrow.flight.Ticket;
3636
import org.apache.arrow.memory.BufferAllocator;
3737
import org.apache.arrow.memory.RootAllocator;
38+
import org.apache.arrow.vector.BigIntVector;
3839
import org.apache.arrow.vector.VarCharVector;
3940
import org.apache.arrow.vector.VectorSchemaRoot;
4041
import org.apache.arrow.vector.types.pojo.ArrowType;
@@ -94,8 +95,9 @@ public static VectorSchemaRoot generateVectorSchemaRoot(final int fieldCount, fi
9495
public static VectorSchemaRoot generateVectorSchemaRootWithNull() {
9596
Field normalField = new Field("normalField", FieldType.nullable(new ArrowType.Utf8()), null);
9697
Field nullField = new Field("nullField", FieldType.nullable(new ArrowType.Utf8()), null);
98+
Field nullField1 = new Field("nullField1", FieldType.nullable(new ArrowType.Int(64, true)), null);
9799

98-
Schema schema = new Schema(List.of(normalField, nullField));
100+
Schema schema = new Schema(List.of(normalField, nullField, nullField1));
99101
VectorSchemaRoot vectorSchemaRoot = VectorSchemaRoot.create(schema, new RootAllocator(Long.MAX_VALUE));
100102

101103
VarCharVector vector = (VarCharVector) vectorSchemaRoot.getVector(normalField);
@@ -106,6 +108,10 @@ public static VectorSchemaRoot generateVectorSchemaRootWithNull() {
106108
vector.allocateNew(1);
107109
vector.setNull(0);
108110

111+
BigIntVector intVector = (BigIntVector) vectorSchemaRoot.getVector(nullField1);
112+
intVector.allocateNew(1);
113+
intVector.setNull(0);
114+
109115
vectorSchemaRoot.setRowCount(1);
110116
return vectorSchemaRoot;
111117
}

0 commit comments

Comments
 (0)