[test](Nereids) add test for uncorrelatedApplyFilter #24220

add test for uncorrelatedApplyFilter rule
2023-09-14 16:10:08 +08:00
parent 4efc68a33d
commit ace3e79498
1 changed files with 73 additions and 0 deletions
--- a/fe/fe-core/src/test/java/org/apache/doris/nereids/rules/rewrite/UnCorrelatedApplyFilterTest.java
+++ b/fe/fe-core/src/test/java/org/apache/doris/nereids/rules/rewrite/UnCorrelatedApplyFilterTest.java
@ -0,0 +1,73 @@
+// Licensed to the Apache Software Foundation (ASF) under one
+// or more contributor license agreements.  See the NOTICE file
+// distributed with this work for additional information
+// regarding copyright ownership.  The ASF licenses this file
+// to you under the Apache License, Version 2.0 (the
+// "License"); you may not use this file except in compliance
+// with the License.  You may obtain a copy of the License at
+//
+//   http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing,
+// software distributed under the License is distributed on an
+// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+// KIND, either express or implied.  See the License for the
+// specific language governing permissions and limitations
+// under the License.
+
+package org.apache.doris.nereids.rules.rewrite;
+
+import org.apache.doris.nereids.pattern.GeneratedPlanPatterns;
+import org.apache.doris.nereids.rules.RulePromise;
+import org.apache.doris.nereids.trees.plans.logical.LogicalOlapScan;
+import org.apache.doris.nereids.util.PlanChecker;
+import org.apache.doris.utframe.TestWithFeService;
+
+import org.junit.jupiter.api.Test;
+
+class UnCorrelatedApplyFilterTest extends TestWithFeService implements GeneratedPlanPatterns {
+
+    @Override
+    protected void runBeforeAll() throws Exception {
+        createDatabase("testApplyFilter");
+        connectContext.setDatabase("default_cluster:testApplyFilter");
+        createTables(
+                "CREATE TABLE t1 (col1 int, col2 int) DISTRIBUTED BY HASH(col2)\n" + "BUCKETS 1\n" + "PROPERTIES(\n"
+                        + "    \"replication_num\"=\"1\"\n" + ");",
+                "CREATE TABLE t2 (col1 int, col2 int) DISTRIBUTED BY HASH(col2)\n" + "BUCKETS 1\n" + "PROPERTIES(\n"
+                        + "    \"replication_num\"=\"1\"\n" + ");"
+        );
+    }
+
+    @Test
+    void testCorrelatedFilter() {
+        String sql = "select * from t1 where exists (select * from t2 where t2.col2 > t1.col2)";
+        PlanChecker.from(connectContext)
+                .parse(sql)
+                .analyze()
+                .applyBottomUp(new PullUpProjectUnderApply())
+                .applyBottomUp(new UnCorrelatedApplyFilter())
+                .matches(logicalApply()
+                        .when(e ->
+                                e.left() instanceof LogicalOlapScan && e.right() instanceof LogicalOlapScan));
+    }
+
+    @Test
+    void testComplexFilter() {
+        String sql = "select * from t1 where exists "
+                + "(select * from t2 where t2.col2 > t1.col2 and t2.col1 > 0)";
+        PlanChecker.from(connectContext)
+                .parse(sql)
+                .analyze()
+                .applyBottomUp(new PullUpProjectUnderApply())
+                .applyBottomUp(new UnCorrelatedApplyFilter())
+                .matches(logicalApply(any(), logicalFilter())
+                        .when(e ->
+                                    e.right().getPredicate().toSql().equals("(col1 > 0)")));
+    }
+
+    @Override
+    public RulePromise defaultPromise() {
+        return RulePromise.REWRITE;
+    }
+}