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

fit mock target checking for kotlin

Browse Source
This commit is contained in:
金戟 2021-05-22 18:31:03 +08:00
parent 3786c6251e
commit 993c178adf
2 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
demo/kotlin-demo/src/test/resources/testable.properties
View File

@ -1 +1,2 @@
omni.constructor.enhance.enable = true
mock.target.checking.enable = true

10
testable-agent/src/main/java/com/alibaba/testable/agent/transformer/MockClassParser.java
View File

@ -9,6 +9,7 @@ import com.alibaba.testable.core.util.LogUtil;
import org.objectweb.asm.Type;
import org.objectweb.asm.tree.AnnotationNode;
import org.objectweb.asm.tree.ClassNode;
import org.objectweb.asm.tree.InnerClassNode;
import org.objectweb.asm.tree.MethodNode;
import java.util.ArrayList;
@ -16,6 +17,7 @@ import java.util.List;
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.KOTLIN_POSTFIX_COMPANION;
import static com.alibaba.testable.agent.util.ClassUtil.toJavaStyleClassName;
import static com.alibaba.testable.agent.util.MethodUtil.isStatic;
import static com.alibaba.testable.core.constant.ConstPool.CONSTRUCTOR;
@ -72,6 +74,14 @@ public class MockClassParser {
mns.addAll(getAllMethods(scn));
}
}
for (InnerClassNode innerClass : cn.innerClasses) {
if (innerClass.name.equals(cn.name + KOTLIN_POSTFIX_COMPANION)) {
ClassNode scn = ClassUtil.getClassNode(innerClass.name);
if (scn != null) {
mns.addAll(getAllMethods(scn));
}
}
}
return mns;
}