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

move dump method to util package

Browse Source
This commit is contained in:
金戟 2021-03-31 10:34:45 +08:00
parent b4da23a90e
commit e0a4a39d55
2 changed files with 34 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -16,17 +16,15 @@ import org.objectweb.asm.tree.ClassNode;
import org.objectweb.asm.tree.InnerClassNode; import org.objectweb.asm.tree.InnerClassNode;
import javax.lang.model.type.NullType; import javax.lang.model.type.NullType;
import java.io.FileOutputStream;
import java.io.IOException;
import java.lang.instrument.ClassFileTransformer; import java.lang.instrument.ClassFileTransformer;
import java.security.ProtectionDomain; import java.security.ProtectionDomain;
import java.util.List; import java.util.List;
import static com.alibaba.testable.agent.constant.ConstPool.KOTLIN_POSTFIX_COMPANION; import static com.alibaba.testable.agent.constant.ConstPool.KOTLIN_POSTFIX_COMPANION;
import static com.alibaba.testable.core.constant.ConstPool.*;
import static com.alibaba.testable.agent.util.ClassUtil.toJavaStyleClassName; import static com.alibaba.testable.agent.util.ClassUtil.toJavaStyleClassName;
import static com.alibaba.testable.core.constant.ConstPool.DOLLAR;
import static com.alibaba.testable.core.constant.ConstPool.TEST_POSTFIX; import static com.alibaba.testable.core.constant.ConstPool.TEST_POSTFIX;
import static org.objectweb.asm.Opcodes.*; import static org.objectweb.asm.Opcodes.ACC_STATIC;
/** /**
* @author flin * @author flin
@ -69,14 +67,14 @@ public class TestableClassTransformer implements ClassFileTransformer {
// it's a mock class // it's a mock class
LogUtil.diagnose("Handling mock class %s", className); LogUtil.diagnose("Handling mock class %s", className);
bytes = new MockClassHandler(className).getBytes(bytes); bytes = new MockClassHandler(className).getBytes(bytes);
dumpByte(className, GlobalConfig.getDumpPath(), bytes); BytecodeUtil.dumpByte(className, GlobalConfig.getDumpPath(), bytes);
} else { } else {
String mockClass = foundMockForTestClass(className); String mockClass = foundMockForTestClass(className);
if (mockClass != null) { if (mockClass != null) {
// it's a test class with testable enabled // it's a test class with testable enabled
LogUtil.diagnose("Handling test class %s", className); LogUtil.diagnose("Handling test class %s", className);
bytes = new TestClassHandler().getBytes(bytes); bytes = new TestClassHandler().getBytes(bytes);
dumpByte(className, GlobalConfig.getDumpPath(), bytes); BytecodeUtil.dumpByte(className, GlobalConfig.getDumpPath(), bytes);
} else { } else {
mockClass = foundMockForSourceClass(className); mockClass = foundMockForSourceClass(className);
if (mockClass != null) { if (mockClass != null) {
@ -84,7 +82,7 @@ public class TestableClassTransformer implements ClassFileTransformer {
List<MethodInfo> injectMethods = mockClassParser.getTestableMockMethods(mockClass); List<MethodInfo> injectMethods = mockClassParser.getTestableMockMethods(mockClass);
LogUtil.diagnose("Handling source class %s", className); LogUtil.diagnose("Handling source class %s", className);
bytes = new SourceClassHandler(injectMethods, mockClass).getBytes(bytes); bytes = new SourceClassHandler(injectMethods, mockClass).getBytes(bytes);
dumpByte(className, GlobalConfig.getDumpPath(), bytes); BytecodeUtil.dumpByte(className, GlobalConfig.getDumpPath(), bytes);
} }
} }
} }
@ -95,26 +93,10 @@ public class TestableClassTransformer implements ClassFileTransformer {
} finally { } finally {
LogUtil.resetLogLevel(); LogUtil.resetLogLevel();
} }
dumpByte(className, getDumpPathByAnnotation(cn), bytes); BytecodeUtil.dumpByte(className, getDumpPathByAnnotation(cn), bytes);
return bytes; return bytes;
} }
private void dumpByte(String className, String dumpPath, byte[] bytes) {
if (dumpPath == null) {
return;
}
try {
String dumpFile = StringUtil.joinPath(dumpPath,
className.replace(SLASH, DOT).replace(DOLLAR, UNDERLINE) + ".class");
LogUtil.verbose("Dump class: " + dumpFile);
FileOutputStream stream = new FileOutputStream(dumpFile);
stream.write(bytes);
stream.close();
} catch (IOException e) {
e.printStackTrace();
}
}
private String foundMockForSourceClass(String className) { private String foundMockForSourceClass(String className) {
String mockClass = lookForMockWithAnnotationAsSourceClass(className); String mockClass = lookForMockWithAnnotationAsSourceClass(className);
if (mockClass != null) { if (mockClass != null) {

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

@ -1,8 +1,14 @@
package com.alibaba.testable.agent.util; package com.alibaba.testable.agent.util;
import com.alibaba.testable.core.util.LogUtil;
import java.io.FileOutputStream;
import java.io.IOException;
import java.util.HashMap; import java.util.HashMap;
import java.util.Map; import java.util.Map;
import static com.alibaba.testable.core.constant.ConstPool.*;
import static com.alibaba.testable.core.constant.ConstPool.UNDERLINE;
import static org.objectweb.asm.Opcodes.*; import static org.objectweb.asm.Opcodes.*;
/** /**
@ -193,4 +199,26 @@ public class BytecodeUtil {
access |= ACC_PUBLIC; access |= ACC_PUBLIC;
return access; return access;
} }
/**
* Dump byte code to specified class file
* @param className original class name
* @param dumpPath folder to store class file
* @param bytes original class bytes
*/
public static void dumpByte(String className, String dumpPath, byte[] bytes) {
if (dumpPath == null) {
return;
}
try {
String dumpFile = StringUtil.joinPath(dumpPath,
className.replace(SLASH, DOT).replace(DOLLAR, UNDERLINE) + ".class");
LogUtil.verbose("Dump class: " + dumpFile);
FileOutputStream stream = new FileOutputStream(dumpFile);
stream.write(bytes);
stream.close();
} catch (IOException e) {
e.printStackTrace();
}
}
} }