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.coprocessor; 019 020import static org.junit.Assert.assertEquals; 021import static org.junit.Assert.assertFalse; 022import static org.junit.Assert.assertTrue; 023 024import java.io.IOException; 025import org.apache.hadoop.hbase.HBaseClassTestRule; 026import org.apache.hadoop.hbase.HBaseTestingUtility; 027import org.apache.hadoop.hbase.master.MasterCoprocessorHost; 028import org.apache.hadoop.hbase.master.MasterServices; 029import org.apache.hadoop.hbase.master.assignment.MockMasterServices; 030import org.apache.hadoop.hbase.testclassification.CoprocessorTests; 031import org.apache.hadoop.hbase.testclassification.SmallTests; 032import org.junit.After; 033import org.junit.Before; 034import org.junit.ClassRule; 035import org.junit.Rule; 036import org.junit.Test; 037import org.junit.experimental.categories.Category; 038import org.junit.rules.TestName; 039 040/** 041 * Test CoreCoprocessor Annotation works giving access to facility not usually available. 042 * Test MasterCoprocessor. 043 */ 044@Category({CoprocessorTests.class, SmallTests.class}) 045public class TestCoreMasterCoprocessor { 046 047 @ClassRule 048 public static final HBaseClassTestRule CLASS_RULE = 049 HBaseClassTestRule.forClass(TestCoreMasterCoprocessor.class); 050 051 @Rule public TestName name = new TestName(); 052 private static final HBaseTestingUtility HTU = HBaseTestingUtility.createLocalHTU(); 053 private MasterServices ms; 054 private MasterCoprocessorHost mch; 055 056 @Before 057 public void before() throws IOException { 058 String methodName = this.name.getMethodName(); 059 this.ms = new MockMasterServices(HTU.getConfiguration(), null); 060 this.mch = new MasterCoprocessorHost(this.ms, HTU.getConfiguration()); 061 this.mch.preMasterInitialization(); 062 } 063 064 @After 065 public void after() throws IOException { 066 this.mch.preStopMaster(); 067 } 068 069 /** 070 * No annotation with CoreCoprocessor. This should make it so I can NOT get at instance of a 071 * MasterServices instance after some gymnastics. 072 */ 073 public static class NotCoreMasterCoprocessor implements MasterCoprocessor { 074 public NotCoreMasterCoprocessor() {} 075 } 076 077 /** 078 * Annotate with CoreCoprocessor. This should make it so I can get at instance of a 079 * MasterServices instance after some gymnastics. 080 */ 081 @CoreCoprocessor 082 public static class CoreMasterCoprocessor implements MasterCoprocessor { 083 public CoreMasterCoprocessor() {} 084 } 085 086 /** 087 * Assert that when a Coprocessor is annotated with CoreCoprocessor, then it is possible to 088 * access a MasterServices instance. Assert the opposite too. 089 * Do it to MasterCoprocessors. 090 * @throws IOException 091 */ 092 @Test 093 public void testCoreRegionCoprocessor() throws IOException { 094 MasterCoprocessorEnvironment env = 095 this.mch.load(null, NotCoreMasterCoprocessor.class.getName(), 0, HTU.getConfiguration()); 096 assertFalse(env instanceof HasMasterServices); 097 env = this.mch.load(null, CoreMasterCoprocessor.class.getName(), 1, HTU.getConfiguration()); 098 assertTrue(env instanceof HasMasterServices); 099 assertEquals(this.ms, ((HasMasterServices)env).getMasterServices()); 100 } 101}