Support passing statsCalculator to RuleAssert

Author: rschlussel-zz

presto-main/src/test/java/com/facebook/presto/sql/planner/iterative/rule/test/RuleAssert.java

@@ -51,7 +51,7 @@
 public class RuleAssert
 {
     private final Metadata metadata;
-    private final StatsCalculator statsCalculator;
+    private StatsCalculator statsCalculator;
     private final CostCalculator costCalculator;
     private Session session;
     private final Rule rule;
@@ -96,6 +96,13 @@ public RuleAssert withSession(Session session)
         return this;
     }
 
+    public RuleAssert withStatsCalculator(StatsCalculator statsCalculator)
+    {
+        checkState(lookup == null, "lookup has been set");
+        this.statsCalculator = statsCalculator;
+        return this;
+    }
+
     public RuleAssert on(Function<PlanBuilder, PlanNode> planProvider)
     {
         checkArgument(plan == null, "plan has already been set");