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

junit test class can also has parent

Browse Source
This commit is contained in:
金戟 2021-04-03 22:10:18 +08:00
parent b5c0afb3da
commit c9c8ba6c24
1 changed files with 14 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
testable-agent/src/main/java/com/alibaba/testable/agent/handler/OmniClassHandler.java
View File

@ -31,37 +31,33 @@ public class OmniClassHandler extends BaseClassHandler {
@Override
protected void transform(ClassNode cn) {
if ((cn.access & ACC_INTERFACE) != 0 || cn.superName == null) {
if (isInterface(cn) || isJunitTestClass(cn)) {
return;
}
MethodNode constructor = new MethodNode(ACC_PUBLIC, CONSTRUCTOR,
METHOD_START + ClassUtil.toByteCodeClassName(NULL_TYPE) + VOID_METHOD_END, null, null);
LabelNode start = new LabelNode(new Label());
LabelNode end = new LabelNode(new Label());
if (cn.superName.equals(CLASS_OBJECT)) {
// JUnit require test class contains only one constructor
if (isJUnitTestClass(cn)) {
return;
}
MethodNode constructor = new MethodNode(ACC_PUBLIC, CONSTRUCTOR,
METHOD_START + ClassUtil.toByteCodeClassName(NULL_TYPE) + VOID_METHOD_END, null, null);
LabelNode start = new LabelNode(new Label());
LabelNode end = new LabelNode(new Label());
constructor.instructions = invokeSuperWithoutParameter(start, end);
constructor.localVariables = createLocalVariables(cn, start, end);
constructor.maxStack = 1;
constructor.maxLocals = 2;
cn.methods.add(constructor);
} else {
MethodNode constructor = new MethodNode(ACC_PUBLIC, CONSTRUCTOR,
METHOD_START + ClassUtil.toByteCodeClassName(NULL_TYPE) + VOID_METHOD_END, null, null);
LabelNode start = new LabelNode(new Label());
LabelNode end = new LabelNode(new Label());
constructor.instructions = invokeSuperWithTestableNullParameter(cn, start, end);
constructor.localVariables = createLocalVariables(cn, start, end);
constructor.maxStack = 3;
constructor.maxLocals = 2;
cn.methods.add(constructor);
}
constructor.maxLocals = 2;
cn.methods.add(constructor);
}
private boolean isJUnitTestClass(ClassNode cn) {
private boolean isInterface(ClassNode cn) {
// is interface or the object class
return (cn.access & ACC_INTERFACE) != 0 || cn.superName == null;
}
private boolean isJunitTestClass(ClassNode cn) {
// junit require test class contains only one constructor
for (MethodNode mn : cn.methods) {
if (mn.visibleAnnotations == null) {
continue;