Add index join related nodes to plan matcher

Author: zacw7

presto-main-base/src/test/java/com/facebook/presto/sql/planner/assertions/PlanMatchPattern.java

@@ -61,6 +61,7 @@
 import com.facebook.presto.sql.planner.plan.EnforceSingleRowNode;
 import com.facebook.presto.sql.planner.plan.ExchangeNode;
 import com.facebook.presto.sql.planner.plan.GroupIdNode;
+import com.facebook.presto.sql.planner.plan.IndexJoinNode;
 import com.facebook.presto.sql.planner.plan.LateralJoinNode;
 import com.facebook.presto.sql.planner.plan.OffsetNode;
 import com.facebook.presto.sql.planner.plan.RemoteSourceNode;
@@ -180,6 +181,12 @@ public static PlanMatchPattern constrainedTableScanWithTableLayout(String expect
         return result.addColumnReferences(expectedTableName, columnReferences);
     }
 
+    public static PlanMatchPattern indexSource(String expectedTableName)
+    {
+        return node(IndexSourceNode.class)
+                .with(new IndexSourceMatcher(expectedTableName));
+    }
+
     public static PlanMatchPattern constrainedIndexSource(String expectedTableName, Map<String, Domain> constraint, Map<String, String> columnReferences)
     {
         return node(IndexSourceNode.class)
@@ -449,6 +456,11 @@ public static PlanMatchPattern join(
                 .with(joinMatcher);
     }
 
+    public static PlanMatchPattern indexJoin(PlanMatchPattern left, PlanMatchPattern right)
+    {
+        return node(IndexJoinNode.class, left, right);
+    }
+
     public static PlanMatchPattern cteConsumer(String cteName)
     {
         CteConsumerMatcher cteConsumerMatcher = new CteConsumerMatcher(cteName);