This is an automated email from the ASF dual-hosted git repository. Wei-hao-Li pushed a commit to branch mppEx in repository https://gitbox.apache.org/repos/asf/iotdb.git
commit 36cc05dc6a6649d14ab3a837a7130bee6cf35ba4 Author: Weihao Li <[email protected]> AuthorDate: Wed Sep 16 10:05:34 2026 +0800 fix UT Signed-off-by: Weihao Li <[email protected]> --- .../iotdb/db/queryengine/execution/exchange/SourceHandleTest.java | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/iotdb-core/datanode/src/test/java/org/apache/iotdb/db/queryengine/execution/exchange/SourceHandleTest.java b/iotdb-core/datanode/src/test/java/org/apache/iotdb/db/queryengine/execution/exchange/SourceHandleTest.java index 03220a3d10b..2e8c8a8c9d3 100644 --- a/iotdb-core/datanode/src/test/java/org/apache/iotdb/db/queryengine/execution/exchange/SourceHandleTest.java +++ b/iotdb-core/datanode/src/test/java/org/apache/iotdb/db/queryengine/execution/exchange/SourceHandleTest.java @@ -638,9 +638,11 @@ public class SourceHandleTest { for (int i = 0; i < request.getEndSequenceId() - request.getStartSequenceId() - 1; i++) { - shortResponse.add(ByteBuffer.allocate(0)); + shortResponse.add(ByteBuffer.allocate(1)); } - return new TGetDataBlockResponse(shortResponse); + TGetDataBlockResponse response = new TGetDataBlockResponse(shortResponse); + response.setOffset(1); + return response; }) .when(mockClient) .getDataBlock(Mockito.any(TGetDataBlockRequest.class)); @@ -676,7 +678,7 @@ public class SourceHandleTest { Assert.fail(e.getMessage()); } Mockito.verify(mockSourceHandleListener, Mockito.timeout(10_000).times(1)) - .onFailure(Mockito.eq(sourceHandle), Mockito.any(TException.class)); + .onFailure(Mockito.eq(sourceHandle), Mockito.any(Throwable.class)); Assert.assertFalse(blocked.isDone()); Assert.assertEquals(0L, sourceHandle.getBufferRetainedSizeInBytes());
