[jboss-cvs] JBossAS SVN: r57560 - trunk/testsuite/src/jdk15/org/jboss/test/aop/test

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

Author: kabir.khan at jboss.com
Date: 2006-10-11 05:22:07 -0400 (Wed, 11 Oct 2006)
New Revision: 57560

Added:
   trunk/testsuite/src/jdk15/org/jboss/test/aop/test/AOPTestSetup.java
Log:
Duplicate AOPTestSetup in the jdk15 source folder

Added: trunk/testsuite/src/jdk15/org/jboss/test/aop/test/AOPTestSetup.java
===================================================================
--- trunk/testsuite/src/jdk15/org/jboss/test/aop/test/AOPTestSetup.java	2006-10-11 04:54:23 UTC (rev 57559)
+++ trunk/testsuite/src/jdk15/org/jboss/test/aop/test/AOPTestSetup.java	2006-10-11 09:22:07 UTC (rev 57560)
@@ -0,0 +1,92 @@
+/*
+  * JBoss, Home of Professional Open Source
+  * Copyright 2005, JBoss Inc., and individual contributors as indicated
+  * by the @authors tag. See the copyright.txt in the distribution for a
+  * full listing of individual contributors.
+  *
+  * This is free software; you can redistribute it and/or modify it
+  * under the terms of the GNU Lesser General Public License as
+  * published by the Free Software Foundation; either version 2.1 of
+  * the License, or (at your option) any later version.
+  *
+  * This software is distributed in the hope that it will be useful,
+  * but WITHOUT ANY WARRANTY; without even the implied warranty of
+  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+  * Lesser General Public License for more details.
+  *
+  * You should have received a copy of the GNU Lesser General Public
+  * License along with this software; if not, write to the Free
+  * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
+  * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
+  */
+package org.jboss.test.aop.test;
+
+import javax.management.ObjectName;
+import javax.management.Attribute;
+
+import junit.framework.TestSuite;
+import org.jboss.test.JBossTestSetup;
+
+/** A test setup wrapper that enables the AspectManager transformation mode
+ * on setUp and disables it on tearDown.
+ *
+ * @author Scott.Stark at jboss.org
+ * @version $Revision: 37406 $
+ */
+public class AOPTestSetup extends JBossTestSetup
+{
+   public static String ASPECT_MANAGER_NAME = "jboss.aop:service=AspectManager";
+
+   private String jar;
+
+   // Create an initializer for the test suite
+   public AOPTestSetup(TestSuite suite, String jar) throws Exception
+   {
+      super(suite);
+      this.jar = jar;
+   }
+
+   protected void setUp() throws Exception
+   {
+      super.setUp();
+      ObjectName aspectManager = new ObjectName(ASPECT_MANAGER_NAME);
+      Attribute enableTransformer = new Attribute("EnableTransformer", Boolean.TRUE);
+      getServer().setAttribute(aspectManager, enableTransformer);
+      try
+      {
+         redeploy(jar);
+      }
+      catch(Exception e)
+      {
+         // Reset the EnableTransformer to false
+         try
+         {
+            enableTransformer = new Attribute("EnableTransformer", Boolean.FALSE);
+            getServer().setAttribute(aspectManager, enableTransformer);
+         }
+         catch(Exception ex)
+         {
+            getLog().error("Failed to set EnableTransformer to false", ex);
+         }
+         throw e;
+      }
+   }
+   protected void tearDown() throws Exception
+   {
+      Exception undeployException = null;
+      try
+      {
+         undeploy(jar);
+      }
+      catch(Exception e)
+      {
+         undeployException = e;
+      }
+      ObjectName aspectManager = new ObjectName(ASPECT_MANAGER_NAME);
+      Attribute enableTransformer = new Attribute("EnableTransformer", Boolean.FALSE);
+      getServer().setAttribute(aspectManager, enableTransformer);
+      if( undeployException != null )
+         throw undeployException;
+      super.tearDown();
+   }
+}