Add BasePlanTest#assertPlan that accepts session

trinodb · Oct 2, 2019 · eade180 · eade180
1 parent ad16ab7
commit eade180
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/presto-main/src/test/java/io/prestosql/sql/planner/assertions/BasePlanTest.java b/presto-main/src/test/java/io/prestosql/sql/planner/assertions/BasePlanTest.java
@@ -108,6 +108,11 @@ protected void assertPlan(String sql, PlanMatchPattern pattern)
         assertPlan(sql, LogicalPlanner.Stage.OPTIMIZED_AND_VALIDATED, pattern);
     }
 
+    protected void assertPlan(String sql, Session session, PlanMatchPattern pattern)
+    {
+        assertPlanWithSession(sql, session, true, pattern);
+    }
+
     protected void assertPlan(String sql, LogicalPlanner.Stage stage, PlanMatchPattern pattern)
     {
         List<PlanOptimizer> optimizers = queryRunner.getPlanOptimizers(true);