Add QueryAssertions.matches(PlanMatchPattern) method.

trinodb · Jul 15, 2021 · c1cf825 · c1cf825
1 parent 47301fb
commit c1cf825
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/core/trino-main/src/test/java/io/trino/sql/query/QueryAssertions.java b/core/trino-main/src/test/java/io/trino/sql/query/QueryAssertions.java
@@ -341,6 +341,21 @@ public QueryAssert matches(MaterializedResult expected)
             });
         }
 
+        public final QueryAssert matches(PlanMatchPattern expectedPlan)
+        {
+            transaction(runner.getTransactionManager(), runner.getAccessControl())
+                    .execute(session, session -> {
+                        Plan plan = runner.createPlan(session, query, WarningCollector.NOOP);
+                        assertPlan(
+                                session,
+                                runner.getMetadata(),
+                                noopStatsCalculator(),
+                                plan,
+                                expectedPlan);
+                    });
+            return this;
+        }
+
         public QueryAssert containsAll(@Language("SQL") String query)
         {
             MaterializedResult expected = runner.execute(session, query);