/*
* Copyright 2012 Red Hat, Inc. and/or its affiliates.
*
* Licensed under the Eclipse Public License version 1.0, available at
* http://www.eclipse.org/legal/epl-v10.html
*/
package org.jboss.forge.furnace.proxy.classloader;
import org.jboss.arquillian.container.test.api.Deployment;
import org.jboss.arquillian.junit.Arquillian;
import org.jboss.forge.arquillian.archive.ForgeArchive;
import org.jboss.forge.arquillian.services.LocalServices;
import org.jboss.forge.classloader.mock.result.AbstractClass;
import org.jboss.forge.classloader.mock.result.BasicInterface;
import org.jboss.forge.classloader.mock.result.Implementation;
import org.jboss.forge.classloader.mock.result.InstanceFactory;
import org.jboss.forge.classloader.mock.result.InstanceFactoryImpl;
import org.jboss.forge.classloader.mock.result.SuperInterface;
import org.jboss.forge.furnace.addons.AddonId;
import org.jboss.forge.furnace.addons.AddonRegistry;
import org.jboss.forge.furnace.proxy.ClassLoaderAdapterBuilder;
import org.jboss.forge.furnace.proxy.Proxies;
import org.jboss.forge.furnace.util.ClassLoaders;
import org.jboss.shrinkwrap.api.ShrinkWrap;
import org.junit.Assert;
import org.junit.Test;
import org.junit.runner.RunWith;
@RunWith(Arquillian.class)
public class ClassLoaderAdapterResultProxyTest
{
@Deployment(order = 3)
public static ForgeArchive getDeployment()
{
ForgeArchive archive = ShrinkWrap
.create(ForgeArchive.class)
.addClasses(BasicInterface.class,
SuperInterface.class,
AbstractClass.class,
Implementation.class,
InstanceFactory.class,
InstanceFactoryImpl.class)
.addAsLocalServices(ClassLoaderAdapterResultProxyTest.class);
return archive;
}
@Deployment(name = "dep,1", testable = false, order = 2)
public static ForgeArchive getDeploymentDep1()
{
ForgeArchive archive = ShrinkWrap.create(ForgeArchive.class)
.addClasses(BasicInterface.class,
SuperInterface.class,
AbstractClass.class,
Implementation.class,
InstanceFactory.class,
InstanceFactoryImpl.class)
.addBeansXML();
return archive;
}
@Test
public void testProxiesUpwrapProxyTypeUsesSpecifiedClassloader() throws Exception
{
AddonRegistry registry = LocalServices.getFurnace(getClass().getClassLoader())
.getAddonRegistry();
ClassLoader thisLoader = ClassLoaderAdapterResultProxyTest.class.getClassLoader();
ClassLoader dep1Loader = registry.getAddon(AddonId.from("dep", "1")).getClassLoader();
Class<?> foreignType = dep1Loader.loadClass(InstanceFactoryImpl.class.getName());
try
{
Implementation local = (Implementation) foreignType.getMethod("getInstance")
.invoke(foreignType.newInstance());
Assert.fail("Should have received a " + ClassCastException.class.getName() + " but got a real object ["
+ local + "]");
}
catch (ClassCastException e)
{
}
catch (Exception e)
{
Assert.fail("Should have received a " + ClassCastException.class.getName() + " but was: " + e);
}
Object delegate = foreignType.newInstance();
InstanceFactoryImpl enhancedFactory = (InstanceFactoryImpl) ClassLoaderAdapterBuilder.callingLoader(thisLoader)
.delegateLoader(dep1Loader).enhance(delegate);
Assert.assertTrue(Proxies.isForgeProxy(enhancedFactory));
Implementation enhancedInstance = enhancedFactory.getInstance();
Class<?> unwrappedType = Proxies.unwrapProxyTypes(enhancedInstance.getClass(), dep1Loader);
Assert.assertEquals(ClassLoaders.loadClass(dep1Loader, Implementation.class.getName()), unwrappedType);
}
@Test
public void testSharedImplementationTypeIncludedInProxy() throws Exception
{
AddonRegistry registry = LocalServices.getFurnace(getClass().getClassLoader())
.getAddonRegistry();
ClassLoader thisLoader = ClassLoaderAdapterResultProxyTest.class.getClassLoader();
ClassLoader dep1Loader = registry.getAddon(AddonId.from("dep", "1")).getClassLoader();
Class<?> foreignType = dep1Loader.loadClass(InstanceFactoryImpl.class.getName());
try
{
Implementation local = (Implementation) foreignType.getMethod("getInstance")
.invoke(foreignType.newInstance());
Assert.fail("Should have received a " + ClassCastException.class.getName() + " but got a real object ["
+ local + "]");
}
catch (ClassCastException e)
{
}
catch (Exception e)
{
Assert.fail("Should have received a " + ClassCastException.class.getName() + " but was: " + e);
}
Object delegate = foreignType.newInstance();
InstanceFactoryImpl enhancedFactory = (InstanceFactoryImpl) ClassLoaderAdapterBuilder.callingLoader(thisLoader)
.delegateLoader(dep1Loader).enhance(delegate);
Assert.assertTrue(Proxies.isForgeProxy(enhancedFactory));
Implementation enhancedInstance = enhancedFactory.getInstance();
Assert.assertTrue(Proxies.isForgeProxy(enhancedInstance));
}
}