com.oneandone.ejbcdiunit.resourcesimulators.SessionContextSimulation Maven / Gradle / Ivy
Show all versions of ejb-cdi-unit Show documentation
package com.oneandone.ejbcdiunit.resourcesimulators;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.util.Set;
import javax.ejb.EJBLocalObject;
import javax.ejb.EJBObject;
import javax.ejb.SessionContext;
import javax.enterprise.inject.Vetoed;
import javax.enterprise.inject.spi.Bean;
import javax.enterprise.inject.spi.BeanManager;
import javax.xml.rpc.handler.MessageContext;
import org.jboss.weld.bean.proxy.InterceptionDecorationContext;
import net.sf.cglib.proxy.Enhancer;
import net.sf.cglib.proxy.InvocationHandler;
/**
* cdi-unit simulation of sessioncontext
*
* @author aschoerk
*/
@Vetoed
public class SessionContextSimulation extends EjbContextSimulation implements SessionContext {
private BeanManager beanManager;
/**
* this Object may only be created using new.
*
* @param preventInject has no function, just prevent default constructor
*/
public SessionContextSimulation(String preventInject) {
assert beanManager == null;
}
/**
* Used later to be able to handle getBusinessObject
*
* @param containerP the injecting bean (originally having @Resource)
* @param beanManagerP the CDI-Beanmanager
*/
public void setContainer(Bean containerP, BeanManager beanManagerP) {
this.beanManager = beanManagerP;
}
/**
* Obtain a reference to the EJB local object that is
* associated with the instance.
*
*
An instance of a session bean can call this method at
* anytime between the PostConstruct or
* ejbCreate and PreDestroy or
* ejbRemove methods, including from within these
* methods.
*
*
An instance can use this method, for example, when it wants to
* pass a reference to itself in a method argument or result.
*
* @return The EJB local object currently associated with the instance.
* @throws IllegalStateException Thrown if the instance invokes this
* method while the instance is in a state that does not allow the
* instance to invoke this method, or if the instance does not have
* a local interface.
* @since EJB 2.0
*/
@Override
public EJBLocalObject getEJBLocalObject() throws IllegalStateException {
throw new NotImplementedException("getEJBLocalObject not implemented in SessionContextSimulation of ejb-cdi-unit");
}
/**
* Obtain a reference to the EJB object that is currently associated with
* the instance.
*
*
An instance of a session enterprise Bean can call this
* method at anytime between the PostConstruct or
* ejbCreate and the PreDestroy or
* ejbRemove methods, including from within these
* methods.
*
*
An instance can use this method, for example, when it wants to
* pass a reference to itself in a method argument or result.
*
* @return The EJB object currently associated with the instance.
* @throws IllegalStateException Thrown if the instance invokes this
* method while the instance is in a state that does not allow the
* instance to invoke this method, or if the instance does not have
* a remote interface.
*/
@Override
public EJBObject getEJBObject() throws IllegalStateException {
throw new NotImplementedException("getObject not implemented in SessionContextSimulation of ejb-cdi-unit");
}
/**
* Obtain a reference to the JAX-RPC MessageContext.
*
*
An instance of a stateless session bean can call this method
* from any business method invoked through its web service
* endpoint interface.
*
*
Note: Support for web services invocations using JAX-RPC is optional as of EJB 3.2
*
* @return The MessageContext for this web service invocation.
* @throws IllegalStateException Thrown if this method is invoked
* while the instance is in a state that does not allow access
* to this method.
* @since EJB 2.1
*/
@Override
public MessageContext getMessageContext() throws IllegalStateException {
throw new NotImplementedException("getMessageContext not implemented in SessionContextSimulation of ejb-cdi-unit");
}
public static boolean startInterceptionDecorationContext() {
Method[] methods = InterceptionDecorationContext.class.getMethods();
for (Method m : methods) {
if (m.getParameterTypes().length == 0) {
if (m.getName().equals("startInterceptorContext")) {
callMethodThrowRTEIfNecessary(m);
return true;
}
if (m.getName().equals("startIfNotEmpty") || m.getName().equals("startIfNotOnTop")) {
Object result = callMethodThrowRTEIfNecessary(m);
return result != null;
}
}
}
return false;
}
public static Object callMethodThrowRTEIfNecessary(Method m) {
try {
return m.invoke(null);
} catch (Throwable e) {
throw new RuntimeException(e);
}
}
/**
* Obtain an object that can be used to invoke the current bean through
* a particular business interface view or its no-interface view.
*
* @param businessInterface One of the local business interfaces
* or remote business interfaces for this session bean.
* In addition, the bean class type can be used to acquire
* a reference to the bean's no-interface view.
* @return The business object corresponding to the given business
* interface or no-interface view.
* @throws IllegalStateException Thrown if invoked with a parameter
* that does not correspond to one of the beans' business interfaces
* or no-interface view.
* @since EJB 3.0
*/
@Override
public T getBusinessObject(Class businessInterface) throws IllegalStateException {
Set> beans = beanManager.getBeans(businessInterface);
if (beans.isEmpty() && businessInterface.getName().endsWith("_WeldSubclass")) {
beans = beanManager.getBeans(businessInterface.getSuperclass());
}
Bean bean = (Bean) beanManager.resolve(beans);
final Object testBean1 = beanManager.getReference(bean, bean.getBeanClass(), beanManager.createCreationalContext(bean));
Enhancer enhancer = new Enhancer();
enhancer.setSuperclass(businessInterface);
enhancer.setCallback(new InvocationHandler() {
@Override
public Object invoke(Object o, Method method, Object[] objects) throws Throwable {
startInterceptionDecorationContext();
try {
return method.invoke(testBean1, objects);
} catch (Throwable thw) {
if (thw instanceof InvocationTargetException) {
throw thw.getCause();
} else {
throw thw;
}
} finally {
InterceptionDecorationContext.endInterceptorContext();
}
}
});
Object proxy = enhancer.create();
return (T) proxy;
}
/**
* Obtain the business interface or no-interface view type through which the
* current business method invocation was made.
*
* @throws IllegalStateException Thrown if this method is called
* and the bean has not been invoked through a business interface or
* no-interface view.
* @since EJB 3.0
*/
@Override
public Class getInvokedBusinessInterface() throws IllegalStateException {
throw new NotImplementedException("getInvokedBusinessInterface not implemented in SessionContextSimulation of ejb-cdi-unit");
}
/**
* Check whether a client invoked the cancel method on the
* client Future object corresponding to the currently executing
* asynchronous business method.
*
* @return true if the client has invoked Future.cancel with a value of
* true for the mayInterruptIfRunning parameter.
* @throws IllegalStateException Thrown if not invoked from within an
* asynchronous business method invocation with return type
* Future<V>.
* @since EJB 3.1
*/
@Override
public boolean wasCancelCalled() throws IllegalStateException {
return false;
}
}