Package io.trino.testing

Class AbstractTestQueryFramework

java.lang.Object

io.trino.testing.AbstractTestQueryFramework

Direct Known Subclasses:

AbstractTestAggregations, AbstractTestIndexedQueries, AbstractTestIntegrationSmokeTest, AbstractTestJoinQueries, AbstractTestOrderByQueries, AbstractTestQueries, AbstractTestWindowQueries

public abstract class AbstractTestQueryFramework
extends Object

Constructor Summary

Constructors

Constructor	Description
AbstractTestQueryFramework()

Method Summary

Modifier and Type	Method	Description
protected void	assertAccessAllowed(Session session, String sql, TestingAccessControlManager.TestingPrivilege... deniedPrivileges)
protected void	assertAccessAllowed(String sql, TestingAccessControlManager.TestingPrivilege... deniedPrivileges)
protected void	assertAccessDenied(Session session, String sql, String exceptionsMessageRegExp, TestingAccessControlManager.TestingPrivilege... deniedPrivileges)
protected void	assertAccessDenied(String sql, String exceptionsMessageRegExp, TestingAccessControlManager.TestingPrivilege... deniedPrivileges)
protected void	assertExplainAnalyze(Session session, String query, String... expectedExplainRegExps)
protected void	assertExplainAnalyze(String query, String... expectedExplainRegExps)
protected void	assertQuery(Session session, String sql)
protected void	assertQuery(Session session, String actual, String expected)
protected void	assertQuery(Session session, String actual, String expected, Consumer<Plan> planAssertion)
protected void	assertQuery(String sql)
protected void	assertQuery(String actual, String expected)
protected void	assertQueryEventually(Session session, String actual, String expected, io.airlift.units.Duration timeout)
protected void	assertQueryFails(Session session, String sql, String expectedMessageRegExp)
protected void	assertQueryFails(String sql, String expectedMessageRegExp)
protected void	assertQueryFailsEventually(String sql, String expectedMessageRegExp, io.airlift.units.Duration timeout)
protected void	assertQueryOrdered(Session session, String sql)
protected void	assertQueryOrdered(Session session, String actual, String expected)
protected void	assertQueryOrdered(String sql)
protected void	assertQueryOrdered(String actual, String expected)
protected void	assertQueryReturnsEmptyResult(Session session, String sql)
protected void	assertQueryReturnsEmptyResult(String sql)
protected void	assertQuerySucceeds(Session session, String sql)
protected void	assertQuerySucceeds(String sql)
protected void	assertTableColumnNames(String tableName, String... columnNames)
protected void	assertUpdate(Session session, String sql)
protected void	assertUpdate(Session session, String sql, long count)
protected void	assertUpdate(Session session, String sql, long count, Consumer<Plan> planAssertion)
protected void	assertUpdate(Session session, String actual, String expected)
protected void	assertUpdate(String sql)
protected void	assertUpdate(String sql, long count)
protected void	assertUpdate(String actual, String expected)
void	close()
protected <T extends AutoCloseable> T	closeAfterClass(T resource)
protected MaterializedResult	computeActual(Session session, String sql)
protected MaterializedResult	computeActual(String sql)
protected MaterializedResult	computeExpected(String sql, List<? extends Type> resultTypes)
protected Object	computeScalar(String sql)
protected abstract QueryRunner	createQueryRunner()
protected void	executeExclusively(Runnable executionBlock)
protected String	formatSqlText(String sql)
protected DistributedQueryRunner	getDistributedQueryRunner()
protected String	getExplainPlan(String query, ExplainType.Type planType)
protected String	getGraphvizExplainPlan(String query, ExplainType.Type planType)
protected int	getNodeCount()
protected QueryRunner	getQueryRunner()
protected Session	getSession()
void	init()
protected Session	noJoinReordering()
protected Session	noJoinReordering(FeaturesConfig.JoinDistributionType distributionType)
protected org.assertj.core.api.AssertProvider<io.trino.sql.query.QueryAssertions.QueryAssert>	query(Session session, String sql)
protected org.assertj.core.api.AssertProvider<io.trino.sql.query.QueryAssertions.QueryAssert>	query(String sql)
protected OperatorStats	searchScanFilterAndProjectOperatorStats(QueryId queryId, String tableName)
protected static void	skipTestUnless(boolean requirement)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

AbstractTestQueryFramework

public AbstractTestQueryFramework()

Method Detail

init

public void init()
          throws Exception

Throws:

Exception

createQueryRunner

protected abstract QueryRunner createQueryRunner()
                                          throws Exception

Throws:

Exception

close

public void close()
           throws Exception

Throws:

Exception

getSession

protected Session getSession()

getNodeCount

protected final int getNodeCount()

computeActual

protected MaterializedResult computeActual(String sql)

computeActual

protected MaterializedResult computeActual(Session session,
                                           String sql)

