org.apache.drill.exec.hive

Class HiveTestFixture

java.lang.Object

org.apache.drill.exec.hive.HiveTestFixture

public class HiveTestFixture
extends Object

Test fixture for configuration of Hive tests, which allows granular control over initialization of test data and hive storage plugin.

Below is example of usage HiveTestFixture along with ClusterFixture:

   // Note that HiveTestFixture doesn't require extension of ClusterTest,
   // this is just the simplest way for configuring test drillbit
   public class HiveTestExample extends ClusterTest {

       private static HiveTestFixture hiveTestFixture;

        @BeforeClass
       public static void setUp() throws Exception {
            startCluster(ClusterFixture.builder(dirTestWatcher));

            // Below is minimal config which uses defaults from HiveTestFixture.Builder
            // constructor, but any option for driver or storage plugin may be
            // overridden using builder's methods
            hiveTestFixture = HiveTestFixture.builder(dirTestWatcher).build();

            // Use driver manager to configure test data in Hive metastore
            hiveTestFixture.getDriverManager().runWithinSession(HiveTestExample::generateData);

            // Use plugin manager to add, remove, update hive storage plugin of one or many test drillbits
            hiveTestFixture.getPluginManager().addHivePluginTo(cluster.drillbits());
       }

       private static void generateData(Driver driver) {
            // Set up data using HiveTestUtilities.executeQuery(driver, sql)
       }

        @AfterClass
       public static void tearDown() throws Exception {
            if (nonNull(hiveTestFixture)) {
                hiveTestFixture.getPluginManager().removeHivePluginFrom(cluster.drillbits());
            }
       }
 }
 

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	HiveTestFixture.Builder
class	HiveTestFixture.HiveDriverManager Implements method for initialization and passing of Hive to consumer instances in order to be used for test data generation within session.
class	HiveTestFixture.HivePluginManager Implements addition, update and deletion of Hive storage plugin for drillbits passed from outside.

Method Summary

Modifier and Type	Method and Description
static HiveTestFixture.Builder	builder(BaseDirTestWatcher dirWatcher)
static HiveTestFixture.Builder	builder(File baseDir)
HiveTestFixture.HiveDriverManager	getDriverManager()
HiveTestFixture.HivePluginManager	getPluginManager()
String	getWarehouseDir() Returns current value of 'hive.metastore.warehouse.dir' option which expected to represent location of metastore warehouse directory.

Methods inherited from class java.lang.Object

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

Method Detail

builder

public static HiveTestFixture.Builder builder(BaseDirTestWatcher dirWatcher)

builder

public static HiveTestFixture.Builder builder(File baseDir)

getPluginManager

public HiveTestFixture.HivePluginManager getPluginManager()

getDriverManager

public HiveTestFixture.HiveDriverManager getDriverManager()

getWarehouseDir

public String getWarehouseDir()

Returns current value of 'hive.metastore.warehouse.dir' option which expected to represent location of metastore warehouse directory. Builder's user can override any option either of pluginConf or driverConf. Since setting of the option is not enforced, this method just tries to find it in any of the conf maps.

Returns:

current value of 'hive.metastore.warehouse.dir' option from pluginConf or driverConf