[jboss-cvs] JBossAS SVN: r78706 - projects/ejb3/trunk/core/src/test/java/org/jboss/ejb3/core/test/ejbthree670/unit.

[jboss-cvs-commits at lists.jboss.org]

Author: wolfc
Date: 2008-09-19 08:02:02 -0400 (Fri, 19 Sep 2008)
New Revision: 78706

Modified:
   projects/ejb3/trunk/core/src/test/java/org/jboss/ejb3/core/test/ejbthree670/unit/PreDestroyCallsTestCase.java
Log:
EJBTHREE-670: cleanup of containers

Modified: projects/ejb3/trunk/core/src/test/java/org/jboss/ejb3/core/test/ejbthree670/unit/PreDestroyCallsTestCase.java
===================================================================
--- projects/ejb3/trunk/core/src/test/java/org/jboss/ejb3/core/test/ejbthree670/unit/PreDestroyCallsTestCase.java	2008-09-19 12:01:38 UTC (rev 78705)
+++ projects/ejb3/trunk/core/src/test/java/org/jboss/ejb3/core/test/ejbthree670/unit/PreDestroyCallsTestCase.java	2008-09-19 12:02:02 UTC (rev 78706)
@@ -24,6 +24,9 @@
 import static org.junit.Assert.assertEquals;
 import static org.junit.Assert.fail;
 
+import java.util.ArrayList;
+import java.util.List;
+
 import javax.transaction.TransactionManager;
 
 import org.jboss.ejb3.core.test.common.AbstractEJB3TestCase;
@@ -32,6 +35,7 @@
 import org.jboss.ejb3.core.test.ejbthree670.MyStateful21Bean;
 import org.jboss.ejb3.core.test.ejbthree670.MyStateful21Home;
 import org.jboss.ejb3.core.test.ejbthree670.MyStatefulBean;
+import org.jboss.ejb3.session.SessionContainer;
 import org.junit.AfterClass;
 import org.junit.BeforeClass;
 import org.junit.Test;
@@ -44,9 +48,15 @@
  */
 public class PreDestroyCallsTestCase extends AbstractEJB3TestCase
 {
+   private static List<SessionContainer> containers = new ArrayList<SessionContainer>();
+   
    @AfterClass
    public static void afterClass() throws Exception
    {
+      for(SessionContainer container : containers)
+         undeployEjb(container);
+      containers.clear();
+      
       AbstractEJB3TestCase.afterClass();
    }
    
@@ -55,8 +65,8 @@
    {
       AbstractEJB3TestCase.beforeClass();
       
-      deploySessionEjb(MyStatefulBean.class);
-      deploySessionEjb(MyStateful21Bean.class);
+      containers.add(deploySessionEjb(MyStatefulBean.class));
+      containers.add(deploySessionEjb(MyStateful21Bean.class));
    }
    
    @Test