How to use

TestContainer

in

org.jboss.arquillian.container.weld.ee.embedded_1_1.mock

/**
 * Bootstrap and shutdown the container.
 * 
 * @param runnable a {@link Runnable} to be called whilst the container is
 *           active
 * 
 */
public void run(Runnable runnable) throws Exception
{
  TestContainer container = null;
  try
  {
   container = new TestContainer(beansXml, classes);
   container.startContainer().ensureRequestActive();
   runnable.run(container.getBeanManager(container.getDeployment().getBeanDeploymentArchives().iterator().next()));
  }
  finally
  {
   if (container != null)
   {
     container.stopContainer();
   }
  }
}

public boolean deploy(Collection<Class<?>> classes, Collection<URL> beansXml) {
  this.testContainer = new TestContainer(beansXml, classes);
  try {
    testContainer.startContainer();
  } catch (Exception e) {
    this.deploymentException = new DeploymentException("Error deploying beans", e);
    return false;
  }
  testContainer.ensureRequestActive();
  return true;
}

public ProtocolMetaData deploy(Archive<?> archive) throws DeploymentException
{  
 ShrinkWrapClassLoader classLoader = getClassLoader(archive);
 ContextClassLoaderManager classLoaderManager = new ContextClassLoaderManager(classLoader);
 classLoaderManager.enable();
 
 TestContainer container = new TestContainer(findArchiveId(archive), findBeansXml(archive), findBeanClasses(archive, classLoader), true);
 Bootstrap bootstrap = container.getBootstrap();
 contextClassLoaderManagerProducer.set(classLoaderManager);
 try {
   container.startContainer();
 } finally {
   // always set container - even if an exception occurs, we want the container to be cleaned up properly in undeploy()
   testContainerProducer.set(container);
   bootstrapProducer.set(bootstrap);
 }
 // Assume a flat structure
 WeldManager manager = container.getBeanManager(container.getDeployment().getBeanDeploymentArchives().iterator().next());
 weldManagerProducer.set(manager);
 beanManagerProducer.set(manager);
 return new ProtocolMetaData();
}

public void undeploy() {
  try {
    testContainer.stopContainer();
  } catch (Exception e) {
    log.log(Level.SEVERE, "Could not shut down container", e);
  }
  testContainer = null;
  deploymentException = null;
}

public TestContainer ensureRequestActive()
{
 RequestContext requestContext = instance().select(RequestContext.class, UnboundLiteral.INSTANCE).get();
 requestContext.activate();
 
 // TODO deactivate the conversation context
 
 BoundSessionContext sessionContext = instance().select(BoundSessionContext.class).get();
 sessionContext.associate(sessionStore);
 sessionContext.activate();
 return this;
}

classLoaderManager.enable();
TestContainer container = new TestContainer(findArchiveId(archive), findBeansXml(archive), findBeanClasses(archive, classLoader));
Bootstrap bootstrap = container.getBootstrap();
container.startContainer();
context.add(WeldManager.class, container.getBeanManager(container.getDeployment().getBeanDeploymentArchives().iterator().next()));

public boolean deploy(Collection<Class<?>> classes, Collection<URL> beansXml) {
  this.testContainer = new TestContainer(beansXml, classes);
  try {
    testContainer.startContainer();
  } catch (Exception e) {
    this.deploymentException = new DeploymentException("Error deploying beans", e);
    return false;
  }
  testContainer.ensureRequestActive();
  return true;
}

public void undeploy() {
  try {
    testContainer.stopContainer();
  } catch (Exception e) {
    log.log(Level.SEVERE, "Could not shut down container", e);
  }
  testContainer = null;
  deploymentException = null;
}

public TestContainer ensureRequestActive()
{
 RequestContext requestContext = instance().select(RequestContext.class, UnboundLiteral.INSTANCE).get();
 requestContext.activate();
 
 // TODO deactivate the conversation context
 
 BoundSessionContext sessionContext = instance().select(BoundSessionContext.class).get();
 sessionContext.associate(sessionStore);
 sessionContext.activate();
 return this;
}

/**
 * Bootstrap and shutdown the container.
 * 
 * @param runnable a {@link Runnable} to be called whilst the container is
 *           active
 * 
 */
public void run(Runnable runnable) throws Exception
{
  TestContainer container = null;
  try
  {
   container = new TestContainer(beansXml, classes);
   container.startContainer().ensureRequestActive();
   runnable.run(container.getBeanManager(container.getDeployment().getBeanDeploymentArchives().iterator().next()));
  }
  finally
  {
   if (container != null)
   {
     container.stopContainer();
   }
  }
}

public void undeploy(Context context, Archive<?> archive) throws DeploymentException
{
 TestContainer container = context.get(TestContainer.class);
 if(container != null)
 {
   container.stopContainer();
 }
 ContextClassLoaderManager classLoaderManager = context.get(ContextClassLoaderManager.class);
 classLoaderManager.disable();
}

/**
* Clean up the container, ending any active contexts
* 
*/
public TestContainer stopContainer()
{
 RequestContext requestContext = instance().select(RequestContext.class, UnboundLiteral.INSTANCE).get();
 if (requestContext.isActive())
 {
   requestContext.invalidate();
   requestContext.deactivate();
 }
 
 // TODO deactivate the conversation context
 
 BoundSessionContext sessionContext = instance().select(BoundSessionContext.class).get();
 if (sessionContext.isActive())
 {
   sessionContext.invalidate();
   sessionContext.deactivate();
   sessionContext.dissociate(sessionStore);
 }
 
 bootstrap.shutdown();
 
 return this;
}

public void undeploy(Archive<?> archive) throws DeploymentException
{
 TestContainer container = testContainerProducer.get();
 if(container != null)
 {
   container.stopContainer();
 }
 ContextClassLoaderManager classLoaderManager = contextClassLoaderManagerProducer.get();
 classLoaderManager.disable();
}

/**
* Clean up the container, ending any active contexts
* 
*/
public TestContainer stopContainer()
{
 RequestContext requestContext = instance().select(RequestContext.class, UnboundLiteral.INSTANCE).get();
 if (requestContext.isActive())
 {
   requestContext.invalidate();
   requestContext.deactivate();
 }
 
 // TODO deactivate the conversation context
 
 BoundSessionContext sessionContext = instance().select(BoundSessionContext.class).get();
 if (sessionContext.isActive())
 {
   sessionContext.invalidate();
   sessionContext.deactivate();
   sessionContext.dissociate(sessionStore);
 }
 
 bootstrap.shutdown();
 
 return this;
}