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

fit private access for gradle build

Browse Source
This commit is contained in:
金戟 2020-11-30 21:32:13 +08:00
parent 1bf6ab3b21
commit 1af6b8c8b6
1 changed files with 15 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
testable-processor/src/main/java/com/alibaba/testable/processor/translator/EnablePrivateAccessTranslator.java
View File

@ -10,6 +10,8 @@ import com.sun.tools.javac.util.Name;
import java.lang.reflect.Field;
import java.lang.reflect.Method;
import java.lang.reflect.Modifier;
import java.net.URL;
import java.net.URLClassLoader;
/**
* Travel AST
@ -41,7 +43,19 @@ public class EnablePrivateAccessTranslator extends BaseTranslator {
this.sourceClassName = testClassName.substring(0, testClassName.length() - ConstPool.TEST_POSTFIX.length());
this.privateAccessStatementGenerator = new PrivateAccessStatementGenerator(cx);
try {
Class<?> cls = Class.forName(pkgName + "." + sourceClassName);
Class<?> cls = null;
String sourceClassFullName = pkgName + "." + sourceClassName;
try {
cls = Class.forName(sourceClassFullName);
} catch (ClassNotFoundException e) {
// fit for gradle build
String path = "file:" + System.getProperty("user.dir") + "/build/classes/java/main/";
cls = new URLClassLoader(new URL[]{new URL(path)}).loadClass(sourceClassFullName);
}
if (cls == null) {
System.err.println("Failed to load source class: " + sourceClassFullName);
return;
}
Field[] fields = cls.getDeclaredFields();
for (Field f : fields) {
if (Modifier.isFinal(f.getModifiers()) || Modifier.isPrivate(f.getModifiers())) {