Update MethodUtilsTest.java

Author: mercyblitz

microsphere-java-core/src/test/java/io/microsphere/reflect/MethodUtilsTest.java

@@ -17,6 +17,8 @@
 package io.microsphere.reflect;
 
 import io.microsphere.lang.Prioritized;
+import org.junit.jupiter.api.AfterAll;
+import org.junit.jupiter.api.BeforeEach;
 import org.junit.jupiter.api.Test;
 
 import java.lang.reflect.Method;
@@ -28,6 +30,7 @@
 
 import static io.microsphere.AbstractTestCase.JACOCO_AGENT_INSTRUCTED;
 import static io.microsphere.reflect.MemberUtils.PUBLIC_MEMBER_PREDICATE;
+import static io.microsphere.reflect.MethodUtils.BANNED_METHODS_PROPERTY_NAME;
 import static io.microsphere.reflect.MethodUtils.FINAL_METHOD_PREDICATE;
 import static io.microsphere.reflect.MethodUtils.NON_PRIVATE_METHOD_PREDICATE;
 import static io.microsphere.reflect.MethodUtils.NON_STATIC_METHOD_PREDICATE;
@@ -37,6 +40,7 @@
 import static io.microsphere.reflect.MethodUtils.STATIC_METHOD_PREDICATE;
 import static io.microsphere.reflect.MethodUtils.buildKey;
 import static io.microsphere.reflect.MethodUtils.buildSignature;
+import static io.microsphere.reflect.MethodUtils.clearBannedMethods;
 import static io.microsphere.reflect.MethodUtils.excludedDeclaredClass;
 import static io.microsphere.reflect.MethodUtils.findAllDeclaredMethods;
 import static io.microsphere.reflect.MethodUtils.findAllMethods;
@@ -50,13 +54,15 @@
 import static io.microsphere.reflect.MethodUtils.getDeclaredMethods;
 import static io.microsphere.reflect.MethodUtils.getMethods;
 import static io.microsphere.reflect.MethodUtils.getSignature;
+import static io.microsphere.reflect.MethodUtils.initBannedMethods;
 import static io.microsphere.reflect.MethodUtils.invokeMethod;
 import static io.microsphere.reflect.MethodUtils.invokeStaticMethod;
 import static io.microsphere.reflect.MethodUtils.isObjectMethod;
 import static io.microsphere.reflect.MethodUtils.overrides;
 import static io.microsphere.util.ArrayUtils.EMPTY_CLASS_ARRAY;
 import static io.microsphere.util.ClassUtils.PRIMITIVE_TYPES;
 import static java.lang.Integer.valueOf;
+import static java.lang.System.setProperty;
 import static java.util.Collections.emptyList;
 import static org.junit.jupiter.api.Assertions.assertEquals;
 import static org.junit.jupiter.api.Assertions.assertFalse;
@@ -80,6 +86,17 @@ class MethodUtilsTest {
         JACOCO_ADDED_METHOD_COUNT = JACOCO_AGENT_INSTRUCTED ? 1 : 0;
     }
 
+    @BeforeEach
+    void setUp() {
+        afterAll();
+    }
+
+    @AfterAll
+    static void afterAll() {
+        System.getProperties().remove(BANNED_METHODS_PROPERTY_NAME);
+        clearBannedMethods();
+    }
+
     @Test
     void testSTATIC_METHOD_PREDICATE() {
         assertTrue(STATIC_METHOD_PREDICATE.test(findMethod(ReflectionTest.class, "staticMethod")));
@@ -107,6 +124,23 @@ void testNON_PRIVATE_METHOD_PREDICATE() {
         assertTrue(NON_PRIVATE_METHOD_PREDICATE.test(findMethod(ReflectionTest.class, "packagePrivateMethod", String.class)));
     }
 
+    @Test
+    void testInitBannedMethods() {
+        String signature = buildSignature(String.class, "substring", int.class);
+        String signature2 = buildSignature(String.class, "substring", int.class, int.class);
+        setProperty(BANNED_METHODS_PROPERTY_NAME, signature + " | " + signature2);
+        initBannedMethods();
+        assertNull(findMethod(String.class, "substring", int.class));
+        assertNull(findMethod(String.class, "substring", int.class, int.class));
+    }
+
+    @Test
+    void testInitBannedMethodsWithoutProperty() {
+        initBannedMethods();
+        assertNotNull(findMethod(String.class, "substring", int.class));
+        assertNotNull(findMethod(String.class, "substring", int.class, int.class));
+    }
+
     @Test
     void testExcludedDeclaredClass() {
         Predicate<? super Method> predicate = excludedDeclaredClass(Object.class);