fix testcase. #1184

This commit is contained in:
hengyunabc
2020-05-24 02:12:27 +08:00
parent f66e1c9d59
commit c20e9f9170
2 changed files with 4 additions and 8 deletions
@@ -197,7 +197,7 @@ public class Enhancer implements ClassFileTransformer {
// 这里要再次过滤一次,为啥?因为在transform的过程中,有可能还会再诞生新的类
// 所以需要将之前需要转换的类集合传递下来,再次进行判断
if (!matchingClasses.contains(classBeingRedefined)) {
if (matchingClasses != null && !matchingClasses.contains(classBeingRedefined)) {
return null;
}
@@ -41,14 +41,10 @@ public class EnhancerTest {
AdviceListener listener = Mockito.mock(AdviceListener.class);
EnhancerAffect affect = new EnhancerAffect();
EqualsMatcher<String> methodNameMatcher = new EqualsMatcher<String>("print");
EqualsMatcher<String> classNameMatcher = new EqualsMatcher<String>(MathGame.class.getName());
Set<Class<?>> matchingClasses = new HashSet<Class<?>>();
matchingClasses.add(MathGame.class);
EqualsMatcher<String> matcher = new EqualsMatcher<String>("print");
Enhancer enhancer = new Enhancer(listener, true, false, matchingClasses, matcher, affect);
Enhancer enhancer = new Enhancer(listener, true, false, classNameMatcher, methodNameMatcher);
ClassLoader inClassLoader = MathGame.class.getClassLoader();
String className = MathGame.class.getName();