org.testng.CheTestNG Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of che-plugin-testing-testng-runtime Show documentation

There is a newer version: 7.0.0-rc-3.0

Show newest version

/*
 * Copyright (c) 2012-2018 Red Hat, Inc.
 * This program and the accompanying materials are made
 * available under the terms of the Eclipse Public License 2.0
 * which is available at https://www.eclipse.org/legal/epl-2.0/
 *
 * SPDX-License-Identifier: EPL-2.0
 *
 * Contributors:
 *   Red Hat, Inc. - initial API and implementation
 */
package org.testng;

import java.util.ArrayList;
import java.util.List;
import org.testng.listeners.CheSuiteListener;
import org.testng.listeners.CheTestListener;
import org.testng.xml.XmlSuite;

/** Class for running tests in the TestNG framework. */
public class CheTestNG extends TestNG {

  @Override
  public void run() {
    try {
      initializeSuitesAndJarFile();

      List suites = new ArrayList<>();
      flatSuites(m_suites, suites);

      if (suites.isEmpty()) {
        TestingMessageHelper.reporterAttached(System.out);
        System.err.print("No test found to run.");
      } else {
        addCheListeners();
        super.run();
        System.exit(0);
      }

    } catch (Throwable throwable) {
      throwable.printStackTrace(System.err);
    }
  }

  private void addCheListeners() {
    CheTestNGListener listener = new CheTestNGListener();
    addListener(new CheSuiteListener(listener));
    addListener(new CheTestListener(listener));

    try {
      Class configurationListenerClass =
          Class.forName("org.testng.listeners.CheConfigurationListener");
      Object confListener =
          configurationListenerClass.getConstructor(CheTestNGListener.class).newInstance(listener);
      addListener(confListener);

      Class methodListenerClass = Class.forName("org.testng.listeners.CheInvokedMethodListener");
      Object methodListener =
          methodListenerClass.getConstructor(CheTestNGListener.class).newInstance(listener);
      addListener(methodListener);

      configurationListenerClass.getMethod("setIgnoreStarted").invoke(confListener);
    } catch (Throwable ignored) {
    }
  }

  private void flatSuites(List suites, List result) {
    for (XmlSuite suite : suites) {
      result.add(suite);
      flatSuites(suite.getChildSuites(), result);
    }
  }
}