Skip to content

Commit 0678fe0

Browse files
committed
fix compile error
Signed-off-by: Lantao Jin <ltjin@amazon.com>
1 parent 79cdc1c commit 0678fe0

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

opensearch/src/test/java/org/opensearch/sql/opensearch/storage/scan/OpenSearchIndexScanTest.java

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -347,7 +347,7 @@ void query_some_results_with_scroll() {
347347
void close_mid_pagination_without_cursor_serialized_should_force_cleanup() {
348348
var request = mock(OpenSearchRequest.class);
349349
when(request.hasAnotherBatch()).thenReturn(true);
350-
var indexScan = new OpenSearchIndexScan(client, request);
350+
var indexScan = new OpenSearchIndexScan(client, 10, request);
351351
indexScan.close();
352352
verify(client).forceCleanup(request);
353353
verify(client, never()).cleanup(any());
@@ -362,7 +362,7 @@ void close_mid_pagination_without_cursor_serialized_should_force_cleanup() {
362362
void close_mid_pagination_with_cursor_serialized_should_cleanup() {
363363
var request = mock(OpenSearchRequest.class);
364364
when(request.hasAnotherBatch()).thenReturn(true);
365-
var indexScan = new OpenSearchIndexScan(client, request);
365+
var indexScan = new OpenSearchIndexScan(client, 10, request);
366366

367367
// Simulate successful cursor serialization by calling writeExternal
368368
var out = mock(ObjectOutput.class);
@@ -377,7 +377,7 @@ void close_mid_pagination_with_cursor_serialized_should_cleanup() {
377377
@Test
378378
void forceClose_should_always_force_cleanup() {
379379
var request = mock(OpenSearchRequest.class);
380-
var indexScan = new OpenSearchIndexScan(client, request);
380+
var indexScan = new OpenSearchIndexScan(client, 10, request);
381381
indexScan.forceClose();
382382
verify(client).forceCleanup(request);
383383
}

0 commit comments

Comments
 (0)