Add a constructor to QueryAssertions

Cherry-pick of trinodb/trino@7efb49c Co-authored-by: Martin Traverso <mtraverso@gmail.com>
prestodb · Dec 20, 2022 · f6a0a03 · f6a0a03
1 parent 13ca62f
commit f6a0a03
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/presto-main/src/test/java/com/facebook/presto/sql/query/QueryAssertions.java b/presto-main/src/test/java/com/facebook/presto/sql/query/QueryAssertions.java
@@ -33,6 +33,7 @@
 import static com.facebook.presto.testing.TestingSession.testSessionBuilder;
 import static com.google.common.base.Strings.nullToEmpty;
 import static java.lang.String.format;
+import static java.util.Objects.requireNonNull;
 import static org.testng.Assert.assertEquals;
 import static org.testng.Assert.fail;
 
@@ -63,6 +64,11 @@ public QueryAssertions(Session session)
         runner = new LocalQueryRunner(session);
     }
 
+    public QueryAssertions(QueryRunner runner)
+    {
+        this.runner = requireNonNull(runner, "runner is null");
+    }
+
     public QueryRunner getQueryRunner()
     {
         return runner;