Skip to content

Commit 818fa7b

Browse files
committed
Fix OffsetOperatorTest null TsBlock handling
1 parent 66d0c07 commit 818fa7b

1 file changed

Lines changed: 12 additions & 0 deletions

File tree

iotdb-core/datanode/src/test/java/org/apache/iotdb/db/queryengine/execution/operator/OffsetOperatorTest.java

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -169,6 +169,9 @@ public void batchTest1() throws Exception {
169169
int count = 100;
170170
while (limitOperator.isBlocked().isDone() && limitOperator.hasNext()) {
171171
TsBlock tsBlock = limitOperator.next();
172+
if (tsBlock == null) {
173+
continue;
174+
}
172175
assertEquals(2, tsBlock.getValueColumnCount());
173176
assertTrue(tsBlock.getColumn(0) instanceof IntColumn);
174177
assertTrue(tsBlock.getColumn(1) instanceof IntColumn);
@@ -277,6 +280,9 @@ public void batchTest2() throws Exception {
277280
int count = 0;
278281
while (offsetOperator.isBlocked().isDone() && offsetOperator.hasNext()) {
279282
TsBlock tsBlock = offsetOperator.next();
283+
if (tsBlock == null) {
284+
continue;
285+
}
280286
assertEquals(2, tsBlock.getValueColumnCount());
281287
assertTrue(tsBlock.getColumn(0) instanceof IntColumn);
282288
assertTrue(tsBlock.getColumn(1) instanceof IntColumn);
@@ -383,6 +389,9 @@ public void batchTest3() throws Exception {
383389

384390
while (offsetOperator.isBlocked().isDone() && offsetOperator.hasNext()) {
385391
TsBlock tsBlock = offsetOperator.next();
392+
if (tsBlock == null) {
393+
continue;
394+
}
386395
assertEquals(2, tsBlock.getValueColumnCount());
387396
assertTrue(tsBlock.getColumn(0) instanceof IntColumn);
388397
assertTrue(tsBlock.getColumn(1) instanceof IntColumn);
@@ -475,6 +484,9 @@ public void batchTest4() throws Exception {
475484

476485
while (offsetOperator.isBlocked().isDone() && offsetOperator.hasNext()) {
477486
TsBlock tsBlock = offsetOperator.next();
487+
if (tsBlock == null) {
488+
continue;
489+
}
478490
assertEquals(2, tsBlock.getValueColumnCount());
479491
assertTrue(tsBlock.getColumn(0) instanceof IntColumn);
480492
assertTrue(tsBlock.getColumn(1) instanceof IntColumn);

0 commit comments

Comments
 (0)