github-repos/testable-mock
1
0
Fork 0
mirror of https://github.com/alibaba/testable-mock.git synced 2025-01-10 20:30:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

dynamically update testable inject method access

Browse Source
This commit is contained in:
金戟 2020-10-12 19:20:03 +08:00
parent efccedc996
commit ce0e7e4d3a
3 changed files with 18 additions and 33 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
agent/src/main/java/com/alibaba/testable/agent/handler/TestClassHandler.java
View File

@ -27,14 +27,27 @@ public class TestClassHandler extends BaseClassHandler {
for (AnnotationNode n : mn.visibleAnnotations) {
visibleAnnotationNames.add(n.desc);
}
if (!visibleAnnotationNames.contains(ConstPool.TESTABLE_INJECT) && couldBeTestMethod(mn)) {
InsnList il = new InsnList();
il.add(new VarInsnNode(ALOAD, 0));
il.add(new FieldInsnNode(PUTSTATIC, cn.name, ConstPool.TESTABLE_INJECT_REF, ClassUtil.toByteCodeClassName(cn.name)));
mn.instructions.insertBefore(mn.instructions.get(0), il);
if (visibleAnnotationNames.contains(ClassUtil.toByteCodeClassName(ConstPool.TESTABLE_INJECT))) {
mn.access &= ~ACC_PRIVATE;
mn.access &= ~ACC_PROTECTED;
mn.access |= ACC_PUBLIC;
} else if (couldBeTestMethod(mn)) {
injectTestableRef(cn, mn);
}
}
private void injectTestableRef(ClassNode cn, MethodNode mn) {
InsnList il = new InsnList();
il.add(new VarInsnNode(ALOAD, 0));
il.add(new FieldInsnNode(PUTSTATIC, cn.name, ConstPool.TESTABLE_INJECT_REF,
ClassUtil.toByteCodeClassName(cn.name)));
mn.instructions.insertBefore(mn.instructions.get(0), il);
}
/**
* Different unit test framework may have different @Test annotation
* but they should always NOT private, protected or static
*/
private boolean couldBeTestMethod(MethodNode mn) {
return (mn.access & (ACC_PRIVATE | ACC_PROTECTED | ACC_STATIC)) == 0 ;
}

13
core/src/main/java/com/alibaba/testable/core/translator/EnableTestableTranslator.java
View File

@ -77,19 +77,6 @@ public class EnableTestableTranslator extends BaseTranslator {
super.visitExec(jcExpressionStatement);
}
/**
* Search for TestableInject annotations
*/
@Override
public void visitMethodDef(JCMethodDecl jcMethodDecl) {
for (JCAnnotation a : jcMethodDecl.mods.annotations) {
if (a.type != null && ConstPool.ANNOTATION_TESTABLE_INJECT.equals(a.type.tsym.toString())) {
TypeUtil.toPublicFlags(jcMethodDecl.getModifiers());
}
}
super.visitMethodDef(jcMethodDecl);
}
/**
* Generate test setup method to initialize n.e.pool
*/

15
core/src/main/java/com/alibaba/testable/core/util/TypeUtil.java
View File

@ -1,27 +1,12 @@
package com.alibaba.testable.core.util;
import com.sun.tools.javac.tree.JCTree;
import java.lang.reflect.Constructor;
import java.lang.reflect.Method;
import java.lang.reflect.Modifier;
import java.util.ArrayList;
import java.util.List;
/**
* @author flin
*/
public class TypeUtil {
/**
* Convert modifier to public
*/
public static void toPublicFlags(JCTree.JCModifiers modifiers) {
modifiers.flags &= ~Modifier.PRIVATE;
modifiers.flags &= ~Modifier.PROTECTED;
modifiers.flags |= Modifier.PUBLIC;
}
/**
* get classes of parameter objects
*/