org.apache.drill.yarn.zk

Class TestZkRegistry

java.lang.Object

org.apache.drill.test.BaseTest

org.apache.drill.yarn.zk.TestZkRegistry

public class TestZkRegistry
extends BaseTest

Tests for the AM version of the cluster coordinator. The AM version has no dependencies on the DoY config system or other systems, making it easy to test in isolation using the Curator-provided test server.

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	TestZkRegistry.TestTask

Field Summary

Fields

Modifier and Type	Field and Description
static String	CLUSTER_ID
static int	TEST_CONTROL_PORT
static int	TEST_DATA_PORT
static int	TEST_USER_PORT
static String	ZK_ROOT

Constructor Summary

Constructors

Constructor and Description
TestZkRegistry()

Method Summary

Modifier and Type	Method and Description
static org.apache.curator.framework.CuratorFramework	connectToZk(String connectString) Connect to the test ZK for the simulated Drillbit side of the test.
static String	makeKey(String host)
void	testBasics() Basic setup: start a ZK and verify that the initial endpoint list is empty.
void	testCycle() Test a typical life cycle: existing Drillbit on AM start, add a Drilbit (simulates a drillbit starting), and remove a drillbit (simulates a Drillbit ending).
void	testFormat() Validate that the key format used for endpoint is the same as that generated for hosts coming from YARN.
void	testZKRegistry()

Methods inherited from class java.lang.Object

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

Field Detail

TEST_USER_PORT

public static final int TEST_USER_PORT

See Also:

Constant Field Values

TEST_CONTROL_PORT

public static final int TEST_CONTROL_PORT

See Also:

Constant Field Values

TEST_DATA_PORT

public static final int TEST_DATA_PORT

See Also:

Constant Field Values

ZK_ROOT

public static final String ZK_ROOT

See Also:

Constant Field Values

CLUSTER_ID

public static final String CLUSTER_ID

See Also:

Constant Field Values

Constructor Detail

TestZkRegistry

public TestZkRegistry()

Method Detail

testFormat

public void testFormat()

Validate that the key format used for endpoint is the same as that generated for hosts coming from YARN.

makeKey

public static String makeKey(String host)

testBasics

public void testBasics()
                throws Exception

Basic setup: start a ZK and verify that the initial endpoint list is empty. Also validates the basics of the test setup (mock server, etc.)

Throws:

Exception

testCycle

public void testCycle()
               throws Exception

Test a typical life cycle: existing Drillbit on AM start, add a Drilbit (simulates a drillbit starting), and remove a drillbit (simulates a Drillbit ending).

Throws:

Exception

connectToZk

public static org.apache.curator.framework.CuratorFramework connectToZk(String connectString)

Connect to the test ZK for the simulated Drillbit side of the test. (The AM side of the test uses the actual AM code, which is what we're testing here...)

Parameters:

connectString -

Returns:

testZKRegistry

public void testZKRegistry()
                    throws Exception

Throws:

Exception