package com.j256.ormlite.misc;
import static org.easymock.EasyMock.createMock;
import static org.easymock.EasyMock.expect;
import static org.easymock.EasyMock.isA;
import static org.easymock.EasyMock.replay;
import static org.easymock.EasyMock.verify;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertNull;
import static org.junit.Assert.fail;
import java.sql.SQLException;
import java.sql.Savepoint;
import java.util.concurrent.Callable;
import org.junit.Test;
import com.j256.ormlite.BaseCoreTest;
import com.j256.ormlite.dao.Dao;
import com.j256.ormlite.support.ConnectionSource;
import com.j256.ormlite.support.DatabaseConnection;
public class TransactionManagerTest extends BaseCoreTest {
@Test
public void testTransactionManager() throws Exception {
ConnectionSource connectionSource = createMock(ConnectionSource.class);
DatabaseConnection conn = createMock(DatabaseConnection.class);
expect(conn.isAutoCommitSupported()).andReturn(false);
Savepoint savePoint = createMock(Savepoint.class);
expect(savePoint.getSavepointName()).andReturn("name").anyTimes();
expect(conn.setSavePoint(isA(String.class))).andReturn(savePoint);
conn.commit(savePoint);
expect(connectionSource.getDatabaseType()).andReturn(databaseType);
expect(connectionSource.getReadWriteConnection()).andReturn(conn);
expect(connectionSource.saveSpecialConnection(conn)).andReturn(true);
connectionSource.clearSpecialConnection(conn);
connectionSource.releaseConnection(conn);
replay(connectionSource, conn, savePoint);
TransactionManager tm = new TransactionManager(connectionSource);
tm.callInTransaction(new Callable<Void>() {
public Void call() throws Exception {
return null;
}
});
verify(connectionSource, conn, savePoint);
}
@Test
public void testTransactionManagerSavePointNull() throws Exception {
ConnectionSource connectionSource = createMock(ConnectionSource.class);
DatabaseConnection conn = createMock(DatabaseConnection.class);
expect(conn.isAutoCommitSupported()).andReturn(false);
expect(conn.setSavePoint(isA(String.class))).andReturn(null);
conn.commit(null);
expect(connectionSource.getDatabaseType()).andReturn(databaseType);
expect(connectionSource.getReadWriteConnection()).andReturn(conn);
expect(connectionSource.saveSpecialConnection(conn)).andReturn(true);
connectionSource.clearSpecialConnection(conn);
connectionSource.releaseConnection(conn);
replay(connectionSource, conn);
TransactionManager tm = new TransactionManager(connectionSource);
tm.callInTransaction(new Callable<Void>() {
public Void call() throws Exception {
return null;
}
});
verify(connectionSource, conn);
}
@Test
public void testTransactionManagerRollback() throws Exception {
ConnectionSource connectionSource = createMock(ConnectionSource.class);
DatabaseConnection conn = createMock(DatabaseConnection.class);
expect(conn.isAutoCommitSupported()).andReturn(false);
Savepoint savePoint = createMock(Savepoint.class);
expect(savePoint.getSavepointName()).andReturn("name").anyTimes();
expect(conn.setSavePoint(isA(String.class))).andReturn(savePoint);
conn.rollback(savePoint);
expect(connectionSource.getDatabaseType()).andReturn(databaseType);
expect(connectionSource.getReadWriteConnection()).andReturn(conn);
expect(connectionSource.saveSpecialConnection(conn)).andReturn(true);
connectionSource.clearSpecialConnection(conn);
connectionSource.releaseConnection(conn);
replay(connectionSource, conn, savePoint);
TransactionManager tm = new TransactionManager(connectionSource);
try {
tm.callInTransaction(new Callable<Void>() {
public Void call() throws Exception {
throw new SQLException("you better roll back!!");
}
});
fail("expected an exception");
} catch (SQLException e) {
// expected
}
verify(connectionSource, conn, savePoint);
}
@Test
public void testTransactionManagerRollbackNullSavePoint() throws Exception {
ConnectionSource connectionSource = createMock(ConnectionSource.class);
DatabaseConnection conn = createMock(DatabaseConnection.class);
expect(conn.isAutoCommitSupported()).andReturn(false);
expect(conn.setSavePoint(isA(String.class))).andReturn(null);
conn.rollback(null);
expect(connectionSource.getDatabaseType()).andReturn(databaseType);
expect(connectionSource.getReadWriteConnection()).andReturn(conn);
expect(connectionSource.saveSpecialConnection(conn)).andReturn(true);
connectionSource.clearSpecialConnection(conn);
connectionSource.releaseConnection(conn);
replay(connectionSource, conn);
TransactionManager tm = new TransactionManager(connectionSource);
try {
tm.callInTransaction(new Callable<Void>() {
public Void call() throws Exception {
throw new SQLException("you better roll back!!");
}
});
fail("expected an exception");
} catch (SQLException e) {
// expected
}
verify(connectionSource, conn);
}
@Test
public void testTransactionManagerRollbackOtherException() throws Exception {
ConnectionSource connectionSource = createMock(ConnectionSource.class);
DatabaseConnection conn = createMock(DatabaseConnection.class);
expect(conn.isAutoCommitSupported()).andReturn(false);
Savepoint savePoint = createMock(Savepoint.class);
expect(savePoint.getSavepointName()).andReturn("name").anyTimes();
expect(conn.setSavePoint(isA(String.class))).andReturn(savePoint);
conn.rollback(savePoint);
expect(connectionSource.getDatabaseType()).andReturn(databaseType);
expect(connectionSource.getReadWriteConnection()).andReturn(conn);
expect(connectionSource.saveSpecialConnection(conn)).andReturn(true);
connectionSource.clearSpecialConnection(conn);
connectionSource.releaseConnection(conn);
replay(connectionSource, conn, savePoint);
TransactionManager tm = new TransactionManager(connectionSource);
try {
tm.callInTransaction(new Callable<Void>() {
public Void call() throws Exception {
throw new Exception("you better roll back!!");
}
});
fail("expected an exception");
} catch (Exception e) {
// expected
}
verify(connectionSource, conn, savePoint);
}
@Test
public void testTransactionManagerAutoCommitSupported() throws Exception {
ConnectionSource connectionSource = createMock(ConnectionSource.class);
DatabaseConnection conn = createMock(DatabaseConnection.class);
expect(conn.isAutoCommitSupported()).andReturn(true);
expect(conn.getAutoCommit()).andReturn(false);
Savepoint savePoint = createMock(Savepoint.class);
expect(savePoint.getSavepointName()).andReturn("name").anyTimes();
expect(conn.setSavePoint(isA(String.class))).andReturn(savePoint);
conn.commit(savePoint);
expect(connectionSource.getDatabaseType()).andReturn(databaseType);
expect(connectionSource.getReadWriteConnection()).andReturn(conn);
expect(connectionSource.saveSpecialConnection(conn)).andReturn(true);
connectionSource.clearSpecialConnection(conn);
connectionSource.releaseConnection(conn);
replay(connectionSource, conn, savePoint);
TransactionManager tm = new TransactionManager(connectionSource);
tm.callInTransaction(new Callable<Void>() {
public Void call() throws Exception {
return null;
}
});
verify(connectionSource, conn, savePoint);
}
@Test
public void testTransactionManagerAutoCommitOn() throws Exception {
ConnectionSource connectionSource = createMock(ConnectionSource.class);
DatabaseConnection conn = createMock(DatabaseConnection.class);
expect(conn.isAutoCommitSupported()).andReturn(true);
expect(conn.getAutoCommit()).andReturn(true);
conn.setAutoCommit(false);
Savepoint savePoint = createMock(Savepoint.class);
expect(savePoint.getSavepointName()).andReturn("name").anyTimes();
expect(conn.setSavePoint(isA(String.class))).andReturn(savePoint);
conn.commit(savePoint);
conn.setAutoCommit(true);
expect(connectionSource.getDatabaseType()).andReturn(databaseType);
expect(connectionSource.getReadWriteConnection()).andReturn(conn);
expect(connectionSource.saveSpecialConnection(conn)).andReturn(true);
connectionSource.clearSpecialConnection(conn);
connectionSource.releaseConnection(conn);
replay(connectionSource, conn, savePoint);
TransactionManager tm = new TransactionManager(connectionSource);
tm.callInTransaction(new Callable<Void>() {
public Void call() throws Exception {
return null;
}
});
verify(connectionSource, conn, savePoint);
}
@Test
public void testTransactionManagerSpringWiring() {
TransactionManager tm = new TransactionManager();
tm.setConnectionSource(connectionSource);
tm.initialize();
}
@Test(expected = IllegalStateException.class)
public void testTransactionManagerNoSet() {
TransactionManager tm = new TransactionManager();
tm.initialize();
}
@Test
public void testDaoTransactionManagerCommitted() throws Exception {
if (connectionSource == null) {
return;
}
TransactionManager mgr = new TransactionManager(connectionSource);
final Dao<Foo, String> fooDao = createDao(Foo.class, true);
testTransactionManager(mgr, null, fooDao);
}
@Test
public void testRollBack() throws Exception {
if (connectionSource == null) {
return;
}
TransactionManager mgr = new TransactionManager(connectionSource);
final Dao<Foo, String> fooDao = createDao(Foo.class, true);
testTransactionManager(mgr, new RuntimeException("What!! I protest!!"), fooDao);
}
@Test
public void testSpringWiredRollBack() throws Exception {
if (connectionSource == null) {
return;
}
TransactionManager mgr = new TransactionManager();
mgr.setConnectionSource(connectionSource);
mgr.initialize();
final Dao<Foo, String> fooDao = createDao(Foo.class, true);
testTransactionManager(mgr, new RuntimeException("What!! I protest!!"), fooDao);
}
@Test
public void testNonRuntimeExceptionWiredRollBack() throws Exception {
if (connectionSource == null) {
return;
}
TransactionManager mgr = new TransactionManager();
mgr.setConnectionSource(connectionSource);
mgr.initialize();
final Dao<Foo, String> fooDao = createDao(Foo.class, true);
testTransactionManager(mgr, new Exception("What!! I protest via an Exception!!"), fooDao);
}
@Test
public void testTransactionWithinTransaction() throws Exception {
if (connectionSource == null) {
return;
}
final TransactionManager mgr = new TransactionManager(connectionSource);
final Dao<Foo, String> fooDao = createDao(Foo.class, true);
mgr.callInTransaction(new Callable<Void>() {
public Void call() throws Exception {
testTransactionManager(mgr, null, fooDao);
return null;
}
});
}
private void testTransactionManager(TransactionManager mgr, final Exception exception, final Dao<Foo, String> fooDao)
throws Exception {
final Foo foo1 = new Foo();
int val = 13131511;
foo1.id = "id string";
foo1.val = val;
assertEquals(1, fooDao.create(foo1));
try {
final int ret = 13431231;
int returned = mgr.callInTransaction(new Callable<Integer>() {
public Integer call() throws Exception {
// we delete it inside a transaction
assertEquals(1, fooDao.delete(foo1));
// we can't find it
assertNull(fooDao.queryForId(foo1.id));
if (exception != null) {
// but then we throw an exception which rolls back the transaction
throw exception;
} else {
return ret;
}
}
});
if (exception == null) {
assertEquals(ret, returned);
} else {
fail("Should have thrown");
}
} catch (SQLException e) {
if (exception == null) {
throw e;
} else {
// expected
}
}
if (exception == null) {
// still doesn't find it after we delete it
assertNull(fooDao.queryForId(foo1.id));
} else {
// still finds it after we delete it
Foo foo2 = fooDao.queryForId(foo1.id);
assertNotNull(foo2);
assertEquals(val, foo2.val);
}
}
}