toplinktransactionmanagertests.java

spring的源代码

/*
 * Created on Mar 20, 2005
 *
 */

package org.springframework.orm.toplink;

import java.util.ArrayList;
import java.util.List;

import junit.framework.TestCase;
import oracle.toplink.sessions.Session;
import oracle.toplink.sessions.UnitOfWork;
import org.easymock.MockControl;

import org.springframework.jdbc.support.SQLStateSQLExceptionTranslator;
import org.springframework.transaction.PlatformTransactionManager;
import org.springframework.transaction.TransactionDefinition;
import org.springframework.transaction.TransactionStatus;
import org.springframework.transaction.UnexpectedRollbackException;
import org.springframework.transaction.support.TransactionCallback;
import org.springframework.transaction.support.TransactionSynchronizationManager;
import org.springframework.transaction.support.TransactionTemplate;

/**
 * @author Juergen Hoeller
 * @author <a href="mailto:[e-mail]">James Clark</a>
 * @since 28.04.2005
 */
public class TopLinkTransactionManagerTests extends TestCase {

	public void testTransactionCommit() {
		MockControl sessionControl = MockControl.createControl(Session.class);
		final Session session = (Session) sessionControl.getMock();
		MockControl uowControl = MockControl.createControl(UnitOfWork.class);
		UnitOfWork uow = (UnitOfWork) uowControl.getMock();

		final SessionFactory sf = new MockSessionFactory(session);

		// during commit, TM must get the active UnitOfWork
		session.getActiveUnitOfWork();
		sessionControl.setReturnValue(uow, 2);
		uow.beginEarlyTransaction();
		uowControl.setVoidCallable(1);
		uow.commit();
		uowControl.setVoidCallable();
		// session should be released when it was bound explicitly by the TM
		session.release();
		sessionControl.setVoidCallable();

		sessionControl.replay();
		uowControl.replay();

		TopLinkTransactionManager tm = new TopLinkTransactionManager();
		tm.setJdbcExceptionTranslator(new SQLStateSQLExceptionTranslator());
		tm.setSessionFactory(sf);
		TransactionTemplate tt = new TransactionTemplate(tm);
		tt.setIsolationLevel(TransactionDefinition.ISOLATION_SERIALIZABLE);
		tt.setTimeout(10);
		assertTrue("Hasn't thread session", !TransactionSynchronizationManager.hasResource(sf));
		assertTrue("JTA synchronizations not active", !TransactionSynchronizationManager.isSynchronizationActive());

		Object result = tt.execute(new TransactionCallback() {
			public Object doInTransaction(TransactionStatus status) {
				assertTrue("Has thread session", TransactionSynchronizationManager.hasResource(sf));
				TopLinkTemplate template = new TopLinkTemplate(sf);
				return template.execute(new TopLinkCallback() {
					public Object doInTopLink(Session session) {
						return null;
					}
				});
			}
		});
		assertTrue("Hasn't thread session", !TransactionSynchronizationManager.hasResource(sf));
		assertTrue("JTA synchronizations not active", !TransactionSynchronizationManager.isSynchronizationActive());
		sessionControl.verify();
		uowControl.verify();
	}

	public void testTransactionRollback() {
		MockControl sessionControl = MockControl.createControl(Session.class);
		final Session session = (Session) sessionControl.getMock();
		MockControl uowControl = MockControl.createControl(UnitOfWork.class);
		UnitOfWork uow = (UnitOfWork) uowControl.getMock();

		final SessionFactory sf = new MockSessionFactory(session);

		session.getActiveUnitOfWork();
		sessionControl.setReturnValue(uow, 1);
		uow.beginEarlyTransaction();
		uowControl.setVoidCallable(1);
		session.release();
		sessionControl.setVoidCallable(1);

		sessionControl.replay();
		uowControl.replay();

		TopLinkTransactionManager tm = new TopLinkTransactionManager();
		tm.setSessionFactory(sf);
		tm.setJdbcExceptionTranslator(new SQLStateSQLExceptionTranslator());
		TransactionTemplate tt = new TransactionTemplate(tm);
		tt.setIsolationLevel(TransactionDefinition.ISOLATION_SERIALIZABLE);
		tt.setTimeout(10);
		assertTrue("Hasn't thread session", !TransactionSynchronizationManager.hasResource(sf));
		assertTrue("JTA synchronizations not active", !TransactionSynchronizationManager.isSynchronizationActive());

		try {
			Object result = tt.execute(new TransactionCallback() {
				public Object doInTransaction(TransactionStatus status) {
					assertTrue("Has thread session", TransactionSynchronizationManager.hasResource(sf));
					TopLinkTemplate template = new TopLinkTemplate(sf);
					return template.execute(new TopLinkCallback() {
						public Object doInTopLink(Session session) {
							throw new RuntimeException("failure");
						}
					});
				}
			});
			fail("Should have propagated RuntimeException");
		}
		catch (RuntimeException ex) {
			assertTrue(ex.getMessage().equals("failure"));
		}
		assertTrue("Hasn't thread session", !TransactionSynchronizationManager.hasResource(sf));
		assertTrue("JTA synchronizations not active", !TransactionSynchronizationManager.isSynchronizationActive());

		sessionControl.verify();
		uowControl.verify();
	}

