[jbpm-commits] JBoss JBPM SVN: r3689 - jbpm3/branches/aguizar/modules/core/src/test/java/org/jbpm/perf.

[do-not-reply at jboss.org]

Author: alex.guizar at jboss.com
Date: 2009-01-20 19:28:18 -0500 (Tue, 20 Jan 2009)
New Revision: 3689

Modified:
   jbpm3/branches/aguizar/modules/core/src/test/java/org/jbpm/perf/TasklistEagerLoadingTest.java
Log:
JBPM-1812 clean leftover task instances

Modified: jbpm3/branches/aguizar/modules/core/src/test/java/org/jbpm/perf/TasklistEagerLoadingTest.java
===================================================================
--- jbpm3/branches/aguizar/modules/core/src/test/java/org/jbpm/perf/TasklistEagerLoadingTest.java	2009-01-20 17:36:02 UTC (rev 3688)
+++ jbpm3/branches/aguizar/modules/core/src/test/java/org/jbpm/perf/TasklistEagerLoadingTest.java	2009-01-21 00:28:18 UTC (rev 3689)
@@ -1,6 +1,7 @@
 package org.jbpm.perf;
 
 import java.util.ArrayList;
+import java.util.Collections;
 import java.util.List;
 
 import org.jbpm.db.AbstractDbTestCase;
@@ -8,32 +9,52 @@
 
 public class TasklistEagerLoadingTest extends AbstractDbTestCase {
 
+  private List<Long> taskInstanceIds = new ArrayList<Long>();
+
   public void testTasklistEagerLoading() {
-    for (int i=0; i<20; i++) {
-      TaskInstance taskInstance = new TaskInstance("task "+i);
+    for (int i = 0; i < 20; i++) {
+      TaskInstance taskInstance = new TaskInstance("task " + i);
       taskInstance.setActorId("johndoe");
-      session.save(taskInstance);
+      save(taskInstance);
     }
     newTransaction();
-    
-    assertEquals(20, jbpmContext.getTaskList("johndoe").size());
+    try {
+      assertEquals(20, jbpmContext.getTaskList("johndoe").size());
+    }
+    finally {
+      deleteTaskInstances();
+    }
   }
 
   public void testPooledTasklistEagerLoading() {
-    for (int i=0; i<20; i++) {
-      TaskInstance taskInstance = new TaskInstance("group task "+i);
-      taskInstance.setPooledActors(new String[]{"group"+i});
-      session.save(taskInstance);
+    for (int i = 0; i < 20; i++) {
+      TaskInstance taskInstance = new TaskInstance("group task " + i);
+      taskInstance.setPooledActors(new String[] { "group" + i });
+      save(taskInstance);
     }
-    for (int i=0; i<20; i++) {
-      TaskInstance taskInstance = new TaskInstance("task "+i);
-      taskInstance.setPooledActors(new String[]{"johndoe", "bachelors", "partyanimals", "wildwomen"});
-      session.save(taskInstance);
+    for (int i = 0; i < 20; i++) {
+      TaskInstance taskInstance = new TaskInstance("task " + i);
+      taskInstance.setPooledActors(new String[] { "johndoe", "bachelors", "partyanimals",
+          "wildwomen" });
+      save(taskInstance);
     }
     newTransaction();
-    
-    List actorIds = new ArrayList();
-    actorIds.add("johndoe");
-    assertEquals(20, jbpmContext.getGroupTaskList(actorIds).size());
+    try {
+      assertEquals(20, jbpmContext.getGroupTaskList(Collections.singletonList("johndoe")).size());
+    }
+    finally {
+      deleteTaskInstances();
+    }
   }
+
+  private void save(TaskInstance taskInstance) {
+    session.save(taskInstance);
+    taskInstanceIds.add(taskInstance.getId());
+  }
+
+  private void deleteTaskInstances() {
+    for (Long id : taskInstanceIds) {
+      session.delete(session.load(TaskInstance.class, id));
+    }
+  }
 }