org.apache.drill.test

Class BaseTestQuery

java.lang.Object

DrillTest

org.apache.drill.exec.ExecTest

org.apache.drill.test.BaseTestQuery

Direct Known Subclasses:

Drill2283InfoSchemaVarchar1BugTest, ParquetRecordReaderTest, PlanTestBase, PreparedStatementTestBase, TemporaryTablesAutomaticDropTest, TestAggNullable, TestAltSortQueries, TestBitBitKerberos, TestBroadcast, TestBugFixes, TestCaseNullableTypes, TestCaseSensitivity, TestClassTransformation, TestColumnReaderFactory, TestComplexToJson, TestComplexTypeReader, TestComplexTypeWriter, TestContextFunctions, TestConvertFunctions, TestCountDownLatchInjection, TestCsvHeader, TestCTAS, TestCTTAS, TestCustomTunnel, TestDateAddFunctions, TestDateConversions, TestDateReader, TestDateTruncFunctions, TestDisabledFunctionality, TestDrillParquetReader, TestDrillSQLWorker, TestDynamicUDFSupport, TestEarlyLimit0Optimization, TestEmptyInputSql, TestEmptyPopulation, TestExampleQueries, TestExceptionInjection, TestExtendedTypes, TestExternalSort, TestFastComplexSchema, TestFileSelection, TestFixedlenDecimal, TestFlatten, TestFragmentExecutorCancel, TestFrameworkTest, TestFunctionsQuery, TestFunctionsWithTypeExpoQueries, TestGlob, TestHashAggr, TestImplicitFileColumns, TestInfoSchema, TestInList, TestJoinNullable, TestJsonNanInf, TestJsonReader, TestJsonReaderWithSparseFiles, TestJsonRecordReader, TestLargeInClause, TestLateLimit0Optimization, TestLateralPlans, TestLimitWithExchanges, TestLogicalExprSerDe, TestMathFunctionsWithNanInf, TestMergeJoinWithSchemaChanges, TestMetadataProvider, TestNestedComplexSchema, TestNestedDateTimeTimestamp, TestNewDateFunctions, TestNewMathFunctions, TestOperatorMetrics, TestOptions, TestOptionsAuthEnabled, TestOutOfMemoryOutcome, TestParquetComplex, TestParquetGroupScan, TestParquetLogicalTypes, TestParquetScan, TestParquetWriterEmptyFiles, TestPauseInjection, TestQueriesOnLargeFile, TestRepeatedReaders, TestSchemaChange, TestSchemaNotFoundException, TestSchemaPathMaterialization, TestSelectivity, TestSemiJoin, TestSimpleCastFunctions, TestSort, TestSpoolingBuffer, TestStarQueries, TestStringFunctions, TestSVRemover, TestTextColumn, TestTextJoin, TestTextReader, TestTopNSchemaChanges, TestTpcdsSf1Leaks, TestTpchDistributed, TestTpchDistributedStreaming, TestTpchExplain, TestTpchLimit0, TestTpchSingleMode, TestTrigFunctions, TestUnionAll, TestUnionDistinct, TestUserBitKerberos, TestUserBitKerberosEncryption, TestUserBitSaslCompatibility, TestUserBitSSL, TestUserBitSSLServer, TestUtf8SupportInQueryString, TestVarDecimalFunctions, TestWindowFrame, TestWindowFunctions, TestWithClause, TestWriter

public class BaseTestQuery
extends ExecTest

deprecated Use ClusterTest instead. But, not marked as deprecated because it is still widely used.

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	BaseTestQuery.ClassicTestServices
static class	BaseTestQuery.SilentListener
class	BaseTestQuery.TestResultSet

Field Summary

Fields

Modifier and Type	Field and Description
protected static BufferAllocator	allocator
protected static Drillbit[]	bits
protected static DrillClient	client
protected static DrillConfig	config
protected static RemoteServiceSet	serviceSet

Fields inherited from class org.apache.drill.exec.ExecTest

c, dirTestWatcher, optionManager

Constructor Summary

Constructors

Constructor and Description
BaseTestQuery()

Method Summary

