Add assertDistributedPlan that accepts session
@@ -140,7 +140,12 @@ public class BasePlanTest
protected void assertDistributedPlan(String sql, PlanMatchPattern pattern)
{
- assertPlanWithSession(sql, getQueryRunner().getDefaultSession(), false, pattern);
+ assertDistributedPlan(sql, getQueryRunner().getDefaultSession(), pattern);
+ }
+
+ protected void assertDistributedPlan(String sql, Session session, PlanMatchPattern pattern)
+ {
+ assertPlanWithSession(sql, session, false, pattern);
}
protected void assertMinimallyOptimizedPlan(@Language("SQL") String sql, PlanMatchPattern pattern)
