From 039d9c88f83810428834d4dee4bbae040ed942f1 Mon Sep 17 00:00:00 2001
From: Zoltan Haindrich <kirk@rxd.hu>
Date: Tue, 6 Aug 2024 20:15:16 +0000
Subject: [PATCH] mask 2 failing testcases

---
 .../test/CalciteSelectJoinQueryMSQTest.java   | 22 +++++++++++++++++++
 1 file changed, 22 insertions(+)

diff --git a/extensions-core/multi-stage-query/src/test/java/org/apache/druid/msq/test/CalciteSelectJoinQueryMSQTest.java b/extensions-core/multi-stage-query/src/test/java/org/apache/druid/msq/test/CalciteSelectJoinQueryMSQTest.java
index 2283386f207..e5fbbc3bccc 100644
--- a/extensions-core/multi-stage-query/src/test/java/org/apache/druid/msq/test/CalciteSelectJoinQueryMSQTest.java
+++ b/extensions-core/multi-stage-query/src/test/java/org/apache/druid/msq/test/CalciteSelectJoinQueryMSQTest.java
@@ -20,6 +20,7 @@
 package org.apache.druid.msq.test;
 
 import com.google.common.collect.ImmutableMap;
+import org.apache.druid.common.config.NullHandling;
 import org.apache.druid.msq.sql.MSQTaskSqlEngine;
 import org.apache.druid.sql.calcite.BaseCalciteQueryTest;
 import org.apache.druid.sql.calcite.CalciteJoinQueryTest;
@@ -27,9 +28,13 @@ import org.apache.druid.sql.calcite.QueryTestBuilder;
 import org.apache.druid.sql.calcite.SqlTestFrameworkConfig;
 import org.apache.druid.sql.calcite.planner.JoinAlgorithm;
 import org.apache.druid.sql.calcite.planner.PlannerContext;
+import org.junit.jupiter.params.ParameterizedTest;
+import org.junit.jupiter.params.provider.MethodSource;
 
 import java.util.Map;
 
+import static org.junit.Assume.assumeTrue;
+
 /**
  * Runs {@link CalciteJoinQueryTest} but with MSQ engine.
  */
@@ -78,7 +83,24 @@ public class CalciteSelectJoinQueryMSQTest
     protected JoinAlgorithm joinAlgorithm()
     {
       return JoinAlgorithm.SORT_MERGE;
+    }
 
+    @Override
+    @MethodSource("provideQueryContexts")
+    @ParameterizedTest(name = "{0}")
+    public void testRegressionFilteredAggregatorsSubqueryJoins(Map<String, Object> queryContext)
+    {
+      assumeTrue("Returns incorrect result with replaceWithDefault", NullHandling.sqlCompatible());
+      super.testRegressionFilteredAggregatorsSubqueryJoins(queryContext);
+    }
+
+    @Override
+    @MethodSource("provideQueryContexts")
+    @ParameterizedTest(name = "{0}")
+    public void testFilterAndGroupByLookupUsingJoinOperatorWithNotFilter(Map<String, Object> queryContext)
+    {
+      assumeTrue("Returns incorrect result with replaceWithDefault", NullHandling.sqlCompatible());
+      super.testFilterAndGroupByLookupUsingJoinOperatorWithNotFilter(queryContext);
     }
   }