testany.java

hibernate quickly 一书的各章节源代码

import com.manning.hq.apdxA.BarterPayment;
import com.manning.hq.apdxA.CreditCardPayment;
import com.manning.hq.apdxA.HibernateFactory;
import com.manning.hq.apdxA.Registration;
import junit.framework.TestCase;
import org.hibernate.Session;

import java.math.BigDecimal;

/**
 * 
 */
public class TestAny extends TestCase {
    private Session session1;
    private Session session2;
    protected void setUp() throws Exception {
        HibernateFactory.buildSessionFactory();
        session1 = HibernateFactory.openSession();
        session2 = HibernateFactory.openSession();
    }

    protected void tearDown() throws Exception {
        super.tearDown();
        Session session = HibernateFactory.openSession();
        session.createQuery("delete Registration").executeUpdate();
        session.createQuery("delete CreditCardPayment").executeUpdate();
        session.createQuery("delete BarterPayment").executeUpdate();
        session.flush();
        HibernateFactory.close(session);

        HibernateFactory.close(session1);
        HibernateFactory.close(session2);
    }

    
    /**
     *
     */
    public void testCreditCardPayment() throws Exception {
        Registration r = new Registration();
        CreditCardPayment cc = new CreditCardPayment();
        r.setPaymentDetails(cc);
        cc.setAmount(new BigDecimal(34.50));
        cc.setCurrencyCode("US");

        session1.save(cc);
        session1.save(r);
        session1.flush();

        Registration reg = (Registration) session2.load(Registration.class, r.getId());
        assertEquals("", r.getId(), reg.getId());
        assertEquals("", r.getPaymentDetails().getId(), cc.getId());
        CreditCardPayment cc2 = (CreditCardPayment) r.getPaymentDetails();
        assertEquals("", "US", cc2.getCurrencyCode());

    }

    public void testBarterPayment() throws Exception {
        Registration r = new Registration();
        BarterPayment bp = new BarterPayment();
        r.setPaymentDetails(bp);
        bp.setNumberOfCows(5);
        bp.setNumberOfSheep(10);

        session1.save(bp);
        session1.save(r);
        session1.flush();

        Registration reg = (Registration) session2.load(Registration.class, r.getId());
        assertEquals("", r.getId(), reg.getId());
        assertEquals("", r.getPaymentDetails().getId(), bp.getId());

    }
}