URLClassLoaderTypeTest.java

package org.oddjob.util;

import java.io.File;
import java.io.IOException;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.net.URISyntaxException;
import java.net.URL;
import java.net.URLClassLoader;
import java.util.Arrays;

import org.apache.log4j.Logger;
import org.oddjob.Oddjob;
import org.oddjob.OddjobLookup;
import org.oddjob.arooa.xml.XMLConfiguration;
import org.oddjob.io.FilesType;
import org.oddjob.tools.CompileJob;
import org.oddjob.tools.ConsoleCapture;
import org.oddjob.tools.OurDirs;

import junit.framework.TestCase;

public class URLClassLoaderTypeTest extends TestCase {

	private static final Logger logger = 
		Logger.getLogger(URLClassLoaderTypeTest.class);
	
	@Override
	protected void setUp() throws Exception {
		super.setUp();
		
		logger.debug("-----------------  " + getName() + "  -----------------");

		logger.debug("ClassLoaders: System=" + 
				ClassLoader.getSystemClassLoader() + ", Context=" +
				Thread.currentThread().getContextClassLoader());
		ClassLoader cl = Thread.currentThread().getContextClassLoader();
		if (cl instanceof URLClassLoader) {
			URL[] urls = ((URLClassLoader) cl).getURLs();
			logger.debug("URLS: " + Arrays.toString(urls));
		}
	}
	
	public void testLoadMixedJob() throws Exception {
		
		ClassLoader existingContextClassLoader = 
				Thread.currentThread().getContextClassLoader();
		
		OurDirs dirs = new OurDirs();
		
		File check = dirs.relative("test/classloader/AJob.class");
		if (!check.exists()) {
			compileSample(dirs);
		}
		
		URLClassLoaderType test = new URLClassLoaderType();
		test.setFiles(new File[] { dirs.relative("test/classloader") } );
		test.setParent(getClass().getClassLoader());
		
		assertEquals("URLClassLoaderType: [" + 
				dirs.relative("test/classloader").toString() + "]",
				test.toString());
		
		test.configured();
		ClassLoader classLoader = test.toValue();
		
		Oddjob oddjob = new Oddjob();
		oddjob.setClassLoader(classLoader);
	
		String xml = 
			"<oddjob>" +
			"<job>" +
			"<bean id='test' class='AJob'/>" +
			"</job>" +
			"</oddjob>";
		
		oddjob.setConfiguration(new XMLConfiguration("XML", xml));
		
		oddjob.run();
		
		OddjobLookup lookup = 
			new OddjobLookup(oddjob);
			
		ClassLoader classLoaderWhenRunning = lookup.lookup(
				"test.classLoader", ClassLoader.class);
		
		ClassLoader jobClassLoader = lookup.lookup(
				"test.class.classLoader", ClassLoader.class);
		
		assertEquals(classLoader, jobClassLoader);
		
		assertEquals(classLoader,
				classLoaderWhenRunning);
		
		assertEquals(existingContextClassLoader,
				Thread.currentThread().getContextClassLoader());
	}
	
	static public void compileSample(final OurDirs dirs) throws IOException {

		File dir = dirs.relative("test/classloader");
		if (new File(dir, "AJob.class").exists()) {
			return;
		}
		
		FilesType sources = new FilesType();
		sources.setFiles(dirs.relative(
				"test/classloader").getPath() +
				File.separator + "*.java"); 

		CompileJob compile = new CompileJob();		
		compile.setFiles(sources.toFiles());
		
		compile.run();
		
		if (compile.getResult() != 0) {
			throw new RuntimeException(
					"Compile failed. See standard output for details.");
		}
	}

	public void testInOddjob() throws URISyntaxException, IOException {
	
		OurDirs dirs = new OurDirs();
		
		compileSample(dirs);
		
    	URL url = getClass().getClassLoader().getResource("org/oddjob/util/URLClassLoader.xml");
    	
    	File file = new File(url.toURI());
		
		Oddjob oddjob = new Oddjob();
		oddjob.setArgs(new String[] { dirs.base().getAbsolutePath() });
		oddjob.setFile(file);
		
		oddjob.run();
		
		Object aJob = new OddjobLookup(oddjob).lookup("nested/x");
		
		assertEquals("AJob", aJob.getClass().getName());
	}
	
	public void testNoParent() {
		
		URLClassLoaderType test = new URLClassLoaderType();
		
		test.setFiles(new File[] { });
		test.setNoInherit(true);
		
		test.configured();
		
		ClassLoader loader = test.toValue();
		
		assertNull(loader.getParent());
		
	}
	
	public void testFromLaunchJar() throws SecurityException, NoSuchMethodException, ClassNotFoundException, IllegalArgumentException, IllegalAccessException, InvocationTargetException, IOException {
		
		OurDirs dirs = new OurDirs();
		
		File check = dirs.relative("test/classloader/AJob.class");
		if (!check.exists()) {
			compileSample(dirs);
		}
		
		
		File oddjobFile = dirs.relative("test/classloader/classloader-test.xml");
		
		URLClassLoaderType first = new URLClassLoaderType();
		first.setFiles(new File[] { dirs.relative("run-oddjob.jar")});
		first.setNoInherit(true);
		
		first.configured();
		
		ClassLoader loader = first.toValue();
		
		Class<?> launcher = loader.loadClass("org.oddjob.launch.Launcher");
		
		Method m = launcher.getMethod("main", String[].class);
		
		String[] args = new String[] { "-f", oddjobFile.getCanonicalPath() };
				
		ConsoleCapture console = new ConsoleCapture();
		try (ConsoleCapture.Close close = console.captureConsole()) {
			
			m.invoke(null, (Object) args);
		}
		
		console.dump(logger);
		
		String[] lines = console.getLines();
		
		assertEquals("Worked.", lines[1].trim());
		
	}	
}