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.MockRegionServerServices; 028import org.apache.hadoop.hbase.regionserver.RegionServerCoprocessorHost; 029import org.apache.hadoop.hbase.regionserver.RegionServerServices; 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 RegionServerCoprocessor. 043 */ 044@Category({CoprocessorTests.class, SmallTests.class}) 045public class TestCoreRegionServerCoprocessor { 046 047 @ClassRule 048 public static final HBaseClassTestRule CLASS_RULE = 049 HBaseClassTestRule.forClass(TestCoreRegionServerCoprocessor.class); 050 051 @Rule public TestName name = new TestName(); 052 private static final HBaseTestingUtility HTU = HBaseTestingUtility.createLocalHTU(); 053 private RegionServerServices rss; 054 private RegionServerCoprocessorHost rsch; 055 056 @Before 057 public void before() throws IOException { 058 String methodName = this.name.getMethodName(); 059 this.rss = new MockRegionServerServices(HTU.getConfiguration()); 060 this.rsch = new RegionServerCoprocessorHost(this.rss, HTU.getConfiguration()); 061 } 062 063 @After 064 public void after() throws IOException { 065 this.rsch.preStop("Stopping", null); 066 } 067 068 /** 069 * No annotation with CoreCoprocessor. This should make it so I can NOT get at instance of a 070 * RegionServerServices instance after some gymnastics. 071 */ 072 public static class NotCoreRegionServerCoprocessor implements RegionServerCoprocessor { 073 public NotCoreRegionServerCoprocessor() {} 074 } 075 076 /** 077 * Annotate with CoreCoprocessor. This should make it so I can get at instance of a 078 * RegionServerServices instance after some gymnastics. 079 */ 080 @CoreCoprocessor 081 public static class CoreRegionServerCoprocessor implements RegionServerCoprocessor { 082 public CoreRegionServerCoprocessor() {} 083 } 084 085 /** 086 * Assert that when a Coprocessor is annotated with CoreCoprocessor, then it is possible to 087 * access a RegionServerServices instance. Assert the opposite too. 088 * Do it to RegionServerCoprocessors. 089 * @throws IOException 090 */ 091 @Test 092 public void testCoreRegionCoprocessor() throws IOException { 093 RegionServerCoprocessorEnvironment env = 094 rsch.load(null, NotCoreRegionServerCoprocessor.class.getName(), 0, HTU.getConfiguration()); 095 assertFalse(env instanceof HasRegionServerServices); 096 env = rsch.load(null, CoreRegionServerCoprocessor.class.getName(), 1, HTU.getConfiguration()); 097 assertTrue(env instanceof HasRegionServerServices); 098 assertEquals(this.rss, ((HasRegionServerServices)env).getRegionServerServices()); 099 } 100}