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

fix crash when meet primary parameter type

Browse Source
This commit is contained in:
金戟 2020-05-21 10:37:13 +08:00
parent fecaafca88
commit eeb3117e7b
1 changed files with 29 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
src/main/java/com/alibaba/testable/translator/TestableClassTestRoleTranslator.java
View File

@ -55,15 +55,22 @@ public class TestableClassTestRoleTranslator extends TreeTranslator {
}
}
/**
* Demo d = new Demo() -> DemoTestable d = new Demo()
*/
@Override
public void visitVarDef(JCVariableDecl jcVariableDecl) {
super.visitVarDef(jcVariableDecl);
if (((JCIdent)jcVariableDecl.vartype).name.toString().equals(sourceClassName)) {
if (jcVariableDecl.vartype.getClass().equals(JCIdent.class) &&
((JCIdent)jcVariableDecl.vartype).name.toString().equals(sourceClassName)) {
jcVariableDecl.vartype = getTestableClassIdent(jcVariableDecl.vartype);
sourceClassIns.add(jcVariableDecl.name);
}
}
/**
* Demo d = new Demo() -> Demo d = new DemoTestable()
*/
@Override
public void visitNewClass(JCNewClass jcNewClass) {
super.visitNewClass(jcNewClass);
@ -72,16 +79,9 @@ public class TestableClassTestRoleTranslator extends TreeTranslator {
}
}
private String getSimpleClassName(JCNewClass jcNewClass) {
if (jcNewClass.clazz.getClass().equals(JCIdent.class)) {
return ((JCIdent)jcNewClass.clazz).name.toString();
} else if (jcNewClass.clazz.getClass().equals(JCFieldAccess.class)) {
return ((JCFieldAccess)jcNewClass.clazz).name.toString();
} else {
return "";
}
}
/**
* d.privateField = val -> d.privateFieldTestableSet(val)
*/
@Override
public void visitExec(JCExpressionStatement jcExpressionStatement) {
if (jcExpressionStatement.expr.getClass().equals(JCAssign.class) &&
@ -95,6 +95,9 @@ public class TestableClassTestRoleTranslator extends TreeTranslator {
super.visitExec(jcExpressionStatement);
}
/**
* Search for TestableInject and TestSetup annotations
*/
@Override
public void visitMethodDef(JCMethodDecl jcMethodDecl) {
for (JCAnnotation a : jcMethodDecl.mods.annotations) {
@ -122,6 +125,9 @@ public class TestableClassTestRoleTranslator extends TreeTranslator {
super.visitMethodDef(jcMethodDecl);
}
/**
* Generate test setup method to initialize n.e.pool
*/
@Override
public void visitClassDef(JCClassDecl jcClassDecl) {
super.visitClassDef(jcClassDecl);
@ -135,7 +141,7 @@ public class TestableClassTestRoleTranslator extends TreeTranslator {
}
/**
* For break point
* For setter break point
*/
@Override
public void visitAssign(JCAssign jcAssign) {
@ -143,13 +149,23 @@ public class TestableClassTestRoleTranslator extends TreeTranslator {
}
/**
* For break point
* For getter break point
*/
@Override
public void visitSelect(JCFieldAccess jcFieldAccess) {
super.visitSelect(jcFieldAccess);
}
private String getSimpleClassName(JCNewClass jcNewClass) {
if (jcNewClass.clazz.getClass().equals(JCIdent.class)) {
return ((JCIdent)jcNewClass.clazz).name.toString();
} else if (jcNewClass.clazz.getClass().equals(JCFieldAccess.class)) {
return ((JCFieldAccess)jcNewClass.clazz).name.toString();
} else {
return "";
}
}
private List<JCAnnotation> makeAnnotations(String fullAnnotationName) {
JCExpression setupAnnotation = nameToExpression(fullAnnotationName);
return List.of(cx.treeMaker.Annotation(setupAnnotation, List.<JCExpression>nil()));