X-Git-Url: http://git.argeo.org/?a=blobdiff_plain;f=org.argeo.slc.hibernate%2Fsrc%2Ftest%2Fjava%2Forg%2Fargeo%2Fslc%2Fhibernate%2Ftest%2Ftree%2FTreeTestResultCollectionDaoHibernateTest.java;h=34430c96694362c61c100bd98bc292bd6310a8d5;hb=7a8f034133c1112e187e2a2bfd2c56a8d2452345;hp=344adbdb0e73c99086463380dc1c72d1458f359a;hpb=806325cbe46ab903b04eada8419961aa57102257;p=gpl%2Fargeo-slc.git

diff --git a/org.argeo.slc.hibernate/src/test/java/org/argeo/slc/hibernate/test/tree/TreeTestResultCollectionDaoHibernateTest.java b/org.argeo.slc.hibernate/src/test/java/org/argeo/slc/hibernate/test/tree/TreeTestResultCollectionDaoHibernateTest.java
index 344adbdb0..34430c966 100644
--- a/org.argeo.slc.hibernate/src/test/java/org/argeo/slc/hibernate/test/tree/TreeTestResultCollectionDaoHibernateTest.java
+++ b/org.argeo.slc.hibernate/src/test/java/org/argeo/slc/hibernate/test/tree/TreeTestResultCollectionDaoHibernateTest.java
@@ -1,15 +1,19 @@
 package org.argeo.slc.hibernate.test.tree;
 
+import java.sql.SQLException;
+
 import org.argeo.slc.core.test.tree.TreeTestResult;
 import org.argeo.slc.core.test.tree.TreeTestResultCollection;
 import org.argeo.slc.dao.test.tree.TreeTestResultCollectionDao;
 import org.argeo.slc.dao.test.tree.TreeTestResultDao;
-import org.argeo.slc.unit.AbstractSpringTestCase;
+import org.argeo.slc.hibernate.unit.HibernateTestCase;
 import org.argeo.slc.unit.test.tree.TreeTestResultTestUtils;
 import org.argeo.slc.unit.test.tree.UnitTestTreeUtil;
+import org.hibernate.HibernateException;
+import org.hibernate.Session;
+import org.springframework.orm.hibernate3.HibernateCallback;
 
-public class TreeTestResultCollectionDaoHibernateTest extends
-		AbstractSpringTestCase {
+public class TreeTestResultCollectionDaoHibernateTest extends HibernateTestCase {
 
 	public void testScenario() {
 		TreeTestResultDao ttrDao = getBean(TreeTestResultDao.class);
@@ -27,7 +31,7 @@ public class TreeTestResultCollectionDaoHibernateTest extends
 		ttrc.getResults().add(ttr1);
 		ttrcDao.update(ttrc);
 
-		TreeTestResult ttr2 = TreeTestResultTestUtils
+		final TreeTestResult ttr2 = TreeTestResultTestUtils
 				.createCompleteTreeTestResult();
 		ttrDao.create(ttr2);
 
@@ -37,11 +41,20 @@ public class TreeTestResultCollectionDaoHibernateTest extends
 		ttrc.getResults().remove(ttr1);
 		ttrcDao.update(ttrc);
 
-		TreeTestResultCollection ttrcPersist = ttrcDao
+		final TreeTestResultCollection ttrcPersist = ttrcDao
 				.getTestResultCollection(ttrcName);
-		assertEquals(1, ttrcPersist.getResults().size());
-		UnitTestTreeUtil.assertTreeTestResult(ttr2, ttrcPersist.getResults()
-				.iterator().next());
+
+		getHibernateTemplate().execute(new HibernateCallback() {
+
+			public Object doInHibernate(Session session)
+					throws HibernateException, SQLException {
+				session.refresh(ttrcPersist);
+				assertEquals(1, ttrcPersist.getResults().size());
+				UnitTestTreeUtil.assertTreeTestResult(ttr2, ttrcPersist
+						.getResults().iterator().next());
+				return null;
+			}
+		});
 	}
 
 	@Override