/** * Licensed to the Apache Software Foundation (ASF) under one * or more contributor license agreements. See the NOTICE file * distributed with this work for additional information * regarding copyright ownership. The ASF licenses this file * to you under the Apache License, Version 2.0 (the * "License"); you may not use this file except in compliance * with the License. You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ package org.apache.hadoop.hbase.master; import java.io.IOException; import java.io.StringWriter; import java.util.ArrayList; import java.util.Arrays; import java.util.HashSet; import java.util.List; import java.util.Set; import org.apache.hadoop.conf.Configuration; import org.apache.hadoop.hbase.HBaseClassTestRule; import org.apache.hadoop.hbase.HBaseConfiguration; import org.apache.hadoop.hbase.ServerName; import org.apache.hadoop.hbase.TableName; import org.apache.hadoop.hbase.client.Admin; import org.apache.hadoop.hbase.client.RegionInfo; import org.apache.hadoop.hbase.client.RegionInfoBuilder; import org.apache.hadoop.hbase.client.TableDescriptor; import org.apache.hadoop.hbase.client.TableDescriptorBuilder; import org.apache.hadoop.hbase.master.assignment.AssignmentManager; import org.apache.hadoop.hbase.master.assignment.RegionStates; import org.apache.hadoop.hbase.regionserver.MetricsRegionServer; import org.apache.hadoop.hbase.regionserver.MetricsRegionServerWrapperStub; import org.apache.hadoop.hbase.testclassification.MasterTests; import org.apache.hadoop.hbase.testclassification.MediumTests; import org.apache.hadoop.hbase.tmpl.master.MasterStatusTmpl; import org.apache.hadoop.hbase.util.Bytes; import org.apache.hadoop.hbase.zookeeper.MasterAddressTracker; import org.apache.hadoop.hbase.zookeeper.ZKWatcher; import org.apache.hadoop.hbase.zookeeper.ZNodePaths; import org.junit.Before; import org.junit.ClassRule; import org.junit.Test; import org.junit.experimental.categories.Category; import org.mockito.Mockito; import org.apache.hbase.thirdparty.com.google.common.collect.Lists; /** * Tests for the master status page and its template. */ @Category({MasterTests.class,MediumTests.class}) public class TestMasterStatusServlet { @ClassRule public static final HBaseClassTestRule CLASS_RULE = HBaseClassTestRule.forClass(TestMasterStatusServlet.class); private HMaster master; private Configuration conf; private Admin admin; static final ServerName FAKE_HOST = ServerName.valueOf("fakehost", 12345, 1234567890); static final TableDescriptor FAKE_TABLE = TableDescriptorBuilder.newBuilder(TableName.valueOf("mytable")).build(); static final RegionInfo FAKE_HRI = RegionInfoBuilder.newBuilder(FAKE_TABLE.getTableName()) .setStartKey(Bytes.toBytes("a")).setEndKey(Bytes.toBytes("b")).build(); @Before public void setupBasicMocks() { conf = HBaseConfiguration.create(); master = Mockito.mock(HMaster.class); Mockito.doReturn(FAKE_HOST).when(master).getServerName(); Mockito.doReturn(conf).when(master).getConfiguration(); //Fake DeadServer DeadServer deadServer = Mockito.mock(DeadServer.class); // Fake serverManager ServerManager serverManager = Mockito.mock(ServerManager.class); Mockito.doReturn(1.0).when(serverManager).getAverageLoad(); Mockito.doReturn(serverManager).when(master).getServerManager(); Mockito.doReturn(deadServer).when(serverManager).getDeadServers(); // Fake AssignmentManager and RIT AssignmentManager am = Mockito.mock(AssignmentManager.class); RegionStates rs = Mockito.mock(RegionStates.class); List<RegionState> regionsInTransition = new ArrayList<>(); regionsInTransition.add(new RegionState(FAKE_HRI, RegionState.State.CLOSING, 12345L, FAKE_HOST)); Mockito.doReturn(rs).when(am).getRegionStates(); Mockito.doReturn(regionsInTransition).when(rs).getRegionsInTransition(); Mockito.doReturn(am).when(master).getAssignmentManager(); Mockito.doReturn(serverManager).when(master).getServerManager(); // Fake ZKW ZKWatcher zkw = Mockito.mock(ZKWatcher.class); Mockito.doReturn(new ZNodePaths(conf)).when(zkw).getZNodePaths(); Mockito.doReturn("fakequorum").when(zkw).getQuorum(); Mockito.doReturn(zkw).when(master).getZooKeeper(); // Fake MasterAddressTracker MasterAddressTracker tracker = Mockito.mock(MasterAddressTracker.class); Mockito.doReturn(tracker).when(master).getMasterAddressTracker(); Mockito.doReturn(FAKE_HOST).when(tracker).getMasterAddress(); MetricsRegionServer rms = Mockito.mock(MetricsRegionServer.class); Mockito.doReturn(new MetricsRegionServerWrapperStub()).when(rms).getRegionServerWrapper(); Mockito.doReturn(rms).when(master).getMetrics(); // Mock admin admin = Mockito.mock(Admin.class); } private void setupMockTables() throws IOException { List<TableDescriptor> tables = Arrays.asList(TableDescriptorBuilder.newBuilder(TableName.valueOf("foo")).build(), TableDescriptorBuilder.newBuilder(TableName.valueOf("bar")).build()); Mockito.doReturn(tables).when(admin).listTableDescriptors(); } @Test public void testStatusTemplateNoTables() throws IOException { new MasterStatusTmpl().render(new StringWriter(), master); } @Test public void testStatusTemplateMetaAvailable() throws IOException { setupMockTables(); new MasterStatusTmpl() .setMetaLocation(ServerName.valueOf("metaserver,123,12345")) .render(new StringWriter(), master); } @Test public void testStatusTemplateWithServers() throws IOException { setupMockTables(); List<ServerName> servers = Lists.newArrayList( ServerName.valueOf("rootserver,123,12345"), ServerName.valueOf("metaserver,123,12345")); Set<ServerName> deadServers = new HashSet<>( Lists.newArrayList( ServerName.valueOf("badserver,123,12345"), ServerName.valueOf("uglyserver,123,12345")) ); new MasterStatusTmpl() .setMetaLocation(ServerName.valueOf("metaserver,123,12345")) .setServers(servers) .setDeadServers(deadServers) .render(new StringWriter(), master); } }