From da471704bedb73570fac7a802fce0ad10ac96b8e Mon Sep 17 00:00:00 2001 From: haxiaolin Date: Thu, 8 Feb 2018 19:37:12 +0800 Subject: [PATCH] HBASE-19937 Ensure createRSGroupTable be called after ProcedureExecutor and LoadBalancer are initialized --- .../hadoop/hbase/rsgroup/TestEnableRSGroup.java | 70 ++++++++++++++++++++++ 1 file changed, 70 insertions(+) create mode 100644 hbase-rsgroup/src/test/java/org/apache/hadoop/hbase/rsgroup/TestEnableRSGroup.java diff --git a/hbase-rsgroup/src/test/java/org/apache/hadoop/hbase/rsgroup/TestEnableRSGroup.java b/hbase-rsgroup/src/test/java/org/apache/hadoop/hbase/rsgroup/TestEnableRSGroup.java new file mode 100644 index 0000000..7bd039e --- /dev/null +++ b/hbase-rsgroup/src/test/java/org/apache/hadoop/hbase/rsgroup/TestEnableRSGroup.java @@ -0,0 +1,70 @@ +package org.apache.hadoop.hbase.rsgroup; + +import java.io.IOException; + +import org.apache.hadoop.conf.Configuration; +import org.apache.hadoop.hbase.HBaseClassTestRule; +import org.apache.hadoop.hbase.HBaseTestingUtility; +import org.apache.hadoop.hbase.HConstants; +import org.apache.hadoop.hbase.MiniHBaseCluster; +import org.apache.hadoop.hbase.coprocessor.CoprocessorHost; +import org.apache.hadoop.hbase.testclassification.SmallTests; +import org.junit.After; +import org.junit.Before; +import org.junit.ClassRule; +import org.junit.Test; +import org.junit.experimental.categories.Category; +import org.slf4j.Logger; +import org.slf4j.LoggerFactory; + +import static org.apache.hadoop.hbase.coprocessor.CoprocessorHost.COPROCESSORS_ENABLED_CONF_KEY; +import static org.apache.hadoop.hbase.coprocessor.CoprocessorHost.DEFAULT_COPROCESSORS_ENABLED; +import static org.junit.Assert.assertTrue; + +/** + * Test enable RSGroup + */ +@Category({ SmallTests.class }) public class TestEnableRSGroup { + + @ClassRule public static final HBaseClassTestRule CLASS_RULE = + HBaseClassTestRule.forClass(TestEnableRSGroup.class); + + protected static final Logger LOG = LoggerFactory.getLogger(TestEnableRSGroup.class); + + private static HBaseTestingUtility TEST_UTIL; + private MiniHBaseCluster miniHBaseCluster; + private Configuration conf; + + @Before public void setUp() throws Exception { + TEST_UTIL = new HBaseTestingUtility(); + TEST_UTIL.startMiniCluster(); + miniHBaseCluster = TEST_UTIL.getMiniHBaseCluster(); + conf = TEST_UTIL.getConfiguration(); + } + + @After public void tearDown() throws Exception { + TEST_UTIL.shutdownMiniCluster(); + } + + @Test public void testEnableRSGroup() throws IOException { + miniHBaseCluster.stopMaster(0); + LOG.info("stopped master..."); + + // set configs about enable rsgroup + conf.setBoolean(COPROCESSORS_ENABLED_CONF_KEY, DEFAULT_COPROCESSORS_ENABLED); + conf.set(CoprocessorHost.MASTER_COPROCESSOR_CONF_KEY, RSGroupAdminEndpoint.class.getName()); + conf.set(HConstants.HBASE_MASTER_LOADBALANCER_CLASS, RSGroupBasedLoadBalancer.class.getName()); + miniHBaseCluster.setConf(conf); + miniHBaseCluster.startMaster(); + miniHBaseCluster.waitForActiveAndReadyMaster(60000); + LOG.info("started master..."); + + // check if master started successfully + assertTrue(TEST_UTIL.getMiniHBaseCluster().getMaster() != null); + + RSGroupBasedLoadBalancer loadBalancer = + (RSGroupBasedLoadBalancer) TEST_UTIL.getMiniHBaseCluster().getMaster().getLoadBalancer(); + assertTrue(loadBalancer.isOnline()); + } + +} -- 1.9.1