@Override
public void lookupMessageDrivenContext() throws TestFailureException {
    try {
        try {
            final InitialContext ctx = new InitialContext();
            Assert.assertNotNull("The InitialContext is null", ctx);

            // lookup in enc
            final MessageDrivenContext sctx = (MessageDrivenContext) ctx.lookup("java:comp/env/mdbcontext");
            Assert.assertNotNull("The MessageDrivenContext got from java:comp/env/mdbcontext is null", sctx);

            // lookup using global name
            final EJBContext ejbCtx = (EJBContext) ctx.lookup("java:comp/EJBContext");
            Assert.assertNotNull("The MessageDrivenContext got from java:comp/EJBContext is null ", ejbCtx);

            // verify context was set via legacy set method
            Assert.assertNotNull("The MessageDrivenContext is null from setter method", mdbContext);
        } catch (final Exception e) {
            Assert.fail("Received Exception " + e.getClass() + " : " + e.getMessage());
        }
    } catch (final AssertionFailedError afe) {
        throw new TestFailureException(afe);
    }

}
 

public void lookupMessageDrivenContext() throws TestFailureException {
    try {
        try {
            final InitialContext ctx = new InitialContext();
            Assert.assertNotNull("The InitialContext is null", ctx);

            // lookup in enc
            final MessageDrivenContext sctx = (MessageDrivenContext) ctx.lookup("java:comp/env/mdbcontext");
            Assert.assertNotNull("The MessageDrivenContext got from java:comp/env/mdbcontext is null", sctx);

            // lookup using global name
            final EJBContext ejbCtx = (EJBContext) ctx.lookup("java:comp/EJBContext");
            Assert.assertNotNull("The MessageDrivenContext got from java:comp/EJBContext is null ", ejbCtx);
        } catch (final Exception e) {
            Assert.fail("Received Exception " + e.getClass() + " : " + e.getMessage());
        }
    } catch (final AssertionFailedError afe) {
        throw new TestFailureException(afe);
    }

}
 

@Override
public void lookupMessageDrivenContext() throws TestFailureException {
    try {
        try {
            final InitialContext ctx = new InitialContext();
            Assert.assertNotNull("The InitialContext is null", ctx);

            // lookup in enc
            final MessageDrivenContext messageDrivenContext = (MessageDrivenContext) ctx.lookup("java:comp/env/mdbcontext");
            Assert.assertNotNull("The SessionContext got from java:comp/env/mdbcontext is null", messageDrivenContext);

            // lookup using global name
            final EJBContext ejbCtx = (EJBContext) ctx.lookup("java:comp/EJBContext");
            Assert.assertNotNull("The SessionContext got from java:comp/EJBContext is null ", ejbCtx);

            // verify context was set via legacy set method
            Assert.assertNotNull("The MdbContext is null from setter method", mdbContext);
        } catch (final Exception e) {
            Assert.fail("Received Exception " + e.getClass() + " : " + e.getMessage());
        }
    } catch (final AssertionFailedError afe) {
        throw new TestFailureException(afe);
    }

}
 

@Override
public void setMessageDrivenContext(final MessageDrivenContext ctx) throws EJBException {
    this.mdbContext = ctx;
    try {
        final ConnectionFactory connectionFactory = (ConnectionFactory) new InitialContext().lookup("java:comp/env/jms");
        mdbInvoker = new MdbInvoker(connectionFactory, this);
    } catch (final Exception e) {
        throw new EJBException(e);
    }
}
 

@Override
public void setMessageDrivenContext(final MessageDrivenContext ctx) throws EJBException {
    this.mdbContext = ctx;
    testAllowedOperations("setMessageDrivenContext");
    try {
        final ConnectionFactory connectionFactory = (ConnectionFactory) new InitialContext().lookup("java:comp/env/jms");
        mdbInvoker = new MdbInvoker(connectionFactory, this);
    } catch (final Exception e) {
        throw new EJBException(e);
    }
}
 

