org.finra.hqlunit.rules

Class TestDataLoader

java.lang.Object

org.finra.hqlunit.rules.TestDataLoader

All Implemented Interfaces:

org.junit.rules.TestRule

public class TestDataLoader
extends Object
implements org.junit.rules.TestRule

This TestRule class provides utility functions for extracting testing data from resources and loading it into Hive tables

Constructor Summary

Constructors

Constructor and Description
TestDataLoader(TestHiveServer hiveServer) Constructs a new TestDataLoader

Method Summary

Modifier and Type	Method and Description
org.junit.runners.model.Statement	apply(org.junit.runners.model.Statement base, org.junit.runner.Description description) Uses a TemporaryFolder rule to process a given statement The TemporaryFolder will be used to construct staging files for loading data into Hive
void	loadDataIntoTable(String tableName, TextResource tableDataResource) Loads data from a TextResource into a hive table A temporary file on the local file system is used as a staging ground for the data
void	loadDataIntoTable(String tableName, TextResource tableDataResource, String partitionInfo) Loads data from a TextResource into a hive table A temporary file on the local file system is used as a staging ground for the data

Methods inherited from class java.lang.Object

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

Constructor Detail

TestDataLoader

public TestDataLoader(TestHiveServer hiveServer)

Constructs a new TestDataLoader

Parameters:

hiveServer - the TestHiveServer to load data into

Method Detail

apply

public org.junit.runners.model.Statement apply(org.junit.runners.model.Statement base,
                                               org.junit.runner.Description description)

Uses a TemporaryFolder rule to process a given statement The TemporaryFolder will be used to construct staging files for loading data into Hive

Specified by:

apply in interface org.junit.rules.TestRule

Parameters:

base - the statement to apply a TemporaryFolder to

description - the description to apply a TemporaryFolder to

Returns:

the output of the TemporaryFolder rule

loadDataIntoTable

public void loadDataIntoTable(String tableName,
                              TextResource tableDataResource)

Loads data from a TextResource into a hive table A temporary file on the local file system is used as a staging ground for the data

Parameters:

tableName - the name of the table

tableDataResource - the resource to extract data from

loadDataIntoTable

public void loadDataIntoTable(String tableName,
                              TextResource tableDataResource,
                              String partitionInfo)

Loads data from a TextResource into a hive table A temporary file on the local file system is used as a staging ground for the data

Parameters:

tableName - the name of the table

tableDataResource - the resource to extract data from

partitionInfo - the optional partitioning commands at the end of a 'LOAD DATA' query