com.qa.framework.generator.TestngXmlGenerator Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of smart-api-framework Show documentation

Support web service api automaton test based on testng and httpclient

There is a newer version: 2.5

package com.qa.framework.generator;


import com.qa.framework.library.base.IOHelper;
import com.qa.framework.library.base.XMLHelper;
import org.dom4j.Element;

import java.io.File;

/**
 * The type Testng xml generator.
 */
public class TestngXmlGenerator {
    /**
     * The entry point of application.
     *
     * @param args the input arguments
     */
    public static void main(String[] args) {
        for (String arg : args) {
            System.out.println("------->" + arg);
        }
        autoGenerateFactory(args[0], args[1]);
    }

    /**
     * Auto generate factory.
     *
     * @param outputPath the output path
     * @param threadCnt  the thread cnt
     */
    public static void autoGenerateFactory(String outputPath, String threadCnt) {
        XMLHelper xml = new XMLHelper();
        xml.createDocument();
        Element root = xml.createDocumentRoot("suite");
        xml.addAttribute(root, "name", "Factory_" + threadCnt + "_Thread");
        xml.addAttribute(root, "thread-count", threadCnt);
        xml.addAttribute(root, "parallel", "methods");
        xml.addAttribute(root, "verbose", "1");
        Element listeners = xml.addChildElement(root, "listeners");
        Element listener = xml.addChildElement(listeners, "listener");
        xml.addAttribute(listener, "class-name", "com.qa.framework.testnglistener.RetryListener");
        Element test = xml.addChildElement(root, "test");
        xml.addAttribute(test, "name", "FactoryExecutor_execute");
        xml.addAttribute(test, "timeout", "600000");
        Element classes = xml.addChildElement(test, "classes");
        Element cls = xml.addChildElement(classes, "class");
        xml.addAttribute(cls, "name", "com.qa.framework.factory.ExecutorFactory");
        Element methods = xml.addChildElement(cls, "methods");
        Element include = xml.addChildElement(methods, "include");
        xml.addAttribute(include, "name", "execute");
        IOHelper.deleteDirectory(outputPath);
        IOHelper.createNestDirectory(outputPath);
        if (outputPath.endsWith("/")) {
            xml.saveTo(outputPath + "Factory_" + threadCnt + "_Thread.xml");
        } else {
            xml.saveTo(outputPath + File.separator + "Factory_" + threadCnt + "_Thread.xml");
        }
    }
}