addresstest.java

JPA入门文章源码

package entity;

import junit.framework.JUnit4TestAdapter;
import org.junit.After;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNull;
import org.junit.Before;
import org.junit.Test;

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

/**
 * @author Antonio Goncalves
 */
public class AddressTest {
    private static String PERSISTENCE_UNIT_NAME = "watermelonPU";

    private EntityManagerFactory emf;
    private EntityManager em;
    private EntityTransaction trans;

    public static junit.framework.Test suite() {
        return new JUnit4TestAdapter(AddressTest.class);
    }

    @Before
    public void init() {
        emf = Persistence.createEntityManagerFactory(PERSISTENCE_UNIT_NAME);
        em = emf.createEntityManager();
        trans = em.getTransaction();
    }

    @After
    public void close() {
        em.close();
        emf.close();
    }

    @Test
    public void createAddress() {
        Address address = new Address("St James St", "London", "SW14", "UK");

        // Creates an address
        trans.begin();
        em.persist(address);
        trans.commit();
        Long id = address.getId();

        // Finds the address by primary key
        address = em.find(Address.class, id);
        assertEquals(address.getCountry(), "UK");

        // Updates the address
        trans.begin();
        address.setCountry("US");
        trans.commit();

        // Finds the address by primary key
        address = em.find(Address.class, id);
        assertEquals(address.getCountry(), "US");

        // Deletes the address
        trans.begin();
        em.remove(address);
        trans.commit();

        assertNull("Address should has been deleted", em.find(Address.class, id));
    }
}