org.apache.drill.exec.sql

Class TestBaseViewSupport

java.lang.Object

DrillTest

org.apache.drill.exec.ExecTest

org.apache.drill.test.BaseTestQuery

org.apache.drill.PlanTestBase

org.apache.drill.exec.sql.TestBaseViewSupport

Direct Known Subclasses:

TestViewSupport

public class TestBaseViewSupport
extends PlanTestBase

Base class for view tests. It has utility methods which can be used when writing tests for views on tables in different storage engines such as Hive, HBase etc.

Nested Class Summary

Nested classes/interfaces inherited from class org.apache.drill.test.BaseTestQuery

BaseTestQuery.ClassicTestServices, BaseTestQuery.SilentListener, BaseTestQuery.TestResultSet

Field Summary

Fields inherited from class org.apache.drill.PlanTestBase

EXPECTED_NOT_FOUND, JSON_FORMAT, OPTIQ_FORMAT, UNEXPECTED_FOUND

Fields inherited from class org.apache.drill.test.BaseTestQuery

allocator, bits, client, config, serviceSet

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

c, dirTestWatcher, optionManager

Constructor Summary

Constructors

Constructor and Description
TestBaseViewSupport()

Method Summary

Modifier and Type	Method and Description
protected static String[]	baselineColumns(String... names) Convenient method for defining baselineColumns to be passed into view helper methods.
protected static List<Object[]>	baselineRows(Object[]... rows) Convenient method for grouping of expected rows into list of Object arrays, where each array represents concrete row.
protected static void	createViewHelper(String viewSchema, String viewName, String finalSchema, String viewFields, String viewDef) Create view with given parameters.
protected static void	dropViewHelper(String viewSchema, String viewName, String finalSchema) Drop view with given parameters.
protected static void	dropViewIfExistsHelper(String viewSchema, String viewName, String finalSchema, Boolean ifViewExists) Drop view if exists with given parameters.
protected static String	generateViewName() Generate a view name by appending the test class with a increasing counter.
protected static void	queryViewHelper(String query, String[] baselineColumns, List<Object[]> baselineValues) Execute the given query and check against the given baseline.
protected static Object[]	row(Object... columns) Helper method for conversion of Object varargs into array of objects.
protected static void	testViewHelper(String finalSchema, String viewFields, String viewDef, String queryOnView, String[] expectedBaselineColumns, List<Object[]> expectedBaselineValues) Tests creating a view with given parameters, query the view and check against the provided baselines and finally drop the view.

Methods inherited from class org.apache.drill.PlanTestBase

checkForMetadataFile, createMetadataDir, getPhysicalJsonPlan, getPlanInString, testPhysicalPlan, testPhysicalPlanExecutionBasedOnQuery, testPlanMatchingPatterns, testPlanMatchingPatterns, testPlanMatchingPatterns, testPlanMatchingPatterns, testPlanOneExcludedPattern, testPlanOneExpectedPattern, testPlanOneExpectedPatternOneExcluded, testPlanSubstrPatterns, testPlanWithAttributesMatchingPatterns, testRelLogicalJoinOrder, testRelLogicalPlanLevDigest, testRelLogicalPlanLevExplain, testRelPhysicalJoinOrder, testRelPhysicalPlanLevDigest, testRelPhysicalPlanLevExplain

Methods inherited from class org.apache.drill.test.BaseTestQuery

alterSession, cloneDefaultTestConfigProperties, closeClient, errorMsgTestHelper, getAllocator, getDrillbitContext, getFile, getPhysicalFileFromResource, getResultString, getUserPort, logResult, newTest, parseErrorHelper, printResult, resetAllSessionOptions, resetDrillbitCount, resetSessionOption, runSQL, setColumnWidth, setColumnWidths, setSessionOption, setSessionOption, setSessionOption, setSessionOption, setupDefaultTestCluster, test, test, testBuilder, testLogicalWithResults, testNoResult, testNoResult, testPhysical, testPhysicalFromFile, testPhysicalWithResults, testPreparedStatement, testRunAndPrint, testRunAndReturn, testSql, testSqlWithResults, testWithListener, updateClient, updateClient, updateClient, updateTestCluster, updateTestCluster

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

