TermDefinitionAccessorBeanTest.java from Texai at Krugle

Show TermDefinitionAccessorBeanTest.java syntax highlighted

/*
 * AssociationFinderBeanTest.java
 * JUnit based test
 *
 * Created on November 1, 2006, 10:11 PM
 *
 * Copyright (C) 2006 Stephen L. Reed.
 *
 * This program is free software; you can redistribute it and/or modify it under the terms
 * of the GNU General Public License as published by the Free Software Foundation; either
 * version 2 of the License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
 * without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
 * See the GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License along with this program;
 * if not, write to the Free Software Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
 */

package org.texai.kb.ejb.session;

import java.util.HashSet;
import junit.framework.*;
import java.util.Set;
import javax.persistence.EntityManager;
import javax.persistence.EntityManagerFactory;
import javax.persistence.EntityTransaction;
import javax.persistence.Persistence;
import net.sf.ehcache.CacheManager;
import org.texai.kb.CacheInitializer;
import org.texai.kb.ejb.entity.AbstractReifiedTerm;
import org.texai.kb.ejb.entity.AbstractTerm;
import org.texai.kb.ejb.session.shared.AssociationFinderBean;
import org.texai.kb.ejb.session.shared.TermDefinitionAccessorBean;
import org.texai.kb.ejb.session.shared.TermFinderFacadeBean;

/**
 *
 * @author reed
 */
public class TermDefinitionAccessorBeanTest extends TestCase {
  
  /** the entity manager factory */
  private static EntityManagerFactory entityManagerFactory;
  
  /** the entity manager */
  private static EntityManager entityManager;
  
  public TermDefinitionAccessorBeanTest(String testName) {
    super(testName);
  }

  protected void setUp() throws Exception {
    if (entityManager == null) {
      System.out.println("oneTimeSetup");
      try {
        getClass().getClassLoader().setDefaultAssertionStatus(true);
        CacheInitializer.initializeCaches();
        System.out.println("setUp: creating EntityManager");
        entityManagerFactory = Persistence.createEntityManagerFactory("TestPersistenceUnit");
        entityManager = entityManagerFactory.createEntityManager();
        System.out.println("entityManager created");
     } catch (Exception e) {
        e.printStackTrace();
        fail(e.getMessage());
      }
    }
    if (entityManager.getTransaction().isActive()) {
      entityManager.getTransaction().rollback();
    }
    try {
    } catch (Exception e) {
      e.printStackTrace();
      fail(e.getMessage());
    }
  }

  protected void tearDown() throws Exception {
  }


  /**
   * Test of gatherTypesForTerm method, of class org.texai.kb.ejb.session.AssociationFinderBean.
   */
  public void testGatherTypes() {
    System.out.println("gatherTypes");
    
    EntityTransaction entityTransaction = entityManager.getTransaction();
    entityTransaction.begin();
    assertTrue(entityTransaction.isActive());

    TermFinderFacadeBean termFinderFacade = new TermFinderFacadeBean();
    termFinderFacade.setEntityManager(entityManager);
    AbstractReifiedTerm term = termFinderFacade.findAtomicTermByTermName("Bicycle");
    assertNotNull(term);
    TermDefinitionAccessorBean instance = new TermDefinitionAccessorBean();
    instance.setTermFinderFacade(termFinderFacade);
    AssociationFinderBean associationFinder = new AssociationFinderBean();
    associationFinder.setEntityManager(entityManager);
    TermFinderFacadeBean termFinderFacade2 = new TermFinderFacadeBean();
    termFinderFacade2.setEntityManager(entityManager);
    associationFinder.setTermFinderFacade(termFinderFacade2);
    instance.setAssociationFinder(associationFinder);
    
    Set<AbstractTerm> expResult = new HashSet<AbstractTerm>();
    expResult.add(termFinderFacade.findAtomicTermByTermName("TransportationDeviceType"));
    expResult.add(termFinderFacade.findAtomicTermByTermName("ManufacturedGoodsType"));
    expResult.add(termFinderFacade.findAtomicTermByTermName("ArtifactTypeByGenericCategory"));
    assertNotNull(termFinderFacade.findAtomicTermByTermName("Transportation-Topic"));
    expResult.add(termFinderFacade.findAtomicTermByTermName("Transportation-Topic"));
    assertNotNull(termFinderFacade.findNonAtomicTermByTermName("ConceptNodesForTaskTypeFn(IntelligenceAnalysisTask-Allotment)"));
    expResult.add(termFinderFacade.findNonAtomicTermByTermName("ConceptNodesForTaskTypeFn(IntelligenceAnalysisTask-Allotment)"));
    expResult.add(termFinderFacade.findAtomicTermByTermName("SpatiallyDisjointObjectType"));
    Set<AbstractReifiedTerm> result = instance.gatherTypesForTerm(term);
    assertEquals(expResult, result);
    
    entityTransaction.commit();
    assertTrue(!entityTransaction.isActive());
    System.out.println("  gatherTypes OK");
  }

  /**
   * Test of gatherAllTypesForTerm method, of class org.texai.kb.ejb.session.AssociationFinderBean.
   */
  public void testGatherAllTypes() {
    System.out.println("gatherAllTypes");
    
    EntityTransaction entityTransaction = entityManager.getTransaction();
    entityTransaction.begin();
    assertTrue(entityTransaction.isActive());

    TermFinderFacadeBean termFinderFacade = new TermFinderFacadeBean();
    termFinderFacade.setEntityManager(entityManager);
    AbstractReifiedTerm term = termFinderFacade.findAtomicTermByTermName("EiffelTower");
    assertNotNull(term);
    TermDefinitionAccessorBean instance = new TermDefinitionAccessorBean();
    instance.setTermFinderFacade(termFinderFacade);
    AssociationFinderBean associationFinder = new AssociationFinderBean();
    associationFinder.setEntityManager(entityManager);
    TermFinderFacadeBean termFinderFacade2 = new TermFinderFacadeBean();
    termFinderFacade2.setEntityManager(entityManager);
    associationFinder.setTermFinderFacade(termFinderFacade2);
    instance.setAssociationFinder(associationFinder);
    
    Set<AbstractTerm> expResult = new HashSet<AbstractTerm>();
    Set<AbstractReifiedTerm> result = instance.gatherAllTypesForTerm(
            term,
            termFinderFacade.findAtomicTermByTermName("WorldGeographyMt"));
    assertTrue(result.toString().contains("HumanScaleObject"));
    assertTrue(result.toString().contains("Landmark-Underspecified"));
    assertTrue(result.toString().contains("Thing"));
    
    entityTransaction.commit();
    assertTrue(!entityTransaction.isActive());
    System.out.println("  gatherAllTypes OK");
  }

  
  /** Performs one time tear down of test harness. This must be the last test method. */
  public void testOneTimeTearDown() {
    System.out.println("oneTimeTearDown");
    CacheManager.getInstance().shutdown();
    entityManager.close();
    entityManagerFactory.close();
  }
}
See more files for this project here