(pinot) branch master updated: Fix test failures introduced in #15869 (#15893)

[jackie]

This is an automated email from the ASF dual-hosted git repository.

jackie pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/pinot.git


The following commit(s) were added to refs/heads/master by this push:
     new 5383ec7249 Fix test failures introduced in #15869 (#15893)
5383ec7249 is described below

commit 5383ec7249c17fdfac4874ea961c5138e8b67fa8
Author: Xiaotian (Jackie) Jiang <17555551+jackie-ji...@users.noreply.github.com>
AuthorDate: Fri May 23 16:01:44 2025 -0600

    Fix test failures introduced in #15869 (#15893)
---
 .../apache/pinot/core/operator/combine/CombineSlowOperatorsTest.java  | 3 ++-
 .../org/apache/pinot/core/query/prefetch/DefaultFetchPlannerTest.java | 4 ++--
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git 
a/pinot-core/src/test/java/org/apache/pinot/core/operator/combine/CombineSlowOperatorsTest.java
 
b/pinot-core/src/test/java/org/apache/pinot/core/operator/combine/CombineSlowOperatorsTest.java
index 89dba4b504..b4d62f9ca6 100644
--- 
a/pinot-core/src/test/java/org/apache/pinot/core/operator/combine/CombineSlowOperatorsTest.java
+++ 
b/pinot-core/src/test/java/org/apache/pinot/core/operator/combine/CombineSlowOperatorsTest.java
@@ -47,6 +47,7 @@ import org.testng.annotations.AfterClass;
 import org.testng.annotations.BeforeClass;
 import org.testng.annotations.Test;
 
+import static org.mockito.ArgumentMatchers.any;
 import static org.mockito.ArgumentMatchers.anyString;
 import static org.mockito.Mockito.mock;
 import static org.mockito.Mockito.when;
@@ -138,7 +139,7 @@ public class CombineSlowOperatorsTest {
       DataSourceMetadata dsmd = mock(DataSourceMetadata.class);
       when(dsmd.getMinValue()).thenReturn(100L);
       when(dsmd.getMaxValue()).thenReturn(200L);
-      when(seg.getDataSource(anyString())).thenReturn(ds);
+      when(seg.getDataSource(anyString(), any())).thenReturn(ds);
       when(ds.getDataSourceMetadata()).thenReturn(dsmd);
       return seg;
     });
diff --git 
a/pinot-core/src/test/java/org/apache/pinot/core/query/prefetch/DefaultFetchPlannerTest.java
 
b/pinot-core/src/test/java/org/apache/pinot/core/query/prefetch/DefaultFetchPlannerTest.java
index 7b16059898..2cef552f27 100644
--- 
a/pinot-core/src/test/java/org/apache/pinot/core/query/prefetch/DefaultFetchPlannerTest.java
+++ 
b/pinot-core/src/test/java/org/apache/pinot/core/query/prefetch/DefaultFetchPlannerTest.java
@@ -68,10 +68,10 @@ public class DefaultFetchPlannerTest {
     QueryContext queryContext = 
QueryContextConverterUtils.getQueryContext(query);
     // No Bloomfilter for those columns.
     DataSource ds0 = mock(DataSource.class);
-    when(indexSegment.getDataSource("c0")).thenReturn(ds0);
+    when(indexSegment.getDataSourceNullable("c0")).thenReturn(ds0);
     when(ds0.getBloomFilter()).thenReturn(null);
     DataSource ds2 = mock(DataSource.class);
-    when(indexSegment.getDataSource("c2")).thenReturn(ds2);
+    when(indexSegment.getDataSourceNullable("c2")).thenReturn(ds2);
     when(ds2.getBloomFilter()).thenReturn(null);
     FetchContext fetchContext = planner.planFetchForPruning(indexSegment, 
queryContext);
     assertTrue(fetchContext.isEmpty());


---------------------------------------------------------------------
To unsubscribe, e-mail: commits-unsubscr...@pinot.apache.org
For additional commands, e-mail: commits-h...@pinot.apache.org