github-repos/testable-mock
1
0
Fork 0
mirror of https://github.com/alibaba/testable-mock.git synced 2026-09-04 21:52:56 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: transform mock class according to MockContainer annotation

Browse Source
This commit is contained in:
金戟
2022-11-20 20:11:46 +08:00
parent d912e53ebc
commit db91a0ea34
3 changed files with 72 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
testable-agent/src/main/java/com/alibaba/testable/agent/constant/ConstPool.java
View File

@@ -9,10 +9,12 @@ public class ConstPool {
public static final String FIELD_TARGET_CLASS = "targetClass";
public static final String FIELD_SCOPE = "scope";
public static final String FIELD_VALUE = "value";
public static final String FIELD_INHERITS = "inherits";
public static final String PROPERTY_USER_DIR = "user.dir";
public static final String PROPERTY_TEMP_DIR = "java.io.tmpdir";
public static final String MOCK_CONTAINER = "com.alibaba.testable.core.annotation.MockContainer";
public static final String MOCK_WITH = "com.alibaba.testable.core.annotation.MockWith";
public static final String DUMP_TO = "com.alibaba.testable.core.annotation.DumpTo";
public static final String MOCK_DIAGNOSE = "com.alibaba.testable.core.annotation.MockDiagnose";

69
testable-agent/src/main/java/com/alibaba/testable/agent/handler/MockClassHandler.java
View File

@@ -10,12 +10,15 @@ import org.objectweb.asm.Label;
import org.objectweb.asm.Type;
import org.objectweb.asm.tree.*;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import static com.alibaba.testable.agent.constant.ByteCodeConst.TYPE_ARRAY;
import static com.alibaba.testable.agent.constant.ByteCodeConst.TYPE_CLASS;
import static com.alibaba.testable.agent.constant.ConstPool.CLASS_OBJECT;
import static com.alibaba.testable.agent.constant.ConstPool.*;
import static com.alibaba.testable.core.constant.ConstPool.CONSTRUCTOR;
import static com.alibaba.testable.core.util.CollectionUtil.listOf;
/**
* @author flin
@@ -33,6 +36,7 @@ public class MockClassHandler extends BaseClassWithContextHandler {
private static final String SIGNATURE_IS_ASSOCIATED = "()Z";
private static final String SELF_REF = "__self";
private static final String TESTABLE_REF = "__testable";
private static final String INHERITED_REF = "__inherited_m";
private final String mockClassName;
@@ -63,9 +67,72 @@ public class MockClassHandler extends BaseClassWithContextHandler {
handleTestableUtil(mn);
}
}
// should inject inherited at last to avoid above transfers
injectInheritedMockMethods(cn);
LogUtil.diagnose(" Found %d mock methods", mockMethodCount);
}
/**
* add member field of inherited mock class by @MockContainer annotation
*/
private void injectInheritedMockMethods(ClassNode cn) {
List<Type> inheritedTypes = new ArrayList<Type>();
if (cn.visibleAnnotations != null) {
for (AnnotationNode an : cn.visibleAnnotations) {
if ((ClassUtil.toByteCodeClassName(ConstPool.MOCK_CONTAINER)).equals(an.desc)) {
inheritedTypes.addAll(AnnotationUtil.getAnnotationParameter(an, FIELD_INHERITS,
Collections.<Type>emptyList(), List.class));
}
}
}
for (int i = 0; i < inheritedTypes.size(); i++) {
String className = inheritedTypes.get(i).getClassName();
String fieldName = INHERITED_REF + i;
cn.fields.add(new FieldNode(ACC_PRIVATE | ACC_FINAL, fieldName,
ClassUtil.toByteCodeClassName(className), null, null));
ClassNode inheritedClassNode = ClassUtil.getClassNode(className);
if (inheritedClassNode == null) {
throw new IllegalArgumentException("Failed to load class " + className);
}
for (MethodNode mn : inheritedClassNode.methods) {
if (mn.visibleAnnotations == null) {
continue;
}
// TODO: should check whether method with same descriptor already exists to avoid conflict
for (AnnotationNode an : mn.visibleAnnotations) {
if ((ClassUtil.toByteCodeClassName(ConstPool.MOCK_INVOKE)).equals(an.desc)) {
Type targetClass = AnnotationUtil.getAnnotationParameter(an, FIELD_TARGET_CLASS, null, Type.class);
String targetMethod = AnnotationUtil.getAnnotationParameter(an, FIELD_TARGET_METHOD, null, String.class);
String desc = (targetClass == null) ? mn.desc :
MethodUtil.addParameterAtBegin(mn.desc, ClassUtil.toByteCodeClassName(targetClass.getClassName()));
String name = (targetMethod == null) ? mn.name : targetMethod;
MethodNode mockMethod = new MethodNode(ACC_PUBLIC, name, desc, null, null);
List<Byte> parameters = MethodUtil.getParameterTypes(mn.desc);
int maxStack = 2;
InsnList il = new InsnList();
il.add(new VarInsnNode(ALOAD, 0));
il.add(new FieldInsnNode(GETFIELD, ClassUtil.toSlashSeparatedName(cn.name), fieldName, ClassUtil.toByteCodeClassName(className)));
il.add(new VarInsnNode(ALOAD, 1));
for (int pi = 2; pi < parameters.size() + 2; pi++) {
ImmutablePair<Integer, Integer> codeAndStack = BytecodeUtil.getLoadParameterByteCode(parameters.get(pi - 2));
il.add(new VarInsnNode(codeAndStack.left, pi));
maxStack += codeAndStack.right;
}
il.add(new MethodInsnNode(INVOKEVIRTUAL, ClassUtil.toSlashSeparatedName(className), mn.name, desc, false));
il.add(new InsnNode(ARETURN));
mockMethod.instructions = il;
mockMethod.maxStack = maxStack;
mockMethod.maxLocals = 2 + parameters.size();
mockMethod.visibleAnnotations = listOf(new AnnotationNode(ClassUtil.toByteCodeClassName(MOCK_INVOKE)));
cn.methods.add(mockMethod);
} else if ((ClassUtil.toByteCodeClassName(ConstPool.MOCK_NEW)).equals(an.desc)) {
// TODO: should also support MockNew annotation
}
}
}
}
}
/**
* add method to fetch singleton instance of this mock class
*/

3
testable-core/src/main/java/com/alibaba/testable/core/annotation/MockContainer.java
View File

@@ -1,5 +1,6 @@
package com.alibaba.testable.core.annotation;
import javax.lang.model.type.NullType;
import java.lang.annotation.*;
/**
@@ -16,6 +17,6 @@ public @interface MockContainer {
* specify the classes to inherit methods from
* @return list of class
*/
Class<?>[] inherits();
Class<?>[] inherits() default NullType.class;
}