com.mockrunner.ejb.JNDIUtil Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of mockrunner-ejb Show documentation
Mock classes for enterprise java beans
There is a newer version: 2.0.7
package com.mockrunner.ejb;

import javax.naming.Context;
import javax.naming.InitialContext;
import javax.naming.NamingException;
import javax.transaction.UserTransaction;

import org.mockejb.jndi.MockContextFactory;

import com.mockrunner.base.NestedApplicationException;

/**
 * Util class for creating and managing the JNDI context
 */
public class JNDIUtil
{
    /**
     * Calls MockContextFactory.setAsInitial(), if the
     * MockContextFactory is not already the current
     * context factory.
     */
    public static void initMockContextFactory() throws NamingException
    {
        String factory = System.getProperty(Context.INITIAL_CONTEXT_FACTORY);
        if(null == factory || !factory.equals(MockContextFactory.class.getName()))
        {
            MockContextFactory.setAsInitial();
        }
    }
    
    /**
     * Calls MockContextFactory.revertSetAsInitial(), if the
     * MockContextFactory is the current context factory.
     */
    public static void resetMockContextFactory()
    {
        String factory = System.getProperty(Context.INITIAL_CONTEXT_FACTORY);
        if(null != factory && factory.equals(MockContextFactory.class.getName()))
        {
            MockContextFactory.revertSetAsInitial();
        }
    }
    
    /**
     * Tries to get the JNDI context from the specified configuration.
     * If the configuration returns null for the context,
     * this method initializes the MockEJB JNDI implementation.
     * @param configuration the configuration
     * @return the JNDI context
     */
    public static Context getContext(Configuration configuration)
    {
        Context context = configuration.getContext();
        if(null == context)
        {
            try
            {
                JNDIUtil.initMockContextFactory();
                context = new InitialContext();
            } 
            catch(NamingException exc)
            {
                throw new NestedApplicationException(exc);
            }
            configuration.setContext(context);
        }
        return context;
    }
    
    /**
     * Binds the specified UserTransaction to the specified
     * JNDI context.
     * @param configuration the configuration
     * @param context the JNDI context
     * @param transaction the UserTransaction
     */
    public static void bindUserTransaction(Configuration configuration, Context context, UserTransaction transaction) throws NamingException
    {
        if(configuration.getBindMockUserTransactionToJNDI())
        {
            context.rebind(configuration.getUserTransactionJNDIName(), transaction);
            context.rebind("javax.transaction.UserTransaction", transaction);
            context.rebind("java:comp/UserTransaction", transaction);
        }
    }
}