Package io.trino.testing

Class AbstractTestQueryFramework

java.lang.Object

io.trino.testing.AbstractTestQueryFramework

Direct Known Subclasses:

AbstractTestAggregations, AbstractTestDynamicRowFiltering, AbstractTestEngineOnlyQueries, AbstractTestIndexedQueries, AbstractTestJoinQueries, AbstractTestOrderByQueries, AbstractTestQueries, AbstractTestWindowQueries, BaseComplexTypesPredicatePushDownTest, BaseConnectorSmokeTest, BaseDynamicPartitionPruningTest, BaseFailureRecoveryTest, BaseOrcWithBloomFiltersTest, BaseTestParquetWithBloomFilters

@TestInstance(PER_CLASS)
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	assertExplain(Session session, String query, String... expectedExplainRegExps)
protected void	assertExplain(String query, String... expectedExplainRegExps)
protected void	assertExplainAnalyze(Session session, String query, String... expectedExplainRegExps)
protected void	assertExplainAnalyze(String query, String... expectedExplainRegExps)
protected void	assertFunctionNotFound(Session session, String sql, String functionName, TestingAccessControlManager.TestingPrivilege... deniedPrivileges)
protected void	assertNoDataRead(String sql)
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	assertQuery(String actual, String expected, Consumer<Plan> planAssertion)
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	assertQueryStats(Session session, String query, Consumer<QueryStats> queryStatsAssertion, Consumer<MaterializedResult> resultAssertion)
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)
final void	close()
protected final <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(Session session, String sql)
protected Object	computeScalar(String sql)
protected abstract QueryRunner	createQueryRunner()
void	ensureTestNamingConvention()
protected void	executeExclusively(Runnable executionBlock)
protected OperatorStats	extractOperatorStatsForNodeId(QueryId queryId, PlanNodeId nodeId, String operatorType)
protected String	formatSqlText(String sql)
protected final DistributedQueryRunner	getDistributedQueryRunner()
protected DynamicFilterService.DynamicFiltersStats	getDynamicFilteringStats(QueryId queryId)
protected String	getExplainPlan(Session session, String query, ExplainType.Type planType)
protected String	getExplainPlan(String query, ExplainType.Type planType)
protected String	getGraphvizExplainPlan(String query, ExplainType.Type planType)
protected final int	getNodeCount()
protected QualifiedObjectName	getQualifiedTableName(String tableName)
protected final QueryRunner	getQueryRunner()
protected Session	getSession()
void	init()
protected void	inTransaction(Consumer<Session> callback)
protected TransactionBuilder	newTransaction()
protected Session	noJoinReordering()
protected Session	noJoinReordering(OptimizerConfig.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, QualifiedObjectName catalogSchemaTableName)

Methods inherited from class java.lang.Object

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

Constructor Details

AbstractTestQueryFramework

public AbstractTestQueryFramework()

Method Details

init

@BeforeAll
public void init()
          throws Exception

Throws:

Exception

createQueryRunner

protected abstract QueryRunner createQueryRunner()
                                          throws Exception

Throws:

Exception

close

@AfterAll
public final void close()
                 throws Exception

Throws:

Exception

ensureTestNamingConvention

@Test
public void ensureTestNamingConvention()

getSession

protected Session getSession()

getNodeCount

protected final int getNodeCount()

newTransaction

protected TransactionBuilder newTransaction()

inTransaction

protected void inTransaction(Consumer<Session> callback)

computeActual

protected MaterializedResult computeActual(@Language("SQL")
 String sql)

computeActual

protected MaterializedResult computeActual(Session session,
 @Language("SQL")
 String sql)

computeScalar

protected Object computeScalar(@Language("SQL")
 String sql)

computeScalar

protected Object computeScalar(Session session,
 @Language("SQL")
 String sql)

query

@CheckReturnValue
protected org.assertj.core.api.AssertProvider<io.trino.sql.query.QueryAssertions.QueryAssert> query(@Language("SQL")
 String sql)

query

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

assertQuery

protected void assertQuery(@Language("SQL")
 String sql)

assertQuery

protected void assertQuery(Session session,
 @Language("SQL")
 String sql)

assertQuery

protected void assertQuery(@Language("SQL")
 String actual,
 @Language("SQL")
 String expected)

assertQuery

protected void assertQuery(Session session,
 @Language("SQL")
 String actual,
 @Language("SQL")
 String expected)

assertQuery

protected void assertQuery(@Language("SQL")
 String actual,
 @Language("SQL")
 String expected,
 Consumer<Plan> planAssertion)

assertQuery

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

assertQueryEventually

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

assertQueryOrdered

protected void assertQueryOrdered(@Language("SQL")
 String sql)

assertQueryOrdered

protected void assertQueryOrdered(Session session,
 @Language("SQL")
 String sql)

assertQueryOrdered

protected void assertQueryOrdered(@Language("SQL")
 String actual,
 @Language("SQL")
 String expected)

assertQueryOrdered

protected void assertQueryOrdered(Session session,
 @Language("SQL")
 String actual,
 @Language("SQL")
 String expected)

assertUpdate

protected void assertUpdate(@Language("SQL")
 String actual,
 @Language("SQL")
 String expected)

