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

travel all methods before handle testable injection

Browse Source
This commit is contained in:
金戟 2020-06-01 19:43:31 +08:00
parent 0410990a90
commit 8e95fefb58
3 changed files with 39 additions and 39 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
core/src/main/java/com/alibaba/testable/processor/EnableTestableInjectProcessor.java
View File

@ -3,6 +3,7 @@ package com.alibaba.testable.processor;
import com.alibaba.testable.annotation.EnableTestableInject; import com.alibaba.testable.annotation.EnableTestableInject;
import com.alibaba.testable.generator.StaticNewClassGenerator; import com.alibaba.testable.generator.StaticNewClassGenerator;
import com.alibaba.testable.translator.EnableTestableInjectTranslator; import com.alibaba.testable.translator.EnableTestableInjectTranslator;
import com.alibaba.testable.translator.MethodRecordTranslator;
import com.alibaba.testable.util.ConstPool; import com.alibaba.testable.util.ConstPool;
import com.sun.tools.javac.code.Symbol; import com.sun.tools.javac.code.Symbol;
import com.sun.tools.javac.tree.JCTree; import com.sun.tools.javac.tree.JCTree;
@ -72,8 +73,9 @@ public class EnableTestableInjectProcessor extends BaseProcessor {
private void processClassElement(Symbol.ClassSymbol clazz) { private void processClassElement(Symbol.ClassSymbol clazz) {
JCTree tree = cx.trees.getTree(clazz); JCTree tree = cx.trees.getTree(clazz);
EnableTestableInjectTranslator translator = new EnableTestableInjectTranslator(cx); MethodRecordTranslator methodRecordTranslator = new MethodRecordTranslator();
tree.accept(translator); tree.accept(methodRecordTranslator);
tree.accept(new EnableTestableInjectTranslator(cx, methodRecordTranslator.getMethods()));
} }
private void writeSourceFile(String fullQualityTypeName, String content) throws IOException { private void writeSourceFile(String fullQualityTypeName, String content) throws IOException {

40
core/src/main/java/com/alibaba/testable/translator/EnableTestableInjectTranslator.java
View File

@ -15,36 +15,11 @@ import com.sun.tools.javac.util.Name;
public class EnableTestableInjectTranslator extends BaseTranslator { public class EnableTestableInjectTranslator extends BaseTranslator {
private final TestableContext cx; private final TestableContext cx;
private List<JCMethodDecl> methods;
/** public EnableTestableInjectTranslator(TestableContext cx, List<JCMethodDecl> methods) {
* Methods to inject
*/
private List<JCMethodDecl> methods = List.nil();
/**
* Fields to wrap
*/
private List<JCVariableDecl> fields = List.nil();
public List<JCMethodDecl> getMethods() {
return methods;
}
public List<JCVariableDecl> getFields() {
return fields;
}
public EnableTestableInjectTranslator(TestableContext cx) {
this.cx = cx; this.cx = cx;
} this.methods = methods;
/**
* Record all methods
*/
@Override
public void visitMethodDef(JCMethodDecl jcMethodDecl) {
super.visitMethodDef(jcMethodDecl);
methods = methods.append(jcMethodDecl);
} }
/** /**
@ -68,15 +43,11 @@ public class EnableTestableInjectTranslator extends BaseTranslator {
} }
/** /**
* Record all private fields
* Demo d = new Demo() -> Demo d = n.e.w(Demo.class) * Demo d = new Demo() -> Demo d = n.e.w(Demo.class)
* Demo d = member() -> Demo d = n.e.f(this, "member") * Demo d = member() -> Demo d = n.e.f(this, "member")
*/ */
@Override @Override
public void visitVarDef(JCVariableDecl jcVariableDecl) { public void visitVarDef(JCVariableDecl jcVariableDecl) {
if (isStubbornField(jcVariableDecl.mods)) {
fields = fields.append(jcVariableDecl);
}
jcVariableDecl.init = checkAndExchange(jcVariableDecl.init); jcVariableDecl.init = checkAndExchange(jcVariableDecl.init);
super.visitVarDef(jcVariableDecl); super.visitVarDef(jcVariableDecl);
} }
@ -117,11 +88,6 @@ public class EnableTestableInjectTranslator extends BaseTranslator {
super.visitNewArray(jcNewArray); super.visitNewArray(jcNewArray);
} }
private boolean isStubbornField(JCModifiers mods) {
return mods.getFlags().contains(javax.lang.model.element.Modifier.PRIVATE) ||
mods.getFlags().contains(javax.lang.model.element.Modifier.FINAL);
}
@Override @Override
protected JCExpression checkAndExchange(JCExpression expr) { protected JCExpression checkAndExchange(JCExpression expr) {
if (isNewOperation(expr)) { if (isNewOperation(expr)) {

32
core/src/main/java/com/alibaba/testable/translator/MethodRecordTranslator.java Normal file
View File

@ -0,0 +1,32 @@
package com.alibaba.testable.translator;
import com.sun.tools.javac.tree.JCTree.JCMethodDecl;
import com.sun.tools.javac.tree.TreeTranslator;
import com.sun.tools.javac.util.List;
/**
* Travel AST
*
* @author flin
*/
public class MethodRecordTranslator extends TreeTranslator {
/**
* Member methods
*/
private List<JCMethodDecl> methods = List.nil();
public List<JCMethodDecl> getMethods() {
return methods;
}
/**
* Record all methods
*/
@Override
public void visitMethodDef(JCMethodDecl jcMethodDecl) {
super.visitMethodDef(jcMethodDecl);
methods = methods.append(jcMethodDecl);
}
}