[jboss-svn-commits] JBL Code SVN: r37593 - labs/jbosstm/trunk/XTS/sar/tests/arquillian/src/main/java/org/jboss/jbossts/xts/servicetests/ejb.

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

Author: zhfeng
Date: 2011-10-19 02:56:42 -0400 (Wed, 19 Oct 2011)
New Revision: 37593

Modified:
   labs/jbosstm/trunk/XTS/sar/tests/arquillian/src/main/java/org/jboss/jbossts/xts/servicetests/ejb/XTSServiceTestRunnerEJBImpl.java
Log:
update to add start and stop function in TestRunner

Modified: labs/jbosstm/trunk/XTS/sar/tests/arquillian/src/main/java/org/jboss/jbossts/xts/servicetests/ejb/XTSServiceTestRunnerEJBImpl.java
===================================================================
--- labs/jbosstm/trunk/XTS/sar/tests/arquillian/src/main/java/org/jboss/jbossts/xts/servicetests/ejb/XTSServiceTestRunnerEJBImpl.java	2011-10-19 03:09:50 UTC (rev 37592)
+++ labs/jbosstm/trunk/XTS/sar/tests/arquillian/src/main/java/org/jboss/jbossts/xts/servicetests/ejb/XTSServiceTestRunnerEJBImpl.java	2011-10-19 06:56:42 UTC (rev 37593)
@@ -20,29 +20,45 @@
  */
 package org.jboss.jbossts.xts.servicetests.ejb;
 
-import javax.ejb.Stateless;
+import javax.ejb.Singleton;
+import javax.ejb.Startup;
+import javax.annotation.PostConstruct;
+import javax.annotation.PreDestroy;
 
 import org.jboss.jbossts.xts.servicetests.service.recovery.TestATRecoveryModule;
 import org.jboss.jbossts.xts.servicetests.service.recovery.TestBARecoveryModule;
 import org.jboss.jbossts.xts.servicetests.test.XTSServiceTest;
 import org.jboss.logging.Logger;
 
- at Stateless
+ at Singleton
+ at Startup
 public class XTSServiceTestRunnerEJBImpl implements XTSServiceTestRunnerEJB {
 
-	@SuppressWarnings("rawtypes")
-	public void runTest(String testName) throws Exception {
+	@PostConstruct
+	public void start() {
 		log.info("Starting XTSServiceTestRunner");
 
 		// ensure that the xts service test AT recovery helper module is registered
 		TestATRecoveryModule.register();
 		TestBARecoveryModule.register();
+	}
+	
+	@PreDestroy
+	public void stop() {
+		// ensure that the xts service test AT recovery helper module is unregistered
+		TestATRecoveryModule.unregister();
+		TestBARecoveryModule.unregister();
 
+		log.info("Stopped XTSServiceTestRunner");
+	}
+
+	@SuppressWarnings("rawtypes")
+	public void runTest(String testName) throws Exception {
 		if(testName != null) {
 			log.info("XTSServiceTestRunner run test " + testName);
 			Class testClass;
 			ClassLoader cl = XTSServiceTestRunnerEJBImpl.class.getClassLoader();
-			
+
 			try {
 				testClass = cl.loadClass(testName);
 			} catch (ClassNotFoundException cnfe) {
@@ -81,12 +97,6 @@
 				log.info("Joined test thread " + testName);
 			}
 		}
-		// ensure that the xts service test AT recovery helper module is unregistered
-
-		TestATRecoveryModule.unregister();
-		TestBARecoveryModule.unregister();
-
-		log.info("Stopped XTSServiceTestRunner");
 	}
 
 	private final Logger log = org.jboss.logging.Logger.getLogger(XTSServiceTestRunnerEJBImpl.class);