View Javadoc

1   /**
2   * Licensed to the Apache Software Foundation (ASF) under one
3   * or more contributor license agreements.  See the NOTICE file
4   * distributed with this work for additional information
5   * regarding copyright ownership.  The ASF licenses this file
6   * to you under the Apache License, Version 2.0 (the
7   * "License"); you may not use this file except in compliance
8   * with the License.  You may obtain a copy of the License at
9   *
10  *     http://www.apache.org/licenses/LICENSE-2.0
11  *
12  * Unless required by applicable law or agreed to in writing, software
13  * distributed under the License is distributed on an "AS IS" BASIS,
14  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15  * See the License for the specific language governing permissions and
16  * limitations under the License.
17  */
18  
19  package org.apache.hadoop.hbase.regionserver;
20  
21  import static org.mockito.Mockito.when;
22  
23  import org.apache.hadoop.conf.Configuration;
24  import org.apache.hadoop.hbase.*;
25  import org.apache.hadoop.hbase.testclassification.SmallTests;
26  import org.apache.hadoop.hbase.protobuf.generated.ClientProtos.MultiRequest;
27  import org.junit.Before;
28  import org.junit.Test;
29  import org.junit.experimental.categories.Category;
30  import org.mockito.Mockito;
31  
32  /**
33   * Basic test that qos function is sort of working; i.e. a change in method naming style
34   * over in pb doesn't break it.
35   */
36  @Category({ SmallTests.class})
37  public class TestQosFunction extends QosTestHelper {
38    private Configuration conf;
39    private RSRpcServices rpcServices;
40    private AnnotationReadingPriorityFunction qosFunction;
41  
42    @Before
43    public void setUp() {
44      conf = HBaseConfiguration.create();
45      rpcServices = Mockito.mock(RSRpcServices.class);
46      when(rpcServices.getConfiguration()).thenReturn(conf);
47      qosFunction = new AnnotationReadingPriorityFunction(rpcServices, RSRpcServices.class);
48    }
49  
50    @Test
51    public void testPriority() {
52      // Set method name in pb style with the method name capitalized.
53      checkMethod(conf, "ReplicateWALEntry", HConstants.REPLICATION_QOS, qosFunction);
54      // Set method name in pb style with the method name capitalized.
55      checkMethod(conf, "OpenRegion", HConstants.ADMIN_QOS, qosFunction);
56      // Check multi works.
57      checkMethod(conf, "Multi", HConstants.NORMAL_QOS, qosFunction,
58          MultiRequest.getDefaultInstance());
59  
60    }
61  
62  
63    @Test
64    public void testAnnotations() {
65      checkMethod(conf, "CloseRegion", HConstants.ADMIN_QOS, qosFunction);
66      checkMethod(conf, "CompactRegion", HConstants.ADMIN_QOS, qosFunction);
67      checkMethod(conf, "FlushRegion", HConstants.ADMIN_QOS, qosFunction);
68    }
69  }