fix CalcitePlannerModuleTest

LakshSingla · Sep 14, 2023 · 75652e6 · 75652e6
1 parent 54a664a
commit 75652e6
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/sql/src/test/java/org/apache/druid/sql/calcite/planner/CalcitePlannerModuleTest.java b/sql/src/test/java/org/apache/druid/sql/calcite/planner/CalcitePlannerModuleTest.java
@@ -42,6 +42,7 @@
 import org.apache.druid.sql.calcite.aggregation.SqlAggregator;
 import org.apache.druid.sql.calcite.expression.SqlOperatorConversion;
 import org.apache.druid.sql.calcite.rule.ExtensionCalciteRuleProvider;
+import org.apache.druid.sql.calcite.run.NativeSqlEngine;
 import org.apache.druid.sql.calcite.schema.DruidSchemaCatalog;
 import org.apache.druid.sql.calcite.schema.DruidSchemaName;
 import org.apache.druid.sql.calcite.schema.NamedSchema;
@@ -188,7 +189,8 @@ public void testExtensionCalciteRule()
     PlannerContext context = PlannerContext.create(
         toolbox,
         "SELECT 1",
-        null,
+        // Engine implementation required check engine features
+        new NativeSqlEngine(queryLifecycleFactory, new DefaultObjectMapper()),
         Collections.emptyMap(),
         null
     );