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.regionserver; 019 020import java.io.IOException; 021import java.io.StringWriter; 022import java.util.List; 023import org.apache.hadoop.hbase.HBaseClassTestRule; 024import org.apache.hadoop.hbase.HBaseConfiguration; 025import org.apache.hadoop.hbase.HConstants; 026import org.apache.hadoop.hbase.HTableDescriptor; 027import org.apache.hadoop.hbase.ServerName; 028import org.apache.hadoop.hbase.TableName; 029import org.apache.hadoop.hbase.client.RegionInfo; 030import org.apache.hadoop.hbase.client.RegionInfoBuilder; 031import org.apache.hadoop.hbase.io.hfile.CacheConfig; 032import org.apache.hadoop.hbase.ipc.MetricsHBaseServer; 033import org.apache.hadoop.hbase.ipc.MetricsHBaseServerWrapperStub; 034import org.apache.hadoop.hbase.ipc.RpcServerInterface; 035import org.apache.hadoop.hbase.testclassification.RegionServerTests; 036import org.apache.hadoop.hbase.testclassification.SmallTests; 037import org.apache.hadoop.hbase.tmpl.regionserver.RSStatusTmpl; 038import org.apache.hadoop.hbase.util.Bytes; 039import org.apache.hadoop.hbase.zookeeper.MasterAddressTracker; 040import org.apache.hadoop.hbase.zookeeper.ZKWatcher; 041import org.junit.Before; 042import org.junit.ClassRule; 043import org.junit.Rule; 044import org.junit.Test; 045import org.junit.experimental.categories.Category; 046import org.junit.rules.TestName; 047import org.mockito.Mockito; 048import org.slf4j.Logger; 049import org.slf4j.LoggerFactory; 050 051import org.apache.hbase.thirdparty.com.google.common.collect.Lists; 052import org.apache.hbase.thirdparty.com.google.protobuf.RpcController; 053import org.apache.hbase.thirdparty.com.google.protobuf.ServiceException; 054 055import org.apache.hadoop.hbase.shaded.protobuf.ResponseConverter; 056import org.apache.hadoop.hbase.shaded.protobuf.generated.AdminProtos.GetOnlineRegionRequest; 057import org.apache.hadoop.hbase.shaded.protobuf.generated.AdminProtos.GetServerInfoRequest; 058import org.apache.hadoop.hbase.shaded.protobuf.generated.AdminProtos.GetServerInfoResponse; 059 060/** 061 * Tests for the region server status page and its template. 062 */ 063@Category({RegionServerTests.class, SmallTests.class}) 064public class TestRSStatusServlet { 065 066 @ClassRule 067 public static final HBaseClassTestRule CLASS_RULE = 068 HBaseClassTestRule.forClass(TestRSStatusServlet.class); 069 070 private static final Logger LOG = LoggerFactory.getLogger(TestRSStatusServlet.class); 071 private HRegionServer rs; 072 private RSRpcServices rpcServices; 073 private RpcServerInterface rpcServer; 074 075 static final int FAKE_IPC_PORT = 1585; 076 static final int FAKE_WEB_PORT = 1586; 077 078 private final ServerName fakeServerName = 079 ServerName.valueOf("localhost", FAKE_IPC_PORT, 11111); 080 private final GetServerInfoResponse fakeResponse = 081 ResponseConverter.buildGetServerInfoResponse(fakeServerName, FAKE_WEB_PORT); 082 083 private final ServerName fakeMasterAddress = 084 ServerName.valueOf("localhost", 60010, 1212121212); 085 086 @Rule 087 public TestName name = new TestName(); 088 089 @Before 090 public void setupBasicMocks() throws IOException, ServiceException { 091 rs = Mockito.mock(HRegionServer.class); 092 rpcServices = Mockito.mock(RSRpcServices.class); 093 rpcServer = Mockito.mock(RpcServerInterface.class); 094 Mockito.doReturn(HBaseConfiguration.create()) 095 .when(rs).getConfiguration(); 096 Mockito.doReturn(rpcServices).when(rs).getRSRpcServices(); 097 Mockito.doReturn(rpcServer).when(rs).getRpcServer(); 098 Mockito.doReturn(fakeResponse).when(rpcServices).getServerInfo( 099 (RpcController)Mockito.any(), (GetServerInfoRequest)Mockito.any()); 100 // Fake ZKW 101 ZKWatcher zkw = Mockito.mock(ZKWatcher.class); 102 Mockito.doReturn("fakequorum").when(zkw).getQuorum(); 103 Mockito.doReturn(zkw).when(rs).getZooKeeper(); 104 105 // Fake CacheConfig 106 LOG.warn("The " + HConstants.HFILE_BLOCK_CACHE_SIZE_KEY + " is set to 0"); 107 CacheConfig cacheConf = Mockito.mock(CacheConfig.class); 108 Mockito.doReturn(null).when(cacheConf).getBlockCache(); 109 Mockito.doReturn(cacheConf).when(rs).getCacheConfig(); 110 111 // Fake MasterAddressTracker 112 MasterAddressTracker mat = Mockito.mock(MasterAddressTracker.class); 113 Mockito.doReturn(fakeMasterAddress).when(mat).getMasterAddress(); 114 Mockito.doReturn(mat).when(rs).getMasterAddressTracker(); 115 116 MetricsRegionServer rms = Mockito.mock(MetricsRegionServer.class); 117 Mockito.doReturn(new MetricsRegionServerWrapperStub()).when(rms).getRegionServerWrapper(); 118 Mockito.doReturn(rms).when(rs).getRegionServerMetrics(); 119 120 MetricsHBaseServer ms = Mockito.mock(MetricsHBaseServer.class); 121 Mockito.doReturn(new MetricsHBaseServerWrapperStub()).when(ms).getHBaseServerWrapper(); 122 Mockito.doReturn(ms).when(rpcServer).getMetrics(); 123 } 124 125 @Test 126 public void testBasic() throws IOException, ServiceException { 127 new RSStatusTmpl().render(new StringWriter(), rs); 128 } 129 130 @Test 131 public void testWithRegions() throws IOException, ServiceException { 132 HTableDescriptor htd = new HTableDescriptor(TableName.valueOf(name.getMethodName())); 133 List<RegionInfo> regions = Lists.newArrayList( 134 RegionInfoBuilder.newBuilder(htd.getTableName()) 135 .setStartKey(Bytes.toBytes("a")) 136 .setEndKey(Bytes.toBytes("d")) 137 .build(), 138 RegionInfoBuilder.newBuilder(htd.getTableName()) 139 .setStartKey(Bytes.toBytes("d")) 140 .setEndKey(Bytes.toBytes("z")) 141 .build() 142 ); 143 Mockito.doReturn(ResponseConverter.buildGetOnlineRegionResponse( 144 regions)).when(rpcServices).getOnlineRegion((RpcController)Mockito.any(), 145 (GetOnlineRegionRequest)Mockito.any()); 146 new RSStatusTmpl().render(new StringWriter(), rs); 147 } 148}