hbase-server/src/test/java/org/apache/hadoop/hbase/master/TestRoundRobinAssignmentOnRestart.java

   1 /**
   2  * Licensed to the Apache Software Foundation (ASF) under one
   3  * or more contributor license agreements.  See the NOTICE file
   4  * distributed with this work for additional information
   5  * regarding copyright ownership.  The ASF licenses this file
   6  * to you under the Apache License, Version 2.0 (the
   7  * "License"); you may not use this file except in compliance
   8  * with the License.  You may obtain a copy of the License at
   9  *
  10  *     http://www.apache.org/licenses/LICENSE-2.0
  11  *
  12  * Unless required by applicable law or agreed to in writing, software
  13  * distributed under the License is distributed on an "AS IS" BASIS,
  14  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  15  * See the License for the specific language governing permissions and
  16  * limitations under the License.
  17  */
  18 package org.apache.hadoop.hbase.master;
  19
  20 import static org.junit.Assert.assertEquals;
  21 import static org.junit.Assert.assertNotNull;
  22 import static org.junit.Assert.assertTrue;
  23
  24 import java.util.List;
  25
  26 import org.apache.hadoop.hbase.HBaseClassTestRule;
  27 import org.apache.hadoop.hbase.HConstants;
  28 import org.apache.hadoop.hbase.ServerName;
  29 import org.apache.hadoop.hbase.SingleProcessHBaseCluster;
  30 import org.apache.hadoop.hbase.TableName;
  31 import org.apache.hadoop.hbase.client.RegionInfo;
  32 import org.apache.hadoop.hbase.testclassification.MasterTests;
  33 import org.apache.hadoop.hbase.testclassification.MediumTests;
  34 import org.apache.hadoop.hbase.util.JVMClusterUtil;
  35 import org.junit.ClassRule;
  36 import org.junit.Test;
  37 import org.junit.experimental.categories.Category;
  38 import org.slf4j.Logger;
  39 import org.slf4j.LoggerFactory;
  40
  41 @Category({ MasterTests.class, MediumTests.class })
  42 public class TestRoundRobinAssignmentOnRestart extends AbstractTestRestartCluster {
  43
  44   @ClassRule
  45   public static final HBaseClassTestRule CLASS_RULE =
  46       HBaseClassTestRule.forClass(TestRoundRobinAssignmentOnRestart.class);
  47
  48   private static final Logger LOG =
  49       LoggerFactory.getLogger(TestRoundRobinAssignmentOnRestart.class);
  50
  51   @Override
  52   protected boolean splitWALCoordinatedByZk() {
  53     return true;
  54   }
  55
  56   private final int regionNum = 10;
  57   private final int rsNum = 2;
  58
  59   /**
  60    * This tests retaining assignments on a cluster restart
  61    */
  62   @Test
  63   public void test() throws Exception {
  64     UTIL.startMiniCluster(rsNum);
  65     // Turn off balancer
  66     UTIL.getMiniHBaseCluster().getMaster().getMasterRpcServices().synchronousBalanceSwitch(false);
  67     LOG.info("\n\nCreating tables");
  68     for (TableName TABLE : TABLES) {
  69       UTIL.createMultiRegionTable(TABLE, FAMILY, regionNum);
  70     }
  71     // Wait until all regions are assigned
  72     for (TableName TABLE : TABLES) {
  73       UTIL.waitTableEnabled(TABLE);
  74     }
  75     UTIL.waitUntilNoRegionsInTransition(60000);
  76
  77     SingleProcessHBaseCluster cluster = UTIL.getHBaseCluster();
  78     List<JVMClusterUtil.RegionServerThread> threads = cluster.getLiveRegionServerThreads();
  79     assertEquals(2, threads.size());
  80
  81     ServerName testServer = threads.get(0).getRegionServer().getServerName();
  82     int port = testServer.getPort();
  83     List<RegionInfo> regionInfos =
  84         cluster.getMaster().getAssignmentManager().getRegionsOnServer(testServer);
  85     LOG.debug("RegionServer {} has {} regions", testServer, regionInfos.size());
  86     assertTrue(regionInfos.size() >= (TABLES.length * regionNum / rsNum));
  87
  88     // Restart 1 regionserver
  89     cluster.stopRegionServer(testServer);
  90     cluster.waitForRegionServerToStop(testServer, 60000);
  91     cluster.getConf().setInt(HConstants.REGIONSERVER_PORT, port);
  92     cluster.startRegionServer();
  93
  94     HMaster master = UTIL.getMiniHBaseCluster().getMaster();
  95     List<ServerName> localServers = master.getServerManager().getOnlineServersList();
  96     ServerName newTestServer = null;
  97     for (ServerName serverName : localServers) {
  98       if (serverName.getAddress().equals(testServer.getAddress())) {
  99         newTestServer = serverName;
 100         break;
 101       }
 102     }
 103     assertNotNull(newTestServer);
 104
 105     // Wait until all regions are assigned
 106     for (TableName TABLE : TABLES) {
 107       UTIL.waitTableAvailable(TABLE);
 108     }
 109     UTIL.waitUntilNoRegionsInTransition(60000);
 110
 111     List<RegionInfo> newRegionInfos =
 112         cluster.getMaster().getAssignmentManager().getRegionsOnServer(newTestServer);
 113     LOG.debug("RegionServer {} has {} regions", newTestServer, newRegionInfos.size());
 114     assertTrue("Should not retain all regions when restart",
 115         newRegionInfos.size() < regionInfos.size());
 116   }
 117 }