Don't use reflection where not necessary in TestClientTokenUtil

apache · Sep 18, 2019 · facfe8a · facfe8a
1 parent d1979be
commit facfe8a
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/hbase-client/src/test/java/org/apache/hadoop/hbase/security/token/TestClientTokenUtil.java b/hbase-client/src/test/java/org/apache/hadoop/hbase/security/token/TestClientTokenUtil.java
@@ -23,7 +23,6 @@
 
 import java.io.IOException;
 import java.lang.reflect.Field;
-import java.lang.reflect.InvocationTargetException;
 import java.net.URL;
 import java.net.URLClassLoader;
 import java.util.concurrent.CompletableFuture;
@@ -73,10 +72,9 @@ public void testObtainToken() throws Exception {
     shouldInjectFault.set(null, injected);
 
     try {
-      clientTokenUtil.getMethod("obtainToken", Connection.class)
-          .invoke(null, new Object[] { null });
+      ClientTokenUtil.obtainToken((Connection)null);
       fail("Should have injected exception.");
-    } catch (InvocationTargetException e) {
+    } catch (IOException e) {
       Throwable t = e;
       boolean serviceExceptionFound = false;
       while ((t = t.getCause()) != null) {
@@ -90,8 +88,7 @@ public void testObtainToken() throws Exception {
       }
     }
 
-    CompletableFuture<?> future = (CompletableFuture<?>) clientTokenUtil
-      .getMethod("obtainToken", AsyncConnection.class).invoke(null, new Object[] { null });
+    CompletableFuture<?> future = ClientTokenUtil.obtainToken((AsyncConnection)null);
     try {
       future.get();
       fail("Should have injected exception.");