update test case for JoinDataSource since it has a new JoinAlgorithm …

…field now
apache · Dec 18, 2024 · af51747 · af51747
1 parent c7dc930
commit af51747
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/processing/src/main/java/org/apache/druid/query/RestrictedDataSource.java b/processing/src/main/java/org/apache/druid/query/RestrictedDataSource.java
@@ -112,7 +112,7 @@ public DataSource withChildren(List<DataSource> children)
       throw new IAE("Expected [1] child, got [%d]", children.size());
     }
 
-    return RestrictedDataSource.create(children.get(0), rowFilter);
+    return create(children.get(0), rowFilter);
   }
 
   @Override
@@ -151,7 +151,7 @@ public Function<SegmentReference, SegmentReference> createSegmentMapFunction(
   @Override
   public DataSource withUpdatedDataSource(DataSource newSource)
   {
-    return RestrictedDataSource.create(newSource, rowFilter);
+    return create(newSource, rowFilter);
   }
 
   @Override

diff --git a/processing/src/test/java/org/apache/druid/query/JoinDataSourceTest.java b/processing/src/test/java/org/apache/druid/query/JoinDataSourceTest.java
@@ -516,7 +516,8 @@ public void testGetAnalysisWithRestrictedDS()
         JoinType.LEFT,
         null,
         ExprMacroTable.nil(),
-        null
+        null,
+        JoinAlgorithm.BROADCAST
     );
     DataSourceAnalysis analysis = dataSource.getAnalysis();
     Assert.assertEquals("table1", analysis.getBaseDataSource().getTableNames().iterator().next());