audittest.java

webwork in action 下载。

package org.hibernate.auction.test;

import junit.framework.*;
import junit.textui.TestRunner;
import net.sf.hibernate.*;
import org.hibernate.auction.dao.*;
import org.hibernate.auction.model.*;
import org.hibernate.auction.persistence.HibernateUtil;
import org.hibernate.auction.persistence.audit.*;

import java.math.BigDecimal;
import java.util.*;

public class AuditTest extends TestCase {

	// ********************************************************** //

	public void testAuditLog() throws Exception {

		// Save a user without audit logging
		UserDAO userDAO = new HibernateUserDAO();
		User u1 = new User("Christian", "Bauer", "turin", "abc123", new Email("christian", "hibernate.org"));
		userDAO.makePersistent(u1);
		HibernateUtil.commitTransaction();
		HibernateUtil.closeSession();

		// Enable interceptor
		AuditLogInterceptor interceptor = new AuditLogInterceptor();
		HibernateUtil.registerInterceptor(interceptor);
		interceptor.setSession(HibernateUtil.getSession());
		interceptor.setUserId(u1.getId());

		// Save an item with audit logging enabled
		Item item = new Item("ONE", "Foo",
		        u1,
		        new MonetaryAmount(new BigDecimal("1.99"), Currency.getInstance(Locale.US)),
		        new MonetaryAmount(new BigDecimal("50.33"), Currency.getInstance(Locale.US)),
		        new Date(), new Date());
		ItemDAO itemDAO = new ItemDAO();
		itemDAO.makePersistent(item);

		// Synchronize state to trigger interceptor
		HibernateUtil.getSession().flush();

		// Check audit log
		Query queryAuditOne = HibernateUtil.getSession().createQuery("from AuditLogRecord lr where lr.entityId = :id");
		queryAuditOne.setParameter("id", item.getId());
		AuditLogRecord logRecordOne = (AuditLogRecord)queryAuditOne.uniqueResult();
		assertEquals(logRecordOne.userId, u1.getId());

		HibernateUtil.commitTransaction();
		HibernateUtil.closeSession();

		// Deregister interceptor
		HibernateUtil.registerInterceptor(null);
	}

	// ********************************************************** //

	public AuditTest(String x) {
		super(x);
	}

	public static Test suite() {
		return new TestSuite(AuditTest.class);
	}

	public static void main(String[] args) throws Exception {
		TestRunner.run( suite() );
	}

}