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