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

implement util function to get current member method and test case

Browse Source
This commit is contained in:
金戟 2020-07-26 20:20:32 +08:00
parent 5ac7f354f3
commit a814e51f66
5 changed files with 48 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -11,6 +11,8 @@ public class ConstPool {
public static final String DOT = ".";
public static final String SLASH = "/";
public static final String TEST_POSTFIX = "Test";
public static final List<String> SYS_CLASSES = new ArrayList<String>();
static {
SYS_CLASSES.add("java/lang/StringBuilder");

4
agent/src/main/java/com/alibaba/testable/agent/transformer/TestableClassTransformer.java
View File

@ -1,5 +1,6 @@
package com.alibaba.testable.agent.transformer;
import com.alibaba.testable.agent.constant.ConstPool;
import com.alibaba.testable.agent.handler.SourceClassHandler;
import com.alibaba.testable.agent.handler.TestClassHandler;
import com.alibaba.testable.agent.util.ClassUtil;
@ -19,7 +20,6 @@ public class TestableClassTransformer implements ClassFileTransformer {
private static final String ENABLE_TESTABLE = "com.alibaba.testable.core.annotation.EnableTestable";
private static final String ENABLE_TESTABLE_INJECT = "com.alibaba.testable.core.annotation.EnableTestableInject";
private static final String TEST_POSTFIX = "Test";
private static final Set<String> loadedClassNames = new HashSet<String>();
@ -31,7 +31,7 @@ public class TestableClassTransformer implements ClassFileTransformer {
}
List<String> annotations = ClassUtil.getAnnotations(className);
List<String> testAnnotations = ClassUtil.getAnnotations(className + TEST_POSTFIX);
List<String> testAnnotations = ClassUtil.getAnnotations(className + ConstPool.TEST_POSTFIX);
try {
if (annotations.contains(ENABLE_TESTABLE_INJECT) || testAnnotations.contains(ENABLE_TESTABLE)) {
loadedClassNames.add(className);

5
core/src/main/java/com/alibaba/testable/core/processor/EnableTestableProcessor.java
View File

@ -2,6 +2,7 @@ package com.alibaba.testable.core.processor;
import com.alibaba.testable.core.annotation.EnableTestable;
import com.alibaba.testable.core.translator.EnableTestableTranslator;
import com.alibaba.testable.core.util.ConstPool;
import com.alibaba.testable.core.util.ResourceUtil;
import com.sun.tools.javac.code.Symbol;
import com.sun.tools.javac.tree.JCTree;
@ -42,7 +43,7 @@ public class EnableTestableProcessor extends BaseProcessor {
}
private boolean isTestClass(Name name) {
return name.toString().endsWith("Test");
return name.toString().endsWith(ConstPool.TEST_POSTFIX);
}
private void processClassElement(Symbol.ClassSymbol clazz) {
@ -56,7 +57,7 @@ public class EnableTestableProcessor extends BaseProcessor {
private String getOriginClassName(Symbol.ClassSymbol clazz) {
String testClassName = clazz.getSimpleName().toString();
return testClassName.substring(0, testClassName.length() - "Test".length());
return testClassName.substring(0, testClassName.length() - ConstPool.TEST_POSTFIX.length());
}
private void createTestableAgentJar() {

1
core/src/main/java/com/alibaba/testable/core/util/ConstPool.java
View File

@ -15,5 +15,6 @@ public final class ConstPool {
public static final String TESTABLE_PRIVATE_ACCESSOR = "com.alibaba.testable.core.accessor.PrivateAccessor";
public static final String ANNOTATION_TESTABLE_INJECT = "com.alibaba.testable.core.annotation.TestableInject";
public static final String TESTABLE_SETUP_METHOD_NAME = "testableSetup";
public static final String TEST_POSTFIX = "Test";
}

40
core/src/main/java/com/alibaba/testable/core/util/TestableUtil.java Normal file
View File

@ -0,0 +1,40 @@
package com.alibaba.testable.core.util;
/**
* @author flin
*/
public class TestableUtil {
private static final String TESTABLE_NE = "n.e";
public static String currentMemberMethodName() {
StackTraceElement[] stack = Thread.currentThread().getStackTrace();
for (int i = 0; i < stack.length; i++) {
if (stack[i].getClassName().equals(TESTABLE_NE)) {
return stack[i + 1].getMethodName();
}
}
return "";
}
public static String currentTestCaseName(Object testClassRef) {
StackTraceElement[] stack = getMainThread().getStackTrace();
String testClassName = testClassRef.getClass().getName();
for (int i = stack.length - 1; i >= 0; i--) {
if (stack[i].getClassName().equals(testClassName)) {
return stack[i].getMethodName();
}
}
return "";
}
private static Thread getMainThread() {
for (Thread t : Thread.getAllStackTraces().keySet()) {
if (t.getId() == 1L) {
return t;
}
}
return Thread.currentThread();
}
}