Package io.trino.testing

Class BaseFailureRecoveryTest

java.lang.Object

io.trino.testing.AbstractTestQueryFramework

io.trino.testing.BaseFailureRecoveryTest

Direct Known Subclasses:

ExtendedFailureRecoveryTest

public abstract class BaseFailureRecoveryTest
extends AbstractTestQueryFramework

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
protected class	BaseFailureRecoveryTest.FailureRecoveryAssert

Constructor Summary

Constructors

Modifier	Constructor	Description
protected	BaseFailureRecoveryTest(RetryPolicy retryPolicy)
protected	BaseFailureRecoveryTest(RetryPolicy retryPolicy, int maxParallelTestConcurrency)

Method Summary

Modifier and Type	Method	Description
protected abstract boolean	areWriteRetriesSupported()
protected BaseFailureRecoveryTest.FailureRecoveryAssert	assertThatQuery(String query)
protected static Function<MaterializedResult,Integer>	boundaryCoordinatorStage()
protected static Function<MaterializedResult,Integer>	boundaryDistributedStage()
protected void	checkTemporaryTables(Set<String> queryIds)
protected final QueryRunner	createQueryRunner()
protected abstract QueryRunner	createQueryRunner(List<io.trino.tpch.TpchTable<?>> requiredTpchTables, Map<String,String> configProperties, Map<String,String> coordinatorProperties)
protected static Function<MaterializedResult,Integer>	distributedStage()
protected RetryPolicy	getRetryPolicy()
protected static Function<MaterializedResult,Integer>	intermediateDistributedStage()
protected static Function<MaterializedResult,Integer>	leafStage()
protected Object[][]	moreParallelTests(Object[][] some, Object[]... more)
protected Object[]	parallelTest(String name, Runnable runnable)
Object[][]	parallelTests()
protected static Function<MaterializedResult,Integer>	rootStage()
protected void	testAnalyzeTable()
protected void	testCreateTable()
protected void	testDelete()
protected void	testDeleteWithSubquery()
protected void	testExplainAnalyze()
protected void	testInsert()
protected void	testMerge()
protected void	testNonSelect(Optional<Session> session, Optional<String> setupQuery, String query, Optional<String> cleanupQuery, boolean writesData)
void	testParallel(Runnable runnable)
protected void	testRefreshMaterializedView()
protected void	testRequestTimeouts()
protected void	testSelect(String query)
protected void	testSelect(String query, Optional<Session> session)
protected void	testSelect(String query, Optional<Session> session, Consumer<QueryId> queryAssertion)
protected void	testTableModification(Optional<Session> session, Optional<String> setupQuery, String query, Optional<String> cleanupQuery)
protected void	testTableModification(Optional<String> setupQuery, String query, Optional<String> cleanupQuery)
protected void	testUpdate()
protected void	testUpdateWithSubquery()

Methods inherited from class io.trino.testing.AbstractTestQueryFramework

assertAccessAllowed, assertAccessAllowed, assertAccessDenied, assertAccessDenied, assertExplain, assertExplain, assertExplainAnalyze, assertExplainAnalyze, assertNoDataRead, assertQuery, assertQuery, assertQuery, assertQuery, assertQuery, assertQuery, assertQueryEventually, assertQueryFails, assertQueryFails, assertQueryFailsEventually, assertQueryOrdered, assertQueryOrdered, assertQueryOrdered, assertQueryOrdered, assertQueryReturnsEmptyResult, assertQueryReturnsEmptyResult, assertQueryStats, assertQuerySucceeds, assertQuerySucceeds, assertTableColumnNames, assertUpdate, assertUpdate, assertUpdate, assertUpdate, assertUpdate, assertUpdate, assertUpdate, close, closeAfterClass, computeActual, computeActual, computeExpected, computeScalar, computeScalar, ensureTestNamingConvention, executeExclusively, formatSqlText, getDistributedQueryRunner, getDynamicFilteringStats, getExplainPlan, getExplainPlan, getGraphvizExplainPlan, getNodeCount, getQualifiedTableName, getQueryRunner, getSession, init, inTransaction, newTransaction, noJoinReordering, noJoinReordering, query, query, searchScanFilterAndProjectOperatorStats, skipTestUnless

Methods inherited from class java.lang.Object

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

Constructor Details

BaseFailureRecoveryTest

protected BaseFailureRecoveryTest(RetryPolicy retryPolicy)

BaseFailureRecoveryTest

protected BaseFailureRecoveryTest(RetryPolicy retryPolicy,
 int maxParallelTestConcurrency)

Method Details

getRetryPolicy

protected RetryPolicy getRetryPolicy()

createQueryRunner

protected final QueryRunner createQueryRunner()
                                       throws Exception

Specified by:

createQueryRunner in class AbstractTestQueryFramework

Throws:

Exception

createQueryRunner

protected abstract QueryRunner createQueryRunner(List<io.trino.tpch.TpchTable<?>> requiredTpchTables,
 Map<String,String> configProperties,
 Map<String,String> coordinatorProperties)
                                          throws Exception

Throws:

Exception

areWriteRetriesSupported

protected abstract boolean areWriteRetriesSupported()

testSelect

protected void testSelect(String query)

testSelect

protected void testSelect(String query,
 Optional<Session> session)

testSelect

protected void testSelect(String query,
 Optional<Session> session,
 Consumer<QueryId> queryAssertion)

parallelTests

public Object[][] parallelTests()

testParallel

public void testParallel(Runnable runnable)

testCreateTable

protected void testCreateTable()

testInsert

protected void testInsert()

testDelete

protected void testDelete()

testDeleteWithSubquery

protected void testDeleteWithSubquery()

testUpdate

protected void testUpdate()

testUpdateWithSubquery

protected void testUpdateWithSubquery()

testAnalyzeTable

protected void testAnalyzeTable()

testMerge

protected void testMerge()

testRefreshMaterializedView

protected void testRefreshMaterializedView()

testExplainAnalyze

protected void testExplainAnalyze()

testRequestTimeouts

protected void testRequestTimeouts()

testTableModification

protected void testTableModification(Optional<String> setupQuery,
 String query,
 Optional<String> cleanupQuery)

testTableModification

protected void testTableModification(Optional<Session> session,
 Optional<String> setupQuery,
 String query,
 Optional<String> cleanupQuery)

testNonSelect

protected void testNonSelect(Optional<Session> session,
 Optional<String> setupQuery,
 String query,
 Optional<String> cleanupQuery,
 boolean writesData)

assertThatQuery

protected BaseFailureRecoveryTest.FailureRecoveryAssert assertThatQuery(String query)

checkTemporaryTables

protected void checkTemporaryTables(Set<String> queryIds)

rootStage

protected static Function<MaterializedResult,Integer> rootStage()

boundaryCoordinatorStage

protected static Function<MaterializedResult,Integer> boundaryCoordinatorStage()

boundaryDistributedStage

protected static Function<MaterializedResult,Integer> boundaryDistributedStage()

intermediateDistributedStage

protected static Function<MaterializedResult,Integer> intermediateDistributedStage()

distributedStage

protected static Function<MaterializedResult,Integer> distributedStage()

leafStage

protected static Function<MaterializedResult,Integer> leafStage()

parallelTest

protected Object[] parallelTest(String name,
 Runnable runnable)

moreParallelTests

protected Object[][] moreParallelTests(Object[][] some,
 Object[]... more)