	public void testTransactionRollbackOnly() {
		MockControl sessionControl = MockControl.createControl(Session.class);
		final Session session = (Session) sessionControl.getMock();

		final SessionFactory sf = new MockSessionFactory(session);
		session.release();
		sessionControl.setVoidCallable();
		sessionControl.replay();

		TopLinkTransactionManager tm = new TopLinkTransactionManager();
		tm.setSessionFactory(sf);
		tm.setLazyDatabaseTransaction(true);
		tm.setJdbcExceptionTranslator(new SQLStateSQLExceptionTranslator());
		TransactionTemplate tt = new TransactionTemplate(tm);
		tt.setIsolationLevel(TransactionDefinition.ISOLATION_SERIALIZABLE);
		tt.setTimeout(10);
		assertTrue("Hasn't thread session", !TransactionSynchronizationManager.hasResource(sf));
		assertTrue("JTA synchronizations not active", !TransactionSynchronizationManager.isSynchronizationActive());

		Object result = tt.execute(new TransactionCallback() {
			public Object doInTransaction(TransactionStatus status) {
				assertTrue("Has thread session",
						TransactionSynchronizationManager.hasResource(sf));
				TopLinkTemplate template = new TopLinkTemplate(sf);
				template.execute(new TopLinkCallback() {
					public Object doInTopLink(Session session) {
						return null;
					}
				});
				status.setRollbackOnly();
				return null;
			}
		});
		assertTrue("Hasn't thread session", !TransactionSynchronizationManager.hasResource(sf));
		assertTrue("JTA synchronizations not active", !TransactionSynchronizationManager.isSynchronizationActive());
		sessionControl.verify();
	}

	public void testParticipatingTransactionWithCommit() {
		MockControl sessionControl = MockControl.createControl(Session.class);
		final Session session = (Session) sessionControl.getMock();
		MockControl uowControl = MockControl.createControl(UnitOfWork.class);
		UnitOfWork uow = (UnitOfWork) uowControl.getMock();

		final SessionFactory sf = new MockSessionFactory(session);

		session.getActiveUnitOfWork();
		sessionControl.setReturnValue(uow, 2);
		uow.beginEarlyTransaction();
		uowControl.setVoidCallable(1);
		uow.commit();
		uowControl.setVoidCallable();
		session.release();
		sessionControl.setVoidCallable();

		sessionControl.replay();
		uowControl.replay();

		PlatformTransactionManager tm = new TopLinkTransactionManager(sf);
		final TransactionTemplate tt = new TransactionTemplate(tm);

		Object result = tt.execute(new TransactionCallback() {
			public Object doInTransaction(TransactionStatus status) {
				return tt.execute(new TransactionCallback() {
					public Object doInTransaction(TransactionStatus status) {
						TopLinkTemplate ht = new TopLinkTemplate(sf);
						return ht.executeFind(new TopLinkCallback() {
							public Object doInTopLink(Session injectedSession) {
								assertTrue(session == injectedSession);
								return null;
							}
						});
					}
				});
			}
		});

		sessionControl.verify();
		uowControl.verify();
	}

	public void testParticipatingTransactionWithRollback() {
		MockControl sessionControl = MockControl.createControl(Session.class);
		final Session session = (Session) sessionControl.getMock();

		final SessionFactory sf = new MockSessionFactory(session);

		session.release();
		sessionControl.setVoidCallable();

		sessionControl.replay();

		TopLinkTransactionManager tm = new TopLinkTransactionManager(sf);
		tm.setLazyDatabaseTransaction(true);
		final TransactionTemplate tt = new TransactionTemplate(tm);
		try {
			tt.execute(new TransactionCallback() {
				public Object doInTransaction(TransactionStatus status) {
					return tt.execute(new TransactionCallback() {
						public Object doInTransaction(TransactionStatus status) {
							TopLinkTemplate ht = new TopLinkTemplate(sf);
							return ht.executeFind(new TopLinkCallback() {
								public Object doInTopLink(Session session) {
									throw new RuntimeException("application exception");
								}
							});
						}
					});
				}
			});
			fail("Should not thrown RuntimeException");
		}
		catch (RuntimeException ex) {
			assertTrue(ex.getMessage().equals("application exception"));
		}
		sessionControl.verify();
	}

	public void testParticipatingTransactionWithRollbackOnly() {