computeScalar

protected Object computeScalar(String sql)

query

protected org.assertj.core.api.AssertProvider<io.trino.sql.query.QueryAssertions.QueryAssert> query(String sql)

query

protected org.assertj.core.api.AssertProvider<io.trino.sql.query.QueryAssertions.QueryAssert> query(Session session,
                                                                                                    String sql)

assertQuery

protected void assertQuery(String sql)

assertQuery

protected void assertQuery(Session session,
                           String sql)

assertQuery

protected void assertQuery(String actual,
                           String expected)

assertQuery

protected void assertQuery(Session session,
                           String actual,
                           String expected)

assertQuery

protected void assertQuery(Session session,
                           String actual,
                           String expected,
                           Consumer<Plan> planAssertion)

assertQueryEventually

protected void assertQueryEventually(Session session,
                                     String actual,
                                     String expected,
                                     io.airlift.units.Duration timeout)

assertQueryOrdered

protected void assertQueryOrdered(String sql)

assertQueryOrdered

protected void assertQueryOrdered(Session session,
                                  String sql)

assertQueryOrdered

protected void assertQueryOrdered(String actual,
                                  String expected)

assertQueryOrdered

protected void assertQueryOrdered(Session session,
                                  String actual,
                                  String expected)

assertUpdate

protected void assertUpdate(String actual,
                            String expected)

assertUpdate

protected void assertUpdate(Session session,
                            String actual,
                            String expected)

assertUpdate

protected void assertUpdate(String sql)

assertUpdate

protected void assertUpdate(Session session,
                            String sql)

assertUpdate

protected void assertUpdate(String sql,
                            long count)

assertUpdate

protected void assertUpdate(Session session,
                            String sql,
                            long count)

assertUpdate

protected void assertUpdate(Session session,
                            String sql,
                            long count,
                            Consumer<Plan> planAssertion)

assertQuerySucceeds

protected void assertQuerySucceeds(String sql)

assertQuerySucceeds

protected void assertQuerySucceeds(Session session,
                                   String sql)

assertQueryFailsEventually

protected void assertQueryFailsEventually(String sql,
                                          String expectedMessageRegExp,
                                          io.airlift.units.Duration timeout)

assertQueryFails

protected void assertQueryFails(String sql,
                                String expectedMessageRegExp)

assertQueryFails

protected void assertQueryFails(Session session,
                                String sql,
                                String expectedMessageRegExp)

assertQueryReturnsEmptyResult

protected void assertQueryReturnsEmptyResult(String sql)

assertQueryReturnsEmptyResult

protected void assertQueryReturnsEmptyResult(Session session,
                                             String sql)

assertAccessAllowed

protected void assertAccessAllowed(String sql,
                                   TestingAccessControlManager.TestingPrivilege... deniedPrivileges)

assertAccessAllowed

protected void assertAccessAllowed(Session session,
                                   String sql,
                                   TestingAccessControlManager.TestingPrivilege... deniedPrivileges)

assertAccessDenied

protected void assertAccessDenied(String sql,
                                  String exceptionsMessageRegExp,
                                  TestingAccessControlManager.TestingPrivilege... deniedPrivileges)

assertAccessDenied

protected void assertAccessDenied(Session session,
                                  String sql,
                                  String exceptionsMessageRegExp,
                                  TestingAccessControlManager.TestingPrivilege... deniedPrivileges)

assertTableColumnNames

protected void assertTableColumnNames(String tableName,
                                      String... columnNames)

assertExplainAnalyze

protected void assertExplainAnalyze(String query,
                                    String... expectedExplainRegExps)

assertExplainAnalyze

protected void assertExplainAnalyze(Session session,
                                    String query,
                                    String... expectedExplainRegExps)

computeExpected

protected MaterializedResult computeExpected(String sql,
                                             List<? extends Type> resultTypes)

executeExclusively

protected void executeExclusively(Runnable executionBlock)

formatSqlText

protected String formatSqlText(String sql)

getExplainPlan

protected String getExplainPlan(String query,
                                ExplainType.Type planType)

getGraphvizExplainPlan

protected String getGraphvizExplainPlan(String query,
                                        ExplainType.Type planType)

skipTestUnless

protected static void skipTestUnless(boolean requirement)

getQueryRunner

protected final QueryRunner getQueryRunner()

getDistributedQueryRunner

protected final DistributedQueryRunner getDistributedQueryRunner()

noJoinReordering

protected Session noJoinReordering()

noJoinReordering

protected Session noJoinReordering(FeaturesConfig.JoinDistributionType distributionType)

searchScanFilterAndProjectOperatorStats

protected OperatorStats searchScanFilterAndProjectOperatorStats(QueryId queryId,
                                                                String tableName)

closeAfterClass

@CanIgnoreReturnValue
protected final <T extends AutoCloseable> T closeAfterClass(T resource)