package org.superbiz.ejblookup;

import javax.ejb.EJBException;
import javax.naming.InitialContext;
import javax.naming.NamingException;

//START SNIPPET: code
public class BlueBean implements Friend {

    public String sayHello() {
        return "Blue says, Hello!";
    }

    public String helloFromFriend() {
        try {
            Friend friend = (Friend) new InitialContext().lookup("java:comp/env/myFriend");
            return "My friend " + friend.sayHello();
        } catch (NamingException e) {
            throw new EJBException(e);
        }
    }
}

package org.superbiz.ejblookup;

/**
 * This is an EJB 3 local business interface
 * A local business interface may be annotated with the @Local
 * annotation, but it's optional. A business interface which is
 * not annotated with @Local or @Remote is assumed to be Local
 * if the bean does not implement any other interfaces
 */
//START SNIPPET: code
public interface Friend {

    public String sayHello();

    public String helloFromFriend();
}

package org.superbiz.ejblookup;

import javax.ejb.EJBException;
import javax.naming.InitialContext;
import javax.naming.NamingException;

//START SNIPPET: code
public class RedBean implements Friend {

    public String sayHello() {
        return "Red says, Hello!";
    }

    public String helloFromFriend() {
        try {
            Friend friend = (Friend) new InitialContext().lookup("java:comp/env/myFriend");
            return "My friend " + friend.sayHello();
        } catch (NamingException e) {
            throw new EJBException(e);
        }
    }
}

<ejb-jar xmlns="http://java.sun.com/xml/ns/javaee">

  <!-- Notice this changes the global jndi name -->
  <module-name>wombat</module-name>

  <enterprise-beans>

    <session>
      <ejb-name>BlueBean</ejb-name>
      <business-local>org.superbiz.ejblookup.Friend</business-local>
      <ejb-class>org.superbiz.ejblookup.BlueBean</ejb-class>
      <session-type>Stateless</session-type>
      <transaction-type>Container</transaction-type>
      <ejb-local-ref>
        <ejb-ref-name>myFriend</ejb-ref-name>
        <local>org.superbiz.ejblookup.Friend</local>
        <ejb-link>RedBean</ejb-link>
      </ejb-local-ref>
    </session>

    <session>
      <ejb-name>RedBean</ejb-name>
      <business-local>org.superbiz.ejblookup.Friend</business-local>
      <ejb-class>org.superbiz.ejblookup.RedBean</ejb-class>
      <session-type>Stateless</session-type>
      <transaction-type>Container</transaction-type>
      <ejb-local-ref>
        <ejb-ref-name>myFriend</ejb-ref-name>
        <local>org.superbiz.ejblookup.Friend</local>
        <ejb-link>BlueBean</ejb-link>
      </ejb-local-ref>
    </session>

  </enterprise-beans>
</ejb-jar>

package org.superbiz.ejblookup;

import junit.framework.TestCase;

import javax.ejb.embeddable.EJBContainer;
import javax.naming.Context;

//START SNIPPET: code
public class EjbDependencyTest extends TestCase {

    private Context context;

    protected void setUp() throws Exception {
        context = EJBContainer.createEJBContainer().getContext();
    }

    public void testRed() throws Exception {

        Friend red = (Friend) context.lookup("java:global/wombat/RedBean");

        assertNotNull(red);
        assertEquals("Red says, Hello!", red.sayHello());
        assertEquals("My friend Blue says, Hello!", red.helloFromFriend());
    }

    public void testBlue() throws Exception {

        Friend blue = (Friend) context.lookup("java:global/wombat/BlueBean");

        assertNotNull(blue);
        assertEquals("Blue says, Hello!", blue.sayHello());
        assertEquals("My friend Red says, Hello!", blue.helloFromFriend());
    }
}