Provide assertQueryOrdered overload that accepts session

trinodb · Jul 3, 2017 · c631c5e · c631c5e
1 parent 6996a93
commit c631c5e
Showing 1 changed file with 10 additions and 5 deletions.
diff --git a/presto-tests/src/main/java/com/facebook/presto/tests/AbstractTestQueryFramework.java b/presto-tests/src/main/java/com/facebook/presto/tests/AbstractTestQueryFramework.java
@@ -118,11 +118,6 @@ protected void assertQuery(Session session, @Language("SQL") String sql)
         QueryAssertions.assertQuery(queryRunner, session, sql, h2QueryRunner, sql, false, false);
     }
 
-    public void assertQueryOrdered(@Language("SQL") String sql)
-    {
-        QueryAssertions.assertQuery(queryRunner, getSession(), sql, h2QueryRunner, sql, true, false);
-    }
-
     protected void assertQuery(@Language("SQL") String actual, @Language("SQL") String expected)
     {
         QueryAssertions.assertQuery(queryRunner, getSession(), actual, h2QueryRunner, expected, false, false);
@@ -133,6 +128,16 @@ protected void assertQuery(Session session, @Language("SQL") String actual, @Lan
         QueryAssertions.assertQuery(queryRunner, session, actual, h2QueryRunner, expected, false, false);
     }
 
+    public void assertQueryOrdered(@Language("SQL") String sql)
+    {
+        assertQueryOrdered(getSession(), sql);
+    }
+
+    public void assertQueryOrdered(Session session, @Language("SQL") String sql)
+    {
+        assertQueryOrdered(session, sql, sql);
+    }
+
     protected void assertQueryOrdered(@Language("SQL") String actual, @Language("SQL") String expected)
     {
         assertQueryOrdered(getSession(), actual, expected);