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

make testable inject method public

Browse Source
This commit is contained in:
金戟 2020-07-29 11:57:50 +08:00
parent bd203dc409
commit 47aa73bd13
2 changed files with 17 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -5,6 +5,7 @@ import com.alibaba.testable.core.generator.TestSetupMethodGenerator;
import com.alibaba.testable.core.generator.TestableRefFieldGenerator;
import com.alibaba.testable.core.model.TestableContext;
import com.alibaba.testable.core.constant.ConstPool;
import com.alibaba.testable.core.util.TypeUtil;
import com.sun.tools.javac.tree.JCTree;
import com.sun.tools.javac.tree.JCTree.*;
import com.sun.tools.javac.util.ListBuffer;
@ -24,8 +25,8 @@ import java.util.Arrays;
public class EnableTestableTranslator extends BaseTranslator {
private final TestableContext cx;
private String testClassName;
private String sourceClassName;
private final String testClassName;
private final String sourceClassName;
private final ListBuffer<Name> sourceClassIns = new ListBuffer<>();
private final ListBuffer<String> privateOrFinalFields = new ListBuffer<>();
private final ListBuffer<String> privateMethods = new ListBuffer<>();
@ -85,12 +86,13 @@ public class EnableTestableTranslator extends BaseTranslator {
}
/**
* Search for TestableInject and TestSetup annotations
* 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());
ListBuffer<JCExpression> args = new ListBuffer<>();
for (JCVariableDecl p : jcMethodDecl.params) {
args.add(cx.treeMaker.Select(p.vartype, cx.names.fromString(ConstPool.CLASS_OF_TYPE)));

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

@ -1,7 +1,10 @@
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;
@ -10,6 +13,15 @@ import java.util.List;
*/
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;
}
/**
* Information of substitution method
*/