org.apache.drill.test

Class DrillTestWrapper

java.lang.Object

org.apache.drill.test.DrillTestWrapper

public class DrillTestWrapper
extends Object

Encapsulates the options for a Drill unit test, as well as the execution methods to perform the tests and validation of results.

To construct an instance easily, look at the TestBuilder class. From an implementation of the BaseTestQuery class, and instance of the builder is accessible through the testBuilder() method.

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static interface	DrillTestWrapper.TestServices

Field Summary

Fields

Modifier and Type	Field and Description
static int	EXPECTED_BATCH_COUNT_NOT_SET
static int	EXPECTED_NUM_RECORDS_NOT_SET

Constructor Summary

Constructors

Constructor and Description

DrillTestWrapper(TestBuilder testBuilder, DrillTestWrapper.TestServices services, Object query, UserBitShared.QueryType queryType, String baselineOptionSettingQueries, String testOptionSettingQueries, UserBitShared.QueryType baselineQueryType, boolean ordered, boolean approximateEquality, double tolerance, boolean highPerformanceComparison, String[] baselineColumns, List<Map<String,Object>> baselineRecords, int expectedNumBatches, int expectedNumRecords)

Method Summary

Modifier and Type	Method and Description
static int	addToCombinedVectorResults(Iterable<VectorAccessible> batches, BatchSchema expectedSchema, Long expectedBatchSize, Integer expectedNumBatches, Map<String,List<Object>> combinedVectors, Integer expectedTotalRecords) Add to result vectors and compare batch schema against expected schema while iterating batches.
static Map<String,List<Object>>	addToCombinedVectorResults(Iterable<VectorAccessible> batches, Long expectedBatchSize, Integer expectedNumBatches, Integer expectedTotalRecords) Iterate over batches, and combine the batches into a map, where key is schema path, and value is the list of column values across all the batches.
static void	addToMaterializedResults(List<Map<String,Object>> materializedRecords, List<QueryDataBatch> records, RecordBatchLoader loader)
void	checkColumnDef(BatchSchema batchSchema)
void	compareMergedOnHeapVectors()
static void	compareMergedVectors(Map<String,List<Object>> expectedRecords, Map<String,List<Object>> actualRecords)
protected void	compareOrderedResults() Use this method only if necessary to validate one query against another.
void	compareResultsHyperVector()
protected void	compareSchemaOnly()
protected void	compareUnorderedResults() Use this method only if necessary to validate one query against another.
static boolean	compareValues(Object expected, Object actual, int counter, String column)
static boolean	compareValues(Object expected, Object actual, int counter, String column, boolean approximateEquality, double tolerance)
static boolean	compareValuesErrorOnMismatch(Object expected, Object actual, int counter, String column)
void	run()
static Map<String,List<Object>>	translateRecordListToHeapVectors(List<Map<String,Object>> records)

Methods inherited from class java.lang.Object

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

Field Detail

EXPECTED_BATCH_COUNT_NOT_SET

public static final int EXPECTED_BATCH_COUNT_NOT_SET

See Also:

Constant Field Values

EXPECTED_NUM_RECORDS_NOT_SET

public static final int EXPECTED_NUM_RECORDS_NOT_SET

See Also:

Constant Field Values

Constructor Detail

DrillTestWrapper

public DrillTestWrapper(TestBuilder testBuilder,
                        DrillTestWrapper.TestServices services,
                        Object query,
                        UserBitShared.QueryType queryType,
                        String baselineOptionSettingQueries,
                        String testOptionSettingQueries,
                        UserBitShared.QueryType baselineQueryType,
                        boolean ordered,
                        boolean approximateEquality,
                        double tolerance,
                        boolean highPerformanceComparison,
                        String[] baselineColumns,
                        List<Map<String,Object>> baselineRecords,
                        int expectedNumBatches,
                        int expectedNumRecords)

Method Detail

run

public void run()
         throws Exception

Throws:

Exception

compareMergedVectors

public static void compareMergedVectors(Map<String,List<Object>> expectedRecords,
                                        Map<String,List<Object>> actualRecords)
                                 throws Exception

Throws:

Exception

addToCombinedVectorResults

public static Map<String,List<Object>> addToCombinedVectorResults(Iterable<VectorAccessible> batches,
                                                                  Long expectedBatchSize,
                                                                  Integer expectedNumBatches,
                                                                  Integer expectedTotalRecords)
                                                           throws SchemaChangeException,
                                                                  UnsupportedEncodingException

Iterate over batches, and combine the batches into a map, where key is schema path, and value is the list of column values across all the batches.

Parameters:

batches -

expectedTotalRecords -

Returns:

Throws:

SchemaChangeException

UnsupportedEncodingException

addToCombinedVectorResults

public static int addToCombinedVectorResults(Iterable<VectorAccessible> batches,
                                             BatchSchema expectedSchema,
                                             Long expectedBatchSize,
                                             Integer expectedNumBatches,
                                             Map<String,List<Object>> combinedVectors,
                                             Integer expectedTotalRecords)
                                      throws SchemaChangeException,
                                             UnsupportedEncodingException

Add to result vectors and compare batch schema against expected schema while iterating batches.

Parameters:

batches -

expectedSchema: - the expected schema the batches should contain. Through SchemaChangeException if encounter different batch schema.

combinedVectors: - the vectors to hold the values when iterate the batches.

Returns:

number of batches

Throws:

SchemaChangeException

UnsupportedEncodingException

compareSchemaOnly

protected void compareSchemaOnly()
                          throws Exception

Throws:

Exception

compareUnorderedResults

protected void compareUnorderedResults()
                                throws Exception

Use this method only if necessary to validate one query against another. If you are just validating against a baseline file use one of the simpler interfaces that will write the validation query for you.

Throws:

Exception

checkColumnDef

public void checkColumnDef(BatchSchema batchSchema)
                    throws Exception

Throws:

Exception

compareOrderedResults

protected void compareOrderedResults()
                              throws Exception

Use this method only if necessary to validate one query against another. If you are just validating against a baseline file use one of the simpler interfaces that will write the validation query for you.

Throws:

Exception

compareMergedOnHeapVectors

public void compareMergedOnHeapVectors()
                                throws Exception

Throws:

Exception

translateRecordListToHeapVectors

public static Map<String,List<Object>> translateRecordListToHeapVectors(List<Map<String,Object>> records)

compareResultsHyperVector

public void compareResultsHyperVector()
                               throws Exception

Throws:

Exception

addToMaterializedResults

public static void addToMaterializedResults(List<Map<String,Object>> materializedRecords,
                                            List<QueryDataBatch> records,
                                            RecordBatchLoader loader)
                                     throws SchemaChangeException,
                                            UnsupportedEncodingException

Throws:

SchemaChangeException

UnsupportedEncodingException

compareValuesErrorOnMismatch

public static boolean compareValuesErrorOnMismatch(Object expected,
                                                   Object actual,
                                                   int counter,
                                                   String column)
                                            throws Exception

Throws:

Exception

compareValues

public static boolean compareValues(Object expected,
                                    Object actual,
                                    int counter,
                                    String column)
                             throws Exception

Throws:

Exception

compareValues

public static boolean compareValues(Object expected,
                                    Object actual,
                                    int counter,
                                    String column,
                                    boolean approximateEquality,
                                    double tolerance)
                             throws Exception

Throws:

Exception