persontest.java

Java EE 5 许多新功能都包含经过修补的 EJB 架构

/*
 * Copyright 2006 Borys Burnayev
 * 
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 
 * http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package test.persistence;

import java.util.HashMap;
import java.util.List;
import java.util.Map;

import javax.persistence.EntityManager;
import javax.persistence.EntityManagerFactory;
import javax.persistence.Persistence;

import junit.framework.TestCase;

import com.rdacorp.petstore.domain.Category;
import com.rdacorp.petstore.domain.Language;
import com.rdacorp.petstore.domain.Person;

/**
 * @author Borys Burnayev
 */
public class PersonTest extends TestCase {

   public void testCRUD() throws Exception {
      Map config = new HashMap();
      EntityManagerFactory emf = Persistence.createEntityManagerFactory("manager1", config);
      EntityManager em = emf.createEntityManager();

      em.getTransaction().begin();
      // create a person record
      List categories = em.createQuery("from Category").getResultList();
      Category initialCategory = (Category) categories.get(0);
      Category updatedCategory = (Category) categories.get(1);
      String initialCategoryCode = initialCategory.getCode();
      String updatedCategoryCode = updatedCategory.getCode();
      List languages = em.createQuery("from Language").getResultList();
      Language initialLanguage = (Language) languages.get(0);
      Language updatedLanguage = (Language) languages.get(1);
      String initialLanguageCode = initialLanguage.getCode();
      String updatedLanguageCode = updatedLanguage.getCode();
      Person person = new Person("FirstName", "LastName", "[e-mail]", "1112223333", new Integer(1), new Integer(
               1), initialCategory, initialLanguage);
      em.persist(person);
      em.getTransaction().commit();

      // verify primary key
      Integer personId = person.getPersonId();
      assertNotNull(personId);

      em.getTransaction().begin();
      // read the person
      person = (Person) em.find(Person.class, personId);
      // verify item
      assertEquals("FirstName", person.getFirstName());
      assertEquals("LastName", person.getLastName());
      assertEquals("[e-mail]", person.getEmailAddress());
      assertEquals("1112223333", person.getPhoneNumber());
      assertEquals(new Integer(1), person.getEnableMylist());
      assertEquals(new Integer(1), person.getEnableTips());
      assertEquals(initialCategoryCode, person.getFavoriteCategory().getCode());
      assertEquals(initialLanguageCode, person.getPreferredLanguage().getCode());

      // update the record
      person.setFirstName("NewFirstName");
      person.setLastName("NewLastName");
      person.setEmailAddress("[e-mail]");
      person.setPhoneNumber("3332221111");
      person.setEnableMylist(new Integer(0));
      person.setEnableTips(new Integer(0));
      person.setFavoriteCategory(em.merge(updatedCategory));
      person.setPreferredLanguage(em.merge(updatedLanguage));
      em.persist(person);
      em.getTransaction().commit();

      em.getTransaction().begin();
      // reread the person
      person = (Person) em.find(Person.class, personId);
      // verify item
      assertEquals("NewFirstName", person.getFirstName());
      assertEquals("NewLastName", person.getLastName());
      assertEquals("[e-mail]", person.getEmailAddress());
      assertEquals("3332221111", person.getPhoneNumber());
      assertEquals(new Integer(0), person.getEnableMylist());
      assertEquals(new Integer(0), person.getEnableTips());
      assertEquals(updatedCategoryCode, person.getFavoriteCategory().getCode());
      assertEquals(updatedLanguageCode, person.getPreferredLanguage().getCode());

      // delete the item
      em.remove(person);
      em.getTransaction().commit();

      // retrive the person and verify it is null
      person = (Person) em.find(Person.class, personId);
      assertNull(person);

      em.close();
      emf.close();
   }
}