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

support mock class inheriting

Browse Source
This commit is contained in:
金戟 2021-02-15 14:08:56 +08:00
parent 34f1dc8f73
commit 189ef43565
1 changed files with 13 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
testable-agent/src/main/java/com/alibaba/testable/agent/transformer/TestableClassTransformer.java
View File

@ -44,8 +44,8 @@ public class TestableClassTransformer implements ClassFileTransformer {
private static final String FIELD_TREAT_AS = "treatAs";
private static final String FIELD_DIAGNOSE = "diagnose";
private static final String COMMA = ",";
private static final String CLASS_NAME_MOCK = "Mock";
private static final String CLASS_OBJECT = "java/lang/Object";
/**
* Just avoid spend time to scan those surely non-user classes
@ -172,13 +172,24 @@ public class TestableClassTransformer implements ClassFileTransformer {
if (cn == null) {
return new ArrayList<MethodInfo>();
}
for (MethodNode mn : cn.methods) {
for (MethodNode mn : getAllMethods(cn)) {
checkMethodAnnotation(cn, methodInfos, mn);
}
LogUtil.diagnose(" Found %d mock methods", methodInfos.size());
return methodInfos;
}
private List<MethodNode> getAllMethods(ClassNode cn) {
List<MethodNode> mns = new ArrayList<MethodNode>(cn.methods);
if (cn.superName != null && !cn.superName.equals(CLASS_OBJECT)) {
ClassNode scn = getClassNode(cn.superName);
if (scn != null) {
mns.addAll(getAllMethods(scn));
}
}
return mns;
}
private void checkMethodAnnotation(ClassNode cn, List<MethodInfo> methodInfos, MethodNode mn) {
if (mn.visibleAnnotations == null) {
return;