assertUpdate

protected void assertUpdate(Session session,
 @Language("SQL")
 String actual,
 @Language("SQL")
 String expected)

assertUpdate

protected void assertUpdate(@Language("SQL")
 String sql)

assertUpdate

protected void assertUpdate(Session session,
 @Language("SQL")
 String sql)

assertUpdate

protected void assertUpdate(@Language("SQL")
 String sql,
 long count)

assertUpdate

protected void assertUpdate(Session session,
 @Language("SQL")
 String sql,
 long count)

assertUpdate

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

assertQuerySucceeds

protected void assertQuerySucceeds(@Language("SQL")
 String sql)

assertQuerySucceeds

protected void assertQuerySucceeds(Session session,
 @Language("SQL")
 String sql)

assertQueryFailsEventually

protected void assertQueryFailsEventually(@Language("SQL")
 String sql,
 @Language("RegExp")
 String expectedMessageRegExp,
 io.airlift.units.Duration timeout)

assertQueryFails

protected void assertQueryFails(@Language("SQL")
 String sql,
 @Language("RegExp")
 String expectedMessageRegExp)

assertQueryFails

protected void assertQueryFails(Session session,
 @Language("SQL")
 String sql,
 @Language("RegExp")
 String expectedMessageRegExp)

assertQueryReturnsEmptyResult

protected void assertQueryReturnsEmptyResult(@Language("SQL")
 String sql)

assertQueryReturnsEmptyResult

protected void assertQueryReturnsEmptyResult(Session session,
 @Language("SQL")
 String sql)

assertAccessAllowed

protected void assertAccessAllowed(@Language("SQL")
 String sql,
 TestingAccessControlManager.TestingPrivilege... deniedPrivileges)

assertAccessAllowed

protected void assertAccessAllowed(Session session,
 @Language("SQL")
 String sql,
 TestingAccessControlManager.TestingPrivilege... deniedPrivileges)

assertAccessDenied

protected void assertAccessDenied(@Language("SQL")
 String sql,
 @Language("RegExp")
 String exceptionsMessageRegExp,
 TestingAccessControlManager.TestingPrivilege... deniedPrivileges)

assertAccessDenied

protected void assertAccessDenied(Session session,
 @Language("SQL")
 String sql,
 @Language("RegExp")
 String exceptionsMessageRegExp,
 TestingAccessControlManager.TestingPrivilege... deniedPrivileges)

assertFunctionNotFound

protected void assertFunctionNotFound(Session session,
 @Language("SQL")
 String sql,
 String functionName,
 TestingAccessControlManager.TestingPrivilege... deniedPrivileges)

assertTableColumnNames

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

assertExplain

protected void assertExplain(@Language("SQL")
 String query,
 @Language("RegExp")
 String... expectedExplainRegExps)

assertExplain

protected void assertExplain(Session session,
 @Language("SQL")
 String query,
 @Language("RegExp")
 String... expectedExplainRegExps)

assertExplainAnalyze

protected void assertExplainAnalyze(@Language("SQL")
 String query,
 @Language("RegExp")
 String... expectedExplainRegExps)

assertExplainAnalyze

protected void assertExplainAnalyze(Session session,
 @Language("SQL")
 String query,
 @Language("RegExp")
 String... expectedExplainRegExps)

assertQueryStats

protected void assertQueryStats(Session session,
 @Language("SQL")
 String query,
 Consumer<QueryStats> queryStatsAssertion,
 Consumer<MaterializedResult> resultAssertion)

assertNoDataRead

protected void assertNoDataRead(@Language("SQL")
 String sql)

computeExpected

protected MaterializedResult computeExpected(@Language("SQL")
 String sql,
 List<? extends Type> resultTypes)

executeExclusively

protected void executeExclusively(Runnable executionBlock)

formatSqlText

protected String formatSqlText(@Language("SQL")
 String sql)

getExplainPlan

protected String getExplainPlan(@Language("SQL")
 String query,
 ExplainType.Type planType)

getExplainPlan

protected String getExplainPlan(Session session,
 @Language("SQL")
 String query,
 ExplainType.Type planType)

getGraphvizExplainPlan

protected String getGraphvizExplainPlan(@Language("SQL")
 String query,
 ExplainType.Type planType)

getQueryRunner

protected final QueryRunner getQueryRunner()

getDistributedQueryRunner

protected final DistributedQueryRunner getDistributedQueryRunner()

noJoinReordering

protected Session noJoinReordering()

noJoinReordering

protected Session noJoinReordering(OptimizerConfig.JoinDistributionType distributionType)

searchScanFilterAndProjectOperatorStats

protected OperatorStats searchScanFilterAndProjectOperatorStats(QueryId queryId,
 QualifiedObjectName catalogSchemaTableName)

extractOperatorStatsForNodeId

protected OperatorStats extractOperatorStatsForNodeId(QueryId queryId,
 PlanNodeId nodeId,
 String operatorType)

getDynamicFilteringStats

protected DynamicFilterService.DynamicFiltersStats getDynamicFilteringStats(QueryId queryId)

getQualifiedTableName

protected QualifiedObjectName getQualifiedTableName(String tableName)

closeAfterClass

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