com.blazemeter.taurus.testng.TestNGRunner Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of taurus-java-helpers Show documentation
Taurus Java Helpers
There is a newer version: 1.11
package com.blazemeter.taurus.testng;

import com.blazemeter.taurus.reporting.TaurusReporter;
import org.testng.TestNG;
import org.testng.collections.Lists;

import java.io.FileReader;
import java.io.IOException;
import java.net.URL;
import java.net.URLClassLoader;
import java.util.*;
import java.util.jar.JarEntry;
import java.util.jar.JarFile;
import java.util.logging.Level;
import java.util.logging.Logger;


public class TestNGRunner {
    private static final Logger log = Logger.getLogger(TestNGRunner.class.getName());
    public static final String REPORT_FILE = "report_file";
    public static final String TARGET_PREFIX = "target_";
    public static final String TESTNG_CONFIG = "testng_config";
    public static final String ITERATIONS = "iterations";
    public static final String HOLD = "hold_for";
    public static TaurusReporter reporter;
    public static TestListener testListener;

    static {
        log.setLevel(Level.FINER);
    }

    public static void main(String[] args) throws Exception {
        log.info("Starting: " + Arrays.toString(args));
        if (args.length != 1) {
            throw new IllegalArgumentException("Usage requires 1 parameter, containing path to properties file");
        }

        Properties props = new Properties();
        props.load(new FileReader(args[0]));

        reporter = new TaurusReporter(props.getProperty(REPORT_FILE));
        testListener = new TestListener(reporter);

        long iterations = Long.valueOf(props.getProperty(ITERATIONS, "0"));
        float hold = Float.valueOf(props.getProperty(HOLD, "0"));
        if (iterations == 0) {
            if (hold > 0) {
                iterations = Long.MAX_VALUE;
            } else {
                iterations = 1;
            }
        }

        long startTime = System.currentTimeMillis();
        for (int iteration = 0; iteration < iterations; iteration++) {
            TestNG testNG = new TestNG();
            setupTestNG(testNG, props);
            testNG.run();
            log.info("Elapsed: " + (System.currentTimeMillis() - startTime) + ", limit: " + (hold * 1000));
            if (hold > 0 && System.currentTimeMillis() - startTime > hold * 1000) {
                log.info("Duration limit reached, stopping");
                break;
            }
        }

        reporter.close();
    }

    protected static void setupTestNG(TestNG testNG, Properties props) {
        testNG.setUseDefaultListeners(false);
        testNG.setVerbose(0);
        testNG.addListener(testListener);

        List suites = Lists.newArrayList();
        if (props.getProperty(TESTNG_CONFIG) != null) {
            log.info("Using TestNG config: " + props.getProperty(TESTNG_CONFIG));
            suites.add(props.getProperty(TESTNG_CONFIG));
            testNG.setTestSuites(suites);
        } else {
            log.info("No testng.xml passed, looking for test classes");
            ArrayList classes = getClasses(props);
            if (classes.isEmpty()) {
                throw new RuntimeException("Nothing to test");
            } else {
                log.info("Found test classes: " + classes.size());
            }
            Class[] classArray = classes.toArray(new Class[classes.size()]);
            testNG.setTestClasses(classArray);
        }
    }

    protected static ArrayList getClasses(Properties props) {
        ArrayList result = new ArrayList<>(0);

        Enumeration it = props.propertyNames();
        while (it.hasMoreElements()) {
            String propName = (String) it.nextElement();
            if (propName.startsWith(TARGET_PREFIX)) {
                result.addAll(getClasses(props.getProperty(propName)));
            }
        }

        return result;
    }

    protected static List> getClasses(String jarPath) {
        List> testClasses = new ArrayList<>();
        try {
            processJAR(testClasses, jarPath);
        } catch (IOException | ClassNotFoundException | NoClassDefFoundError | VerifyError e) {
            log.log(Level.WARNING, "Failed to add " + jarPath + "\n", e);
        }
        return testClasses;
    }

    protected static void processJAR(List> testClasses, String jarPath) throws IOException, ClassNotFoundException {
        log.info("Processing JAR: " + jarPath);
        JarFile jarFile = new JarFile(jarPath);
        Enumeration jarEntriesEnum = jarFile.entries();

        URL[] urls = {new URL("jar:file:" + jarPath + "!/")};
        URLClassLoader cl = URLClassLoader.newInstance(urls);

        while (jarEntriesEnum.hasMoreElements()) {
            JarEntry jarEntry = jarEntriesEnum.nextElement();
            if (jarEntry.isDirectory() || !jarEntry.getName().endsWith(".class")) {
                continue;
            }

            String className = jarEntry.getName().substring(0, jarEntry.getName().length() - ".class".length());
            className = className.replace('/', '.');

            // TODO: filter out non-test-suite classes
            Class c = cl.loadClass(className);
            testClasses.add(c);
        }
        jarFile.close();
    }
}