Start line:  
End line:  

Snippet Preview

Snippet HTML Code

Stack Overflow Questions
  /*
   * JBoss, Home of Professional Open Source
   * Copyright 2009, Red Hat Middleware LLC, and individual contributors
   * by the @authors tag. See the copyright.txt in the distribution for a
   * full listing of individual contributors.
   *
   * Licensed under the Apache License, Version 2.0 (the "License");
   * you may not use this file except in compliance with the License.
   * You may obtain a copy of the License at
  * http://www.apache.org/licenses/LICENSE-2.0
  * Unless required by applicable law or agreed to in writing, software
  * distributed under the License is distributed on an "AS IS" BASIS,
  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
 package org.jboss.arquillian.testng;
 
 
Arquillian

Author(s):
Aslak Knutsen
Version:
$Revision: $
 
 public abstract class Arquillian implements IHookable
 {
    public static final String ARQUILLIAN_DATA_PROVIDER = "ARQUILLIAN_DATA_PROVIDER";
 
    private static enum Cycle { BEFORE_SUITE, BEFORE_CLASS, BEFORE, TEST,  AFTER, AFTER_CLASS, AFTER_SUITE }
 
 
    private static ThreadLocal<Stack<Cycle>> cycleStack = new InheritableThreadLocal<Stack<Cycle>>() {
       protected java.util.Stack<CycleinitialValue() {
          return new Stack<Cycle>();
       };
    };
 
    @BeforeSuite(groups = "arquillian", inheritGroups = true)
    public void arquillianBeforeSuite() throws Exception
    {
       if(.get() == null)
       {
          TestRunnerAdaptor adaptor = TestRunnerAdaptorBuilder.build();
          adaptor.beforeSuite();
          .set(adaptor); // don't set TestRunnerAdaptor if beforeSuite fails
          .get().push(.);
       }
    }
 
    @AfterSuite(groups = "arquillian", inheritGroups = true, alwaysRun = true)
    public void arquillianAfterSuite() throws Exception
    {
       if (.get() == null)
       {
          return// beforeSuite failed
       }
       if(.get().empty())
       {
          return;
       }
       if(.get().peek() != .)
       {
          return// Arquillian lifecycle called out of order, expected " + Cycle.BEFORE_SUITE
       }
       else
       {
          .get().pop();
       }
       .get().afterSuite();
       .get().shutdown();
       .set(null);
       .remove();
       .set(null);
       .remove();
    }
 
    @BeforeClass(groups = "arquillian", inheritGroups = true)
    public void arquillianBeforeClass() throws Exception
   {
   }
   @AfterClass(groups = "arquillian", inheritGroups = true, alwaysRun = true)
   public void arquillianAfterClass() throws Exception
   {
      if(.get().empty())
      {
         return;
      }
      if(.get().peek() != .)
      {
         return// Arquillian lifecycle called out of order, expected " + Cycle.BEFORE_CLASS
      }
      else
      {
         .get().pop();
      }
   }
   @BeforeMethod(groups = "arquillian", inheritGroups = true)
   public void arquillianBeforeTest(Method testMethodthrows Exception
   {
      .get().before(thistestMethod.);
   }
   @AfterMethod(groups = "arquillian", inheritGroups = true, alwaysRun = true)
   public void arquillianAfterTest(Method testMethodthrows Exception
   {
      if(.get().empty())
      {
         return;
      }
      if(.get().peek() != .)
      {
         return// Arquillian lifecycle called out of order, expected " + Cycle.BEFORE_CLASS
      }
      else
      {
         .get().pop();
      }
      .get().after(thistestMethod.);
   }
   public void run(final IHookCallBack callbackfinal ITestResult testResult)
   {
      TestResult result;
      try
      {
         result = .get().test(new TestMethodExecutor()
         {
            public void invoke(Object... parametersthrows Throwable
            {
               /*
                *  The parameters are stored in the InvocationHandler, so we can't set them on the test result directly.
                *  Copy the Arquillian found parameters to the InvocationHandlers parameters
                */
               copyParameters(parameterscallback.getParameters());
               callback.runTestMethod(testResult);
               // Parameters can be contextual, so extract information
               swapWithClassNames(callback.getParameters());
               testResult.setParameters(callback.getParameters());
               if (testResult.getThrowable() != null) {
                   throw testResult.getThrowable();
               }
            }
            private void copyParameters(Object[] sourceObject[] target)
            {
               for(int i = 0; i < source.lengthi++)
               {
                  if(source[i] != null)
                  {
                     target[i] = source[i];
                  }
               }
            }
            private void swapWithClassNames(Object[] source)
            {
               // clear parameters. they can be contextual and might fail TestNG during the report writing.
               for(int i = 0; source != null && i < source.lengthi++)
               {
                  Object parameter = source[i];
                  if(parameter != null)
                  {
                     source[i] = parameter.toString();
                  }
                  else
                  {
                     source[i] = "null";
                  }
               }
            }
            public Method getMethod()
            {
               return testResult.getMethod().getMethod();
            }
            public Object getInstance()
            {
               return Arquillian.this;
            }
         });
         if(result.getThrowable() != null)
         {
            testResult.setThrowable(result.getThrowable());
         }
         // calculate test end time. this is overwritten in the testng invoker..
         testResult.setEndMillis( (result.getStart() - result.getEnd()) + testResult.getStartMillis());
      }
      catch (Exception e)
      {
         testResult.setThrowable(e);
      }
   }
   public Object[][] arquillianArgumentProvider(Method method)
   {
      Object[][] values = new Object[1][method.getParameterTypes().length];
      if (.get() == null)
      {
         return values;
      }
      Object[] parameterValues = new Object[method.getParameterTypes().length];
      values[0] = parameterValues;
      return values;
   }
   {
      if(.get() == null)
      {
         throw new IllegalStateException("No TestRunnerAdaptor found, @BeforeSuite has not been called");
      }
   }
New to GrepCode? Check out our FAQ X