crimeportaltest.java

Jboss s J2EE Sample

package org.jboss.docs.cmp2.crimeportal;

import java.util.Collection;
import java.util.HashSet;
import java.util.Set;
import java.util.Iterator;
import javax.naming.InitialContext;
import javax.ejb.EJBLocalObject;

import junit.framework.Test;
import junit.framework.TestCase;
import junit.framework.TestSuite;
import net.sourceforge.junitejb.EJBTestCase;

public class CrimePortalTest extends EJBTestCase
      implements CrimePortalTestConstants {

   public static Test suite() {
      TestSuite testSuite = new TestSuite("CrimePortalTest");
      testSuite.addTestSuite(CrimePortalTest.class);
      return testSuite;
   }   

   public CrimePortalTest(String name) {
      super(name);
   }

   private OrganizationHome organizationHome;
   private GangsterHome gangsterHome;
   private JobHome jobHome;

   /**
    * Looks up all of the home interfaces and creates the initial data. 
    * Looking up objects in JNDI is expensive, so it should be done once 
    * and cached.
    * @throws Exception if a problem occures while finding the home interfaces,
    * or if an problem occures while createing the initial data
    */
   public void setUp() throws Exception {
      InitialContext jndi = new InitialContext();

      organizationHome = 
            (OrganizationHome) jndi.lookup("crimeportal/Organization"); 

      gangsterHome = (GangsterHome) jndi.lookup("crimeportal/Gangster"); 

      jobHome = (JobHome) jndi.lookup("crimeportal/Job"); 
   }

   /** Test Organization-Gangster relationship */
   public void testOrganization() throws Exception {
      Organization yakuza = organizationHome.findByPrimaryKey("Yakuza");
      Collection gangsters = yakuza.getMemberGangsters();
      assertEquals(3, gangsters.size());
      assertTrue(gangsters.contains(gangsterHome.findByPrimaryKey(YOJIMBO)));
      assertTrue(gangsters.contains(gangsterHome.findByPrimaryKey(TAKESHI)));
      assertTrue(gangsters.contains(gangsterHome.findByPrimaryKey(YURIKO)));
   }

   /** Test find bad dudes query */
   public void testFindBadDudes() throws Exception {
      Collection gangsters = gangsterHome.findBadDudes(5);
      assertEquals(5, gangsters.size());

      assertTrue(gangsters.contains(
            gangsterHome.findByPrimaryKey(TAKESHI)));
      assertTrue(gangsters.contains(
            gangsterHome.findByPrimaryKey(CHOW)));
      assertTrue(gangsters.contains(
            gangsterHome.findByPrimaryKey(SHOGI)));
      assertTrue(gangsters.contains(
            gangsterHome.findByPrimaryKey(YOJIMBO)));
      assertTrue(gangsters.contains(
            gangsterHome.findByPrimaryKey(CORLEONE)));
   }

   /** Test select boss query */
   public void testSelectBoss() throws Exception {
      Set gangsters = gangsterHome.selectBoss(" Yojimbo ");
      assertEquals(1, gangsters.size());
      assertTrue(gangsters.contains(gangsterHome.findByPrimaryKey(TAKESHI)));

      gangsters = gangsterHome.selectBoss(" Takeshi ");
      assertEquals(1, gangsters.size());
      assertTrue(gangsters.contains(gangsterHome.findByPrimaryKey(TAKESHI)));

      gangsters = gangsterHome.selectBoss("non-existant");
      assertEquals(0, gangsters.size());
   }
}