X-Git-Url: http://wamblee.org/gitweb/?a=blobdiff_plain;f=test%2Fenterprise%2Fsrc%2Ftest%2Fjava%2Forg%2Fwamblee%2Ftest%2Fpersistence%2FMyEntityExampleTestBase.java;h=0ee5b38a37cf77c655fe6674d9e4b4f23e98954b;hb=a1ab1a2e44c95cc10ad4c4d15420b49e3744d93b;hp=7f19633386378dae98c107d1bbdc687044c73bbe;hpb=59b7e714a1ccf08b603bab171d122a0b41505164;p=utils

diff --git a/test/enterprise/src/test/java/org/wamblee/test/persistence/MyEntityExampleTestBase.java b/test/enterprise/src/test/java/org/wamblee/test/persistence/MyEntityExampleTestBase.java
index 7f196333..0ee5b38a 100644
--- a/test/enterprise/src/test/java/org/wamblee/test/persistence/MyEntityExampleTestBase.java
+++ b/test/enterprise/src/test/java/org/wamblee/test/persistence/MyEntityExampleTestBase.java
@@ -16,18 +16,22 @@
 package org.wamblee.test.persistence;
 
 import static junit.framework.Assert.*;
+import static org.mockito.Mockito.*;
 
 import javax.persistence.EntityManager;
+import javax.transaction.UserTransaction;
+
+import junit.framework.AssertionFailedError;
 
 import org.dbunit.IDatabaseTester;
 import org.dbunit.dataset.ITable;
 import org.junit.After;
 import org.junit.Before;
 import org.junit.Test;
-import org.wamblee.test.persistence.DatabaseUtils;
-import org.wamblee.test.persistence.JpaBuilder;
-import org.wamblee.test.persistence.JpaTester;
 import org.wamblee.test.persistence.JpaBuilder.JpaUnitOfWork;
+import org.wamblee.test.transactions.DefaultUserTransactionFactory;
+import org.wamblee.test.transactions.SimpleTransactionManager;
+import org.wamblee.test.transactions.TransactionResult;
 
 /**
  * This class shows an example of how to test an entity using jpa.
@@ -124,16 +128,17 @@ public class MyEntityExampleTestBase {
         assertEquals(1, dbutils.getTableSize("XYZ_MYENTITY"));
 
     }
-    
+
     @Test
-    public void testEntityPersistenceWithContextualEntityManager() throws Exception {
+    public void testEntityPersistenceWithContextualEntityManager()
+        throws Exception {
 
         // Use the JPA builder to create a transaction scoped entity manager for
         // as and execute the
         // unit of work.
         builder.begin();
 
-        EntityManager em = builder.getContextualEntityManager(); 
+        EntityManager em = builder.getContextualEntityManager();
         MyEntity entity = new MyEntity("a", "b");
         em.persist(entity);
 
@@ -150,4 +155,107 @@ public class MyEntityExampleTestBase {
         assertEquals(1, dbutils.getTableSize("XYZ_MYENTITY"));
 
     }
+
+    @Test
+    public void testEntityPersistenceWithUserTransactions() throws Exception {
+
+        SimpleTransactionManager manager = new SimpleTransactionManager(
+            new DefaultUserTransactionFactory());
+        manager.addResource(jpaTester.getJpaBuilder());
+
+        UserTransaction transaction = manager.getTransaction();
+        transaction.begin();
+
+        EntityManager em = builder.getContextualEntityManager();
+        MyEntity entity = new MyEntity("a", "b");
+        em.persist(entity);
+
+        transaction.commit();
+
+        // Verify one row is written (using Db unit)
+        ITable table = dbtester.getDataSet().getTable("XYZ_MYENTITY");
+        assertEquals(1, table.getRowCount());
+
+        assertEquals("a", table.getValue(0, "SLEUTELTJE"));
+        assertEquals("b", table.getValue(0, "VALUE"));
+
+        // For this simple test, it can also be done through DatabaseUtils
+        assertEquals(1, dbutils.getTableSize("XYZ_MYENTITY"));
+
+    }
+
+    @Test
+    public void testTransactionCommittedCallback() throws Exception {
+        TransactionResultCallback callback = mock(TransactionResultCallback.class);
+        builder.execute(new JpaUnitOfWork<Void>() {
+            public Void execute(EntityManager aEm) {
+                MyEntity entity = new MyEntity("a", "b");
+                aEm.persist(entity);
+                return null;
+            }
+        }, callback);
+        verify(callback).status(TransactionResult.COMMIT);
+        verifyNoMoreInteractions(callback);
+    }
+
+    @Test
+    public void testTransactionMarkedForRollback() throws Exception {
+        TransactionResultCallback callback = mock(TransactionResultCallback.class);
+
+        builder.execute(new JpaUnitOfWork<Void>() {
+            public Void execute(EntityManager aEm) {
+                MyEntity entity = new MyEntity("a", "b");
+                aEm.persist(entity);
+                aEm.getTransaction().setRollbackOnly();
+                return null;
+            }
+        }, callback);
+        verify(callback).status(TransactionResult.ROLLBACK);
+        verifyNoMoreInteractions(callback);
+    }
+
+    @Test
+    public void testTransactionRolledBack() throws Exception {
+        TransactionResultCallback callback = mock(TransactionResultCallback.class);
+
+        builder.execute(new JpaUnitOfWork<Void>() {
+            public Void execute(EntityManager aEm) {
+                MyEntity entity = new MyEntity("a", "b");
+                aEm.persist(entity);
+                aEm.getTransaction().rollback();
+                return null;
+            }
+        }, callback);
+        verify(callback).status(TransactionResult.UNKNOWN);
+        verifyNoMoreInteractions(callback);
+    }
+
+    @Test
+    public void testTransactionCommitted() throws Exception {
+        TransactionResultCallback callback = mock(TransactionResultCallback.class);
+
+        builder.execute(new JpaUnitOfWork<Void>() {
+            public Void execute(EntityManager aEm) {
+                MyEntity entity = new MyEntity("a", "b");
+                aEm.persist(entity);
+                aEm.getTransaction().commit();
+                return null;
+            }
+        }, callback);
+        verify(callback).status(TransactionResult.UNKNOWN);
+        verifyNoMoreInteractions(callback);
+    }
+
+    @Test(expected = AssertionFailedError.class)
+    public void testTransactionNotCommittedUnexpectedly() throws Exception {
+        builder.execute(new JpaUnitOfWork<Void>() {
+            public Void execute(EntityManager aEm) {
+                MyEntity entity = new MyEntity("a", "b");
+                aEm.persist(entity);
+                aEm.getTransaction().rollback();
+                return null;
+            }
+        });
+    }
+
 }