/*

  * 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.container;

import java.util.ArrayList;

import java.util.Arrays;

import java.util.List;

import org.jboss.arquillian.container.test.spi.TestRunner;

import org.jboss.arquillian.test.spi.TestResult;

import org.testng.TestNG;

import org.testng.internal.AnnotationTypeEnum;

import org.testng.xml.XmlClass;

import org.testng.xml.XmlInclude;

import org.testng.xml.XmlSuite;

import org.testng.xml.XmlTest;

public class TestNGTestRunner implements TestRunner

{

   public TestResult execute(Class<?> testClass, String methodName)

   {

      TestListener resultListener = new TestListener();

      TestNG runner = new TestNG(false);

      runner.setVerbose(0);

      runner.addListener(resultListener);

      runner.addListener(new RemoveDependsOnTransformer());

      runner.setXmlSuites(

            Arrays.asList(createSuite(testClass, methodName)));

      runner.run();

      return resultListener.getTestResult(); 

   }

   private XmlSuite createSuite(Class<?> className, String methodName)

   {

      XmlSuite suite = new XmlSuite();

      suite.setName("Arquillian");

      suite.setAnnotations(AnnotationTypeEnum.JDK.getName());

      XmlTest test = new XmlTest(suite);

      test.setName("Arquillian - " + className);

      List<XmlClass> testClasses = new ArrayList<XmlClass>();

      XmlClass testClass = new XmlClass(className);

      testClass.getIncludedMethods().add(new XmlInclude(methodName));

      testClasses.add(testClass);

      test.setXmlClasses(testClasses);

      return suite;

   }

}