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

reduce log reset frequency

Browse Source
This commit is contained in:
金戟 2020-12-07 23:26:18 +08:00
parent 65807eece7
commit bd0ec5f4b7
1 changed files with 11 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -40,22 +40,24 @@ public class TestableClassTransformer implements ClassFileTransformer {
// Ignore system class and reloaded class
return null;
}
byte[] bytes = null;
try {
if (shouldTransformAsSourceClass(className)) {
// it's a source class with testable enabled
LogUtil.diagnose("Handling source class %s", className);
List<MethodInfo> injectMethods = getTestableMockMethods(ClassUtil.getTestClassName(className));
return new SourceClassHandler(injectMethods).getBytes(classFileBuffer);
bytes = new SourceClassHandler(injectMethods).getBytes(classFileBuffer);
resetMockContext();
} else if (shouldTransformAsTestClass(className)) {
// it's a test class with testable enabled
LogUtil.diagnose("Handling test class %s", className);
return new TestClassHandler().getBytes(classFileBuffer);
bytes = new TestClassHandler().getBytes(classFileBuffer);
resetMockContext();
}
LogUtil.resetLogLevel();
} catch (IOException e) {
return null;
LogUtil.warn("Failed to transform class " + className);
}
return null;
return bytes;
}
private boolean shouldTransformAsSourceClass(String className) {
@ -155,6 +157,10 @@ public class TestableClassTransformer implements ClassFileTransformer {
}
}
private void resetMockContext() {
LogUtil.resetLogLevel();
}
/**
* Split desc to "first parameter" and "desc of rest parameters"
* @param desc method desc