org.apache.drill.test

Class OperatorFixture

java.lang.Object

org.apache.drill.test.BaseFixture

org.apache.drill.test.OperatorFixture

All Implemented Interfaces:

AutoCloseable

public class OperatorFixture
extends BaseFixture
implements AutoCloseable

Test fixture for operator and (especially) "sub-operator" tests. These are tests that are done without the full Drillbit server. Instead, this fixture creates a test fixture runtime environment that provides "real" implementations of the classes required by operator internals, but with implementations tuned to the test environment. The services available from this fixture are:

• Configuration (DrillConfig)
• Memory allocator
• Code generation (compilers, code cache, etc.)
• Read-only version of system and session options (which are set when creating the fixture.
• Write-only version of operator stats (which are easy to read to verify in tests.

What is not provided is anything that depends on a live server:

• Network endpoints.
• Persistent storage.
• ZK access.
• Multiple threads of execution.

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	OperatorFixture.Builder Builds an operator fixture based on a set of config options and system/session options.
static class	OperatorFixture.MockExecutorState
static class	OperatorFixture.MockFragmentContext Provide a simplified test-time code generation context that uses the same code generation mechanism as the full Drill, but provide test-specific versions of various other services.
static class	OperatorFixture.MockOperatorContext

Field Summary

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

allocator, config

Constructor Summary

Constructors

Modifier	Constructor and Description
protected	OperatorFixture(OperatorFixture.Builder builder)

Method Summary

Modifier and Type	Method and Description
static OperatorFixture.Builder	builder(BaseDirTestWatcher dirTestWatcher)
void	close()
FragmentContext	getFragmentContext()
OptionManager	getOptionManager()
OperatorContext	newOperatorContext(PhysicalOperator popConfig)
OperatorContext	operatorContext(PhysicalOperator config)
RowSet.ExtendableRowSet	rowSet(BatchSchema schema)
RowSet.ExtendableRowSet	rowSet(TupleMetadata schema)
RowSetBuilder	rowSetBuilder(BatchSchema schema)
RowSetBuilder	rowSetBuilder(TupleMetadata schema)
static OperatorFixture	standardFixture(BaseDirTestWatcher dirTestWatcher)
RowSet	wrap(VectorContainer container)
RowSet	wrap(VectorContainer container, SelectionVector2 sv2)

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

allocator, config, dirtyMemory, dirtyMemory, getTempDir

Methods inherited from class java.lang.Object

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

Constructor Detail

OperatorFixture

protected OperatorFixture(OperatorFixture.Builder builder)

Method Detail

operatorContext

public OperatorContext operatorContext(PhysicalOperator config)

getOptionManager

public OptionManager getOptionManager()

getFragmentContext

public FragmentContext getFragmentContext()

close

public void close()
           throws Exception

Specified by:

close in interface AutoCloseable

Throws:

Exception

builder

public static OperatorFixture.Builder builder(BaseDirTestWatcher dirTestWatcher)

standardFixture

public static OperatorFixture standardFixture(BaseDirTestWatcher dirTestWatcher)

rowSetBuilder

public RowSetBuilder rowSetBuilder(BatchSchema schema)

rowSetBuilder

public RowSetBuilder rowSetBuilder(TupleMetadata schema)

rowSet

public RowSet.ExtendableRowSet rowSet(BatchSchema schema)

rowSet

public RowSet.ExtendableRowSet rowSet(TupleMetadata schema)

wrap

public RowSet wrap(VectorContainer container)

newOperatorContext

public OperatorContext newOperatorContext(PhysicalOperator popConfig)

wrap

public RowSet wrap(VectorContainer container,
                   SelectionVector2 sv2)