@Override
public void setMessageDrivenContext(final MessageDrivenContext ctx) throws EJBException {
    this.mdbContext = ctx;
    try {
        mdbInvoker = new MdbInvoker(connectionFactory, this);
    } catch (final Exception e) {
        throw new EJBException(e);
    }
}
 

@Override
public void setMessageDrivenContext(final MessageDrivenContext ctx) throws EJBException {
    this.mdbContext = ctx;
    try {
        final ConnectionFactory connectionFactory = (ConnectionFactory) new InitialContext().lookup("java:comp/env/jms");
        mdbInvoker = new MdbInvoker(connectionFactory, this);
    } catch (final Exception e) {
        throw new EJBException(e);
    }
}
 

@Override
public void setMessageDrivenContext(final MessageDrivenContext ctx) throws EJBException {
    this.mdbContext = ctx;
    try {
        final ConnectionFactory connectionFactory = (ConnectionFactory) new InitialContext().lookup("java:comp/env/jms");
        mdbInvoker = new MdbInvoker(connectionFactory, this);
    } catch (final Exception e) {
        throw new EJBException(e);
    }
}
 

@Override
public void setMessageDrivenContext(final MessageDrivenContext ctx) throws EJBException {
    this.mdbContext = ctx;
    try {
        final ConnectionFactory connectionFactory = (ConnectionFactory) new InitialContext().lookup("java:comp/env/jms");
        mdbInvoker = new MdbInvoker(connectionFactory, this);
    } catch (final Exception e) {
        throw new EJBException(e);
    }
}
 

private <R extends JndiReference> boolean isPrivateReference(final R a) {
    if (!isResourceRef(a)) {
        return false;
    }

    final Class[] types = {EJBContext.class, EntityContext.class, SessionContext.class, MessageDrivenContext.class, UserTransaction.class};

    for (final Class type : types) {
        if (type.getName().equals(a.getType())) {
            return true;
        }
    }

    return false;
}
 

/**
 * @see javax.ejb.MessageDrivenBean#setMessageDrivenContext(javax.ejb.MessageDrivenContext)
 */
public void setMessageDrivenContext(MessageDrivenContext aContext) {
    context = aContext;
}
 

/**
 * @see javax.ejb.MessageDrivenBean#setMessageDrivenContext(javax.ejb.MessageDrivenContext)
 */
public void setMessageDrivenContext(MessageDrivenContext aContext) {
    context = aContext;
}
 

public void setMessageDrivenContext(MessageDrivenContext mdc) {
}
 

public MessageDrivenContext getMessageDrivenContext() {
    return ejbContext;
}
 

public void setMessageDrivenContext(MessageDrivenContext ctx) throws EJBException {
    ejbContext = ctx;
}
 

@Override
public void setMessageDrivenContext(MessageDrivenContext ejbContext) throws EJBException {
       log.info("Received EJB-MDB Context");
       this.ejbContext = ejbContext;
   }
 

public MessageDrivenContext getEjbContext() {
    return ejbContextField;
}
 

public void setEjbContext(final MessageDrivenContext ejbContext) {
    this.ejbContextField = ejbContext;
}
 

@Resource
public void setMessageDrivenContext(final MessageDrivenContext messageDrivenContext) {
    lifecycle.push(Lifecycle.INJECTION);
}
 

@Resource
public void setMessageDrivenContext(final MessageDrivenContext messageDrivenContext) {
    lifecycle.push(Lifecycle.INJECTION);
}
 

@Resource
public void setMessageDrivenContext(final MessageDrivenContext messageDrivenContext) {
    lifecycle.push(Lifecycle.INJECTION);
}
 

@Resource
public void setMessageDrivenContext(final MessageDrivenContext messageDrivenContext) {
    lifecycle.push(Lifecycle.INJECTION);
}
 

public void setMessageDrivenContext(javax.ejb.MessageDrivenContext context)
{
}