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

get resource could be null

Browse Source
This commit is contained in:
金戟 2021-04-17 18:45:29 +08:00
parent d10478f95f
commit 8dde402a8c
3 changed files with 34 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
testable-agent/src/main/java/com/alibaba/testable/agent/constant/ConstPool.java
View File

@ -10,6 +10,7 @@ public class ConstPool {
public static final String FIELD_SCOPE = "scope";
public static final String PROPERTY_USER_DIR = "user.dir";
public static final String PROPERTY_TEMP_DIR = "java.io.tmpdir";
public static final String MOCK_WITH = "com.alibaba.testable.core.annotation.MockWith";
public static final String DUMP_TO = "com.alibaba.testable.core.annotation.DumpTo";

28
testable-agent/src/main/java/com/alibaba/testable/agent/util/GlobalConfig.java
View File

@ -5,9 +5,11 @@ import com.alibaba.testable.core.model.MockScope;
import com.alibaba.testable.core.util.LogUtil;
import java.io.File;
import java.net.URL;
import java.util.ArrayList;
import java.util.List;
import static com.alibaba.testable.agent.constant.ConstPool.PROPERTY_TEMP_DIR;
import static com.alibaba.testable.agent.constant.ConstPool.PROPERTY_USER_DIR;
import static com.alibaba.testable.core.util.PathUtil.createFolder;
@ -22,8 +24,9 @@ public class GlobalConfig {
private static final String DISABLE_LOG_FILE = "null";
private static final String COMMA = ",";
private static final String DOT = ".";
private static final String TESTABLE_AGENT_LOG = "testable-agent.log";
private static final String DEFAULT_MAVEN_OUTPUT_FOLDER = "target";
private static final String DEFAULT_GRADLE_OUTPUT_FOLDER = "gradle";
private static String logFile = null;
private static String dumpPath = null;
@ -78,16 +81,33 @@ public class GlobalConfig {
public static void setupLogRootPath() {
if (logFile == null) {
String baseFolder = PathUtil.getFirstLevelFolder(System.getProperty(PROPERTY_USER_DIR),
Object.class.getResource("/").getPath());
// Use default log file location
String baseFolder = getBuildOutputFolder();
if (!baseFolder.isEmpty()) {
LogUtil.setGlobalLogPath(PathUtil.join(baseFolder, TESTABLE_AGENT_LOG));
String logFilePath = PathUtil.join(baseFolder, TESTABLE_AGENT_LOG);
LogUtil.setGlobalLogPath(logFilePath);
LogUtil.verbose("Generate testable agent log file at: %s", logFilePath);
}
} else if (!DISABLE_LOG_FILE.equals(logFile)) {
// Use custom log file location
LogUtil.setGlobalLogPath(PathUtil.join(System.getProperty(PROPERTY_USER_DIR), logFile));
}
}
private static String getBuildOutputFolder() {
String contextFolder = System.getProperty(PROPERTY_USER_DIR);
URL rootResourceFolder = Object.class.getResource("/");
if (rootResourceFolder != null) {
return PathUtil.getFirstLevelFolder(contextFolder, rootResourceFolder.getPath());
} else if (PathUtil.folderExists(PathUtil.join(contextFolder, DEFAULT_MAVEN_OUTPUT_FOLDER))) {
return PathUtil.join(contextFolder, DEFAULT_MAVEN_OUTPUT_FOLDER);
} else if (PathUtil.folderExists(PathUtil.join(contextFolder, DEFAULT_GRADLE_OUTPUT_FOLDER))) {
return PathUtil.join(contextFolder, DEFAULT_GRADLE_OUTPUT_FOLDER);
} else {
return System.getProperty(PROPERTY_TEMP_DIR);
}
}
public static void setEnhanceThreadLocal(boolean enabled) {
enhanceThreadLocal = enabled;
}

9
testable-agent/src/main/java/com/alibaba/testable/agent/util/PathUtil.java
View File

@ -28,4 +28,13 @@ public class PathUtil {
int pos = targetPath.indexOf(separator, rootFolder.length() + 1);
return pos > 0 ? targetPath.substring(0, pos) : "";
}
/**
* Check whether specified path exists and is folder
* @param path path to check
* @return exists or not
*/
public static boolean folderExists(String path) {
return new File(path).isDirectory();
}
}