Modifier and Type	Method and Description
static void	alterSession(String option, Object value)
protected static Properties	cloneDefaultTestConfigProperties()
static void	closeClient()
protected static void	errorMsgTestHelper(String testSqlQuery, String expectedErrorMsg) Utility method which tests given query produces a UserException and the exception message contains the given message.
protected static BufferAllocator	getAllocator()
protected static DrillbitContext	getDrillbitContext() Useful for tests that require a DrillbitContext to get/add storage plugins, options etc.
static String	getFile(String resource)
static String	getPhysicalFileFromResource(String resource) Copy the resource (ex.
protected static String	getResultString(List<QueryDataBatch> results, String delimiter)
static int	getUserPort()
protected int	logResult(List<QueryDataBatch> results)
static TestBuilder	newTest()
protected static void	parseErrorHelper(String testSqlQuery) Utility method which tests given query produces a UserException with UserBitShared.DrillPBError.ErrorType being DrillPBError.ErrorType.PARSE the given message.
protected int	printResult(List<QueryDataBatch> results)
static void	resetAllSessionOptions()
static void	resetDrillbitCount()
static void	resetSessionOption(String option)
protected static void	runSQL(String sql)
protected void	setColumnWidth(int columnWidth)
protected void	setColumnWidths(int[] columnWidths)
protected static void	setSessionOption(String option, boolean value)
protected static void	setSessionOption(String option, double value)
protected static void	setSessionOption(String option, long value)
protected static void	setSessionOption(String option, String value)
static void	setupDefaultTestCluster()
static void	test(String query)
static void	test(String query, Object... args)
static TestBuilder	testBuilder()
protected static List<QueryDataBatch>	testLogicalWithResults(String logical)
protected static void	testNoResult(int interation, String query, Object... args)
static void	testNoResult(String query, Object... args)
protected static int	testPhysical(String query)
protected static void	testPhysicalFromFile(String file)
protected static List<QueryDataBatch>	testPhysicalWithResults(String physical)
static List<QueryDataBatch>	testPreparedStatement(UserProtos.PreparedStatementHandle handle)
static int	testRunAndPrint(UserBitShared.QueryType type, String query)
static List<QueryDataBatch>	testRunAndReturn(UserBitShared.QueryType type, Object query)
protected static int	testSql(String query)
protected static List<QueryDataBatch>	testSqlWithResults(String sql)
protected static void	testWithListener(UserBitShared.QueryType type, String query, UserResultsListener resultListener)
static void	updateClient(Properties properties) Close the current client and open a new client using the given properties.
static void	updateClient(String user)
static void	updateClient(String user, String password)
protected static void	updateTestCluster(int newDrillbitCount, DrillConfig newConfig)
protected static void	updateTestCluster(int newDrillbitCount, DrillConfig newConfig, Properties properties)

Methods inherited from class org.apache.drill.exec.ExecTest

clear, getLocalFileSystem, mockDrillbitContext, mockUsDateFormatSymbols, mockUtcDateTimeZone, parseExpr, setupOptionManager

Methods inherited from class java.lang.Object

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

Field Detail

client

protected static DrillClient client

bits

protected static Drillbit[] bits

serviceSet

protected static RemoteServiceSet serviceSet

config

protected static DrillConfig config

allocator

protected static BufferAllocator allocator

Constructor Detail

BaseTestQuery

public BaseTestQuery()

Method Detail

setupDefaultTestCluster

public static void setupDefaultTestCluster()
                                    throws Exception

Throws:

Exception

updateTestCluster

protected static void updateTestCluster(int newDrillbitCount,
                                        DrillConfig newConfig)

updateTestCluster

protected static void updateTestCluster(int newDrillbitCount,
                                        DrillConfig newConfig,
                                        Properties properties)

getDrillbitContext

protected static DrillbitContext getDrillbitContext()

Useful for tests that require a DrillbitContext to get/add storage plugins, options etc.

Returns:

DrillbitContext of first Drillbit in the cluster.

cloneDefaultTestConfigProperties

protected static Properties cloneDefaultTestConfigProperties()

updateClient

public static void updateClient(Properties properties)
                         throws Exception

Close the current client and open a new client using the given properties. All tests executed after this method call use the new client.

Parameters:

properties -

Throws:

Exception

updateClient

public static void updateClient(String user)
                         throws Exception

Throws:

Exception

updateClient

public static void updateClient(String user,
                                String password)
                         throws Exception

Throws:

Exception

getAllocator

protected static BufferAllocator getAllocator()

getUserPort

public static int getUserPort()

newTest

public static TestBuilder newTest()

testBuilder

public static TestBuilder testBuilder()

closeClient

public static void closeClient()
                        throws Exception

Throws:

Exception

resetDrillbitCount

public static void resetDrillbitCount()

runSQL

protected static void runSQL(String sql)
                      throws Exception

Throws:

Exception

testSqlWithResults

protected static List<QueryDataBatch> testSqlWithResults(String sql)
                                                  throws Exception

Throws:

Exception

testLogicalWithResults

protected static List<QueryDataBatch> testLogicalWithResults(String logical)
                                                      throws Exception

Throws:

Exception

testPhysicalWithResults

protected static List<QueryDataBatch> testPhysicalWithResults(String physical)
                                                       throws Exception

Throws:

Exception

testRunAndReturn

public static List<QueryDataBatch> testRunAndReturn(UserBitShared.QueryType type,
                                                    Object query)
                                             throws Exception

Throws:

Exception

testPreparedStatement

public static List<QueryDataBatch> testPreparedStatement(UserProtos.PreparedStatementHandle handle)
                                                  throws Exception

Throws:

Exception

testRunAndPrint

public static int testRunAndPrint(UserBitShared.QueryType type,
                                  String query)
                           throws Exception

Throws:

Exception

testWithListener

protected static void testWithListener(UserBitShared.QueryType type,
                                       String query,
                                       UserResultsListener resultListener)

testNoResult

public static void testNoResult(String query,
                                Object... args)
                         throws Exception

Throws:

Exception

alterSession

public static void alterSession(String option,
                                Object value)

resetSessionOption

public static void resetSessionOption(String option)

resetAllSessionOptions

public static void resetAllSessionOptions()

testNoResult

protected static void testNoResult(int interation,
                                   String query,
                                   Object... args)
                            throws Exception

Throws:

Exception

test

public static void test(String query,
                        Object... args)
                 throws Exception

Throws:

Exception

test

public static void test(String query)
                 throws Exception

Throws:

Exception

testPhysical

protected static int testPhysical(String query)
                           throws Exception

Throws:

Exception

testSql

protected static int testSql(String query)
                      throws Exception

Throws:

Exception

testPhysicalFromFile

protected static void testPhysicalFromFile(String file)
                                    throws Exception

Throws:

Exception

errorMsgTestHelper

protected static void errorMsgTestHelper(String testSqlQuery,
                                         String expectedErrorMsg)
                                  throws Exception

Utility method which tests given query produces a UserException and the exception message contains the given message.

Parameters:

testSqlQuery - Test query

expectedErrorMsg - Expected error message.

Throws:

Exception

parseErrorHelper

protected static void parseErrorHelper(String testSqlQuery)
                                throws Exception

Utility method which tests given query produces a UserException with UserBitShared.DrillPBError.ErrorType being DrillPBError.ErrorType.PARSE the given message.

Parameters:

testSqlQuery - Test query

Throws:

Exception

getFile

public static String getFile(String resource)
                      throws IOException

Throws:

IOException

getPhysicalFileFromResource

public static String getPhysicalFileFromResource(String resource)
                                          throws IOException

Copy the resource (ex. file on classpath) to a physical file on FileSystem.

Parameters:

resource -

Returns:

the file path

Throws:

IOException

setSessionOption

protected static void setSessionOption(String option,
                                       boolean value)

setSessionOption

protected static void setSessionOption(String option,
                                       long value)

setSessionOption

protected static void setSessionOption(String option,
                                       double value)

setSessionOption

protected static void setSessionOption(String option,
                                       String value)

setColumnWidth

protected void setColumnWidth(int columnWidth)

setColumnWidths

protected void setColumnWidths(int[] columnWidths)

logResult

protected int logResult(List<QueryDataBatch> results)

printResult

protected int printResult(List<QueryDataBatch> results)
                   throws SchemaChangeException

Throws:

SchemaChangeException

getResultString

protected static String getResultString(List<QueryDataBatch> results,
                                        String delimiter)
                                 throws SchemaChangeException

Throws:

SchemaChangeException