customerpersistencetests.java

一个hibernate的自动测试框架源码

package net.chrisrichardson.ormunit.hibernate.tests;

import java.io.Serializable;

import net.chrisrichardson.ormunit.hibernate.HibernatePersistenceTests;
import net.chrisrichardson.util.TxnCallback;

public class CustomerPersistenceTests extends
		HibernatePersistenceTests<CustomerPersistenceTests> {

	@Override
	protected String[] getConfigLocations() {
		return new String[] { "classpath:appctx/context.xml" };
	}

	private Serializable id;

	public void testUpdateSimpleV1() {
		doWithTransaction(new TxnCallback(){

			public void execute() throws Throwable {
				Customer c = new Customer("John Doe", new Address("1 High Street",
						null, "Oakland", "CA", "94610"));
				id = save(c);
			}});

		doWithTransaction(new TxnCallback(){
			

			public void execute() throws Throwable {
				Customer c2 = get(Customer.class, id);
				assertNotNull(c2);
			}});

	}

	public void testUpdateSimpleV2() {
		Serializable id = txnThis.saveCustomer();
		txnThis.loadCustomer(id);
	}

	Serializable saveCustomer() {
		Customer c = new Customer("John Doe", new Address("1 High Street",
				null, "Oakland", "CA", "94610"));
		return save(c);
	}

	void loadCustomer(Serializable id) {
		Customer c2 = get(Customer.class, id);
		assertNotNull(c2);
	}

	
	public void testUpdate() {
		Customer c = new Customer("John Doe", new Address("1 High Street",
				null, "Oakland", "CA", "94610"));
		int id = (Integer) save(c);

		Customer c2 = get(Customer.class, id);
		assertNotNull(c2);

		txnThis.updateCustomer(id);

		Customer c3 = get(Customer.class, id);
		assertEquals("Mary Smith", c3.getName());
	}

	// This can't be private - very strange things occur if it is

	void updateCustomer(int id) {
		Customer c2 = get(Customer.class, id);
		c2.setName("Mary Smith");
	}

}