From 0811d801fb03bdefa78f91f6a15cd50d101271cd Mon Sep 17 00:00:00 2001 From: Zoltan Haindrich Date: Wed, 17 Jul 2024 16:33:10 +0000 Subject: [PATCH] make query run --- .../java/org/apache/druid/msq/test/CalciteMSQTestsHelper.java | 3 +++ .../org.apache.druid.msq.exec.MSQQuidemTest/decoupled.iq | 2 +- .../org/apache/druid/sql/calcite/util/SqlTestFramework.java | 2 ++ 3 files changed, 6 insertions(+), 1 deletion(-) diff --git a/extensions-core/multi-stage-query/src/test/java/org/apache/druid/msq/test/CalciteMSQTestsHelper.java b/extensions-core/multi-stage-query/src/test/java/org/apache/druid/msq/test/CalciteMSQTestsHelper.java index 54552e5d5b0..b2582c0b9a7 100644 --- a/extensions-core/multi-stage-query/src/test/java/org/apache/druid/msq/test/CalciteMSQTestsHelper.java +++ b/extensions-core/multi-stage-query/src/test/java/org/apache/druid/msq/test/CalciteMSQTestsHelper.java @@ -388,6 +388,9 @@ public class CalciteMSQTestsHelper .inputTmpDir(tempFolderProducer.apply("tmpDir")) .buildMMappedIndex(); break; + case CalciteTests.WIKIPEDIA: + index = TestDataBuilder.makeWikipediaIndex(tempFolderProducer.apply("tmpDir")); + break; case CalciteTests.WIKIPEDIA_FIRST_LAST: index = TestDataBuilder.makeWikipediaIndexWithAggregation(tempFolderProducer.apply("tmpDir")); break; diff --git a/extensions-core/multi-stage-query/src/test/quidem/org.apache.druid.msq.exec.MSQQuidemTest/decoupled.iq b/extensions-core/multi-stage-query/src/test/quidem/org.apache.druid.msq.exec.MSQQuidemTest/decoupled.iq index 7cfdccc74ec..d17f1d6f34a 100644 --- a/extensions-core/multi-stage-query/src/test/quidem/org.apache.druid.msq.exec.MSQQuidemTest/decoupled.iq +++ b/extensions-core/multi-stage-query/src/test/quidem/org.apache.druid.msq.exec.MSQQuidemTest/decoupled.iq @@ -10,7 +10,7 @@ order by 1; +--------------------------------------------+ | TASK | +--------------------------------------------+ -| query-150dc5d7-57d5-44bf-a3be-a023d91fddb9 | +| query-a0537802-857a-450e-bceb-3c6d1fd142ff | +--------------------------------------------+ (1 row) diff --git a/sql/src/test/java/org/apache/druid/sql/calcite/util/SqlTestFramework.java b/sql/src/test/java/org/apache/druid/sql/calcite/util/SqlTestFramework.java index 4374f1e86e2..7a403ac2fac 100644 --- a/sql/src/test/java/org/apache/druid/sql/calcite/util/SqlTestFramework.java +++ b/sql/src/test/java/org/apache/druid/sql/calcite/util/SqlTestFramework.java @@ -42,6 +42,7 @@ import org.apache.druid.query.QueryRunnerFactoryConglomerate; import org.apache.druid.query.QuerySegmentWalker; import org.apache.druid.query.lookup.LookupExtractorFactoryContainerProvider; import org.apache.druid.query.topn.TopNQueryConfig; +import org.apache.druid.quidem.TestSqlModule; import org.apache.druid.segment.DefaultColumnFormatConfig; import org.apache.druid.segment.join.JoinableFactoryWrapper; import org.apache.druid.server.QueryLifecycle; @@ -557,6 +558,7 @@ public class SqlTestFramework binder.bind(DruidOperatorTable.class).in(LazySingleton.class); binder.bind(DataSegment.PruneSpecsHolder.class).toInstance(DataSegment.PruneSpecsHolder.DEFAULT); binder.bind(DefaultColumnFormatConfig.class).toInstance(new DefaultColumnFormatConfig(null)); + binder.install(new TestSqlModule()); } @Provides