1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18 package org.apache.hadoop.hbase;
19
20 import com.google.protobuf.Message;
21 import org.apache.hadoop.conf.Configuration;
22 import org.apache.hadoop.hbase.protobuf.generated.RPCProtos;
23 import org.apache.hadoop.hbase.regionserver.AnnotationReadingPriorityFunction;
24 import org.apache.hadoop.hbase.security.User;
25
26 import static org.junit.Assert.assertEquals;
27
28 public class QosTestHelper {
29 protected void checkMethod(Configuration conf, final String methodName, final int expected,
30 final AnnotationReadingPriorityFunction qosf) {
31 checkMethod(conf, methodName, expected, qosf, null);
32 }
33
34 protected void checkMethod(Configuration conf, final String methodName, final int expected,
35 final AnnotationReadingPriorityFunction qosf, final Message param) {
36 RPCProtos.RequestHeader.Builder builder = RPCProtos.RequestHeader.newBuilder();
37 builder.setMethodName(methodName);
38 assertEquals(methodName, expected, qosf.getPriority(builder.build(), param,
39 User.createUserForTesting(conf, "someuser", new String[]{"somegroup"})));
40 }
41 }