Constructor Detail

TestBaseViewSupport

public TestBaseViewSupport()

Method Detail

createViewHelper

protected static void createViewHelper(String viewSchema,
                                       String viewName,
                                       String finalSchema,
                                       String viewFields,
                                       String viewDef)
                                throws Exception

Create view with given parameters. Current default schema "dfs" CREATE VIEW tmp.viewName(f1, f2) AS SELECT * FROM cp.`region.json` For the above CREATE VIEW query, function parameters are: viewSchema = "tmp" viewName = "viewName" finalSchema = "dfs.tmp" viewFields = "(f1, f2)" viewDef = "SELECT * FROM cp.`region.json`"

Parameters:

viewSchema - Schema name to prefix when referring to the view.

viewName - Name of the view.

finalSchema - Absolute schema path where the view is created. Pameter viewSchema may refer the schema with respect the current default schema. Combining viewSchema with default schema gives the final schema.

viewFields - If the created view needs to specify the fields, otherwise null

viewDef - Definition of the view.

Throws:

Exception

dropViewHelper

protected static void dropViewHelper(String viewSchema,
                                     String viewName,
                                     String finalSchema)
                              throws Exception

Drop view with given parameters. Current schema "dfs" DROP VIEW tmp.viewName For the above DROP VIEW query, function parameters values are: viewSchema = "tmp" "viewName" = "viewName" "finalSchema" = "dfs.tmp"

Parameters:

viewSchema -

viewName -

finalSchema -

Throws:

Exception

dropViewIfExistsHelper

protected static void dropViewIfExistsHelper(String viewSchema,
                                             String viewName,
                                             String finalSchema,
                                             Boolean ifViewExists)
                                      throws Exception

Drop view if exists with given parameters. Current schema "dfs" DROP VIEW IF EXISTS tmp.viewName For the above DROP VIEW IF EXISTS query, function parameters values are: viewSchema = "tmp" "viewName" = "viewName" "finalSchema" = "dfs.tmp" "ifViewExists" = null

Parameters:

viewSchema -

viewName -

finalSchema -

ifViewExists - Helps to check query result depending from the existing of the view.

Throws:

Exception

queryViewHelper

protected static void queryViewHelper(String query,
                                      String[] baselineColumns,
                                      List<Object[]> baselineValues)
                               throws Exception

Execute the given query and check against the given baseline.

Parameters:

query -

baselineColumns -

baselineValues -

Throws:

Exception

generateViewName

protected static String generateViewName()

Generate a view name by appending the test class with a increasing counter.

Returns:

testViewHelper

protected static void testViewHelper(String finalSchema,
                                     String viewFields,
                                     String viewDef,
                                     String queryOnView,
                                     String[] expectedBaselineColumns,
                                     List<Object[]> expectedBaselineValues)
                              throws Exception

Tests creating a view with given parameters, query the view and check against the provided baselines and finally drop the view.

Parameters:

finalSchema - Absolute schema path where the view is going to be created.

viewFields - If view has any field list

viewDef - Definition of the view.

queryOnView - Query to run on created view. Refer to test view using "TEST_VIEW_NAME". Similarly schema using "TEST_SCHEMA".

expectedBaselineColumns - Expected columns from querying the view.

expectedBaselineValues - Expected row values from querying the view.

Throws:

Exception

baselineColumns

protected static String[] baselineColumns(String... names)

Convenient method for defining baselineColumns to be passed into view helper methods.

Parameters:

names - column names varargs

Returns:

column names array

baselineRows

protected static List<Object[]> baselineRows(Object[]... rows)

Convenient method for grouping of expected rows into list of Object arrays, where each array represents concrete row. This method is used for defining baselineValues and passing it to view helper methods.

Parameters:

rows - rows in form of Object[] varargs

Returns:

list of rows

row

protected static Object[] row(Object... columns)

Helper method for conversion of Object varargs into array of objects. Used for passing rows into TestBaseViewSupport#baselineRows(Object[]...)

Parameters:

columns - Object varargs

Returns:

array of passed objects