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

adapt source class loader to intellij context

Browse Source
This commit is contained in:
金戟 2020-12-26 20:51:13 +08:00
parent d7945f52f3
commit cbdab5a246
2 changed files with 21 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
testable-processor/src/main/java/com/alibaba/testable/processor/EnablePrivateAccessProcessor.java
View File

@ -80,8 +80,7 @@ public class EnablePrivateAccessProcessor extends AbstractProcessor {
private void processClassElement(Symbol.ClassSymbol clazz) { private void processClassElement(Symbol.ClassSymbol clazz) {
JCTree tree = cx.trees.getTree(clazz); JCTree tree = cx.trees.getTree(clazz);
String pkgName = ((Symbol.PackageSymbol)clazz.owner).fullname.toString(); tree.accept(new EnablePrivateAccessTranslator(clazz, cx));
tree.accept(new EnablePrivateAccessTranslator(pkgName, clazz.getSimpleName().toString(), cx));
} }
} }

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

@ -4,6 +4,7 @@ import com.alibaba.testable.processor.constant.ConstPool;
import com.alibaba.testable.processor.generator.PrivateAccessStatementGenerator; import com.alibaba.testable.processor.generator.PrivateAccessStatementGenerator;
import com.alibaba.testable.processor.model.MemberType; import com.alibaba.testable.processor.model.MemberType;
import com.alibaba.testable.processor.model.TestableContext; import com.alibaba.testable.processor.model.TestableContext;
import com.sun.tools.javac.code.Symbol;
import com.sun.tools.javac.tree.JCTree.*; import com.sun.tools.javac.tree.JCTree.*;
import com.sun.tools.javac.util.ListBuffer; import com.sun.tools.javac.util.ListBuffer;
import com.sun.tools.javac.util.Name; import com.sun.tools.javac.util.Name;
@ -21,6 +22,9 @@ import java.net.URLClassLoader;
*/ */
public class EnablePrivateAccessTranslator extends BaseTranslator { public class EnablePrivateAccessTranslator extends BaseTranslator {
private static final String IDEA_PATHS_SELECTOR = "idea.paths.selector";
private static final String USER_DIR = "user.dir";
/** /**
* Name of source class * Name of source class
*/ */
@ -40,7 +44,9 @@ public class EnablePrivateAccessTranslator extends BaseTranslator {
private final PrivateAccessStatementGenerator privateAccessStatementGenerator; private final PrivateAccessStatementGenerator privateAccessStatementGenerator;
public EnablePrivateAccessTranslator(String pkgName, String testClassName, TestableContext cx) { public EnablePrivateAccessTranslator(Symbol.ClassSymbol clazz, TestableContext cx) {
String pkgName = ((Symbol.PackageSymbol)clazz.owner).fullname.toString();
String testClassName = clazz.getSimpleName().toString();
String sourceClass = testClassName.substring(0, testClassName.length() - ConstPool.TEST_POSTFIX.length()); String sourceClass = testClassName.substring(0, testClassName.length() - ConstPool.TEST_POSTFIX.length());
this.privateAccessStatementGenerator = new PrivateAccessStatementGenerator(cx); this.privateAccessStatementGenerator = new PrivateAccessStatementGenerator(cx);
this.sourceClassName = cx.names.fromString(sourceClass); this.sourceClassName = cx.names.fromString(sourceClass);
@ -50,9 +56,19 @@ public class EnablePrivateAccessTranslator extends BaseTranslator {
try { try {
cls = Class.forName(sourceClassFullName); cls = Class.forName(sourceClassFullName);
} catch (ClassNotFoundException e) { } catch (ClassNotFoundException e) {
// fit for gradle build if (System.getProperty(IDEA_PATHS_SELECTOR) != null) {
String path = "file:" + System.getProperty("user.dir") + "/build/classes/java/main/"; // fit for intellij 2020.3+
cls = new URLClassLoader(new URL[]{new URL(path)}).loadClass(sourceClassFullName); String sourceFileWrapperString = clazz.sourcefile.toString();
String sourceFilePath = sourceFileWrapperString.substring(
sourceFileWrapperString.lastIndexOf("[") + 1, sourceFileWrapperString.indexOf("]"));
String targetFolderPath = sourceFilePath.substring(0, sourceFilePath.lastIndexOf("/src/")) +
"/target/classes/";
cls = new URLClassLoader(new URL[] {new URL(targetFolderPath)}).loadClass(sourceClassFullName);
} else {
// 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) { if (cls == null) {
System.err.println("Failed to load source class: " + sourceClassFullName); System.err.println("Failed to load source class: " + sourceClassFullName);