001/** 002 * Licensed to the Apache Software Foundation (ASF) under one 003 * or more contributor license agreements. See the NOTICE file 004 * distributed with this work for additional information 005 * regarding copyright ownership. The ASF licenses this file 006 * to you under the Apache License, Version 2.0 (the 007 * "License"); you may not use this file except in compliance 008 * with the License. You may obtain a copy of the License at 009 * 010 * http://www.apache.org/licenses/LICENSE-2.0 011 * 012 * Unless required by applicable law or agreed to in writing, software 013 * distributed under the License is distributed on an "AS IS" BASIS, 014 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 015 * See the License for the specific language governing permissions and 016 * limitations under the License. 017 */ 018package org.apache.hadoop.hbase.client; 019 020import static org.apache.hadoop.hbase.client.RegionReplicaTestHelper.testLocator; 021 022import org.apache.commons.io.IOUtils; 023import org.apache.hadoop.hbase.HBaseClassTestRule; 024import org.apache.hadoop.hbase.HBaseTestingUtility; 025import org.apache.hadoop.hbase.HConstants; 026import org.apache.hadoop.hbase.HRegionLocation; 027import org.apache.hadoop.hbase.RegionLocations; 028import org.apache.hadoop.hbase.TableName; 029import org.apache.hadoop.hbase.client.RegionReplicaTestHelper.Locator; 030import org.apache.hadoop.hbase.master.balancer.BaseLoadBalancer; 031import org.apache.hadoop.hbase.testclassification.ClientTests; 032import org.apache.hadoop.hbase.testclassification.MediumTests; 033import org.junit.AfterClass; 034import org.junit.BeforeClass; 035import org.junit.ClassRule; 036import org.junit.Test; 037import org.junit.experimental.categories.Category; 038 039@Category({ MediumTests.class, ClientTests.class }) 040public class TestAsyncMetaRegionLocator { 041 042 @ClassRule 043 public static final HBaseClassTestRule CLASS_RULE = 044 HBaseClassTestRule.forClass(TestAsyncMetaRegionLocator.class); 045 046 private static final HBaseTestingUtility TEST_UTIL = new HBaseTestingUtility(); 047 048 private static AsyncRegistry REGISTRY; 049 050 private static AsyncMetaRegionLocator LOCATOR; 051 052 @BeforeClass 053 public static void setUp() throws Exception { 054 TEST_UTIL.getConfiguration().set(BaseLoadBalancer.TABLES_ON_MASTER, "none"); 055 TEST_UTIL.getConfiguration().setInt(HConstants.META_REPLICAS_NUM, 3); 056 TEST_UTIL.startMiniCluster(3); 057 REGISTRY = AsyncRegistryFactory.getRegistry(TEST_UTIL.getConfiguration()); 058 RegionReplicaTestHelper 059 .waitUntilAllMetaReplicasHavingRegionLocation(TEST_UTIL.getConfiguration(), REGISTRY, 3); 060 TEST_UTIL.getAdmin().balancerSwitch(false, true); 061 LOCATOR = new AsyncMetaRegionLocator(REGISTRY); 062 } 063 064 @AfterClass 065 public static void tearDown() throws Exception { 066 IOUtils.closeQuietly(REGISTRY); 067 TEST_UTIL.shutdownMiniCluster(); 068 } 069 070 @Test 071 public void test() throws Exception { 072 testLocator(TEST_UTIL, TableName.META_TABLE_NAME, new Locator() { 073 074 @Override 075 public void updateCachedLocationOnError(HRegionLocation loc, Throwable error) 076 throws Exception { 077 LOCATOR.updateCachedLocationOnError(loc, error); 078 } 079 080 @Override 081 public RegionLocations getRegionLocations(TableName tableName, int replicaId, boolean reload) 082 throws Exception { 083 return LOCATOR.getRegionLocations(replicaId, reload).get(); 084 } 085 }); 086 } 087}