third_codes/mapstruct
1
0
Fork 0
mirror of https://github.com/mapstruct/mapstruct.git synced 2025-07-12 00:00:08 +08:00
Code Issues Packages Projects Releases Wiki Activity

extend current classloader instead of creating a classloader chain.

Browse Source
This commit is contained in:
Andreas Gudian 2014-02-16 22:24:59 +01:00
parent f69069c2f8
commit ace47f7e67
1 changed files with 11 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
processor/src/test/java/org/mapstruct/ap/testutil/MapperTestBase.java
View File

@ -72,6 +72,7 @@ public abstract class MapperTestBase {
private static final String LINE_SEPARATOR = System.getProperty( "line.separator" );
private static final DiagnosticDescriptorComparator COMPARATOR = new DiagnosticDescriptorComparator();
private static volatile boolean enhancedClassloader = false;
private JavaCompiler compiler;
private String sourceDir;
@ -104,12 +105,16 @@ public abstract class MapperTestBase {
createOutputDirs();
Thread.currentThread().setContextClassLoader(
new URLClassLoader(
new URL[] { new File( classOutputDir ).toURI().toURL() },
Thread.currentThread().getContextClassLoader()
)
);
if ( !enhancedClassloader ) {
// we need to make sure that the the generated classes are loaded by the same classloader as the test has
// been loaded already. Otherwise some tests won't work.
URLClassLoader classLoader = (URLClassLoader) ClassLoader.getSystemClassLoader();
Class<?> clazz = URLClassLoader.class;
Method method = clazz.getDeclaredMethod( "addURL", new Class[] { URL.class } );
method.setAccessible( true );
method.invoke( classLoader, new File( classOutputDir ).toURI().toURL() );
enhancedClassloader = true;
}
}
@BeforeMethod