com.tascape.reactor.AbstractCaseRunner Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of reactor Show documentation
Reactor - Automation Framework
There is a newer version: 1.3.19
/*
 * Copyright (c) 2015 - present Nebula Bay.
 * All rights reserved.
 *
 * 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 com.tascape.reactor;

import com.tascape.reactor.db.DbHandler;
import com.tascape.reactor.db.CaseResult;
import java.io.File;
import java.io.IOException;
import java.io.PrintWriter;
import java.nio.charset.Charset;
import java.nio.file.Path;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
import org.apache.commons.io.FileUtils;
import org.apache.log4j.Appender;
import org.apache.log4j.FileAppender;
import org.apache.log4j.PatternLayout;
import org.apache.log4j.spi.Filter;
import org.apache.log4j.spi.LoggingEvent;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 *
 * @author linsong wang
 */
public abstract class AbstractCaseRunner {
    private static final Logger LOG = LoggerFactory.getLogger(AbstractCaseRunner.class);

    private static final ThreadLocal CASE_RESULT = new ThreadLocal<>();

    public static void setCaseResult(CaseResult caseResult) {
        CASE_RESULT.set(caseResult);
    }

    public static CaseResult getCaseResult() {
        return CASE_RESULT.get();
    }

    protected SystemConfiguration sysConfig = SystemConfiguration.getInstance();

    protected DbHandler db = null;

    protected CaseResult tcr = null;

    protected String execId = "";

    public abstract void runTaskCase() throws Exception;

    protected void generateHtml(Path logFile) {
        Pattern http = Pattern.compile("((http|https)://\\S+)");

        Path html = logFile.getParent().resolve("log.html");
        LOG.trace("creating file {}", html);
        try (PrintWriter pw = new PrintWriter(html.toFile())) {
            pw.println("");
            pw.println("Suite Log Directory
Case Log Directory");
            pw.println();
            List lines = FileUtils.readLines(logFile.toFile(),Charset.defaultCharset());
            List files = new ArrayList<>(Arrays.asList(logFile.getParent().toFile().listFiles()));

            for (String line : lines) {
                String newline = line.replaceAll(">", ">");
                newline = newline.replaceAll("<", "<");
                if (newline.contains(" INFO  ")) {
                    newline = "" + newline + "";
                } else if (newline.contains(" WARN  ")) {
                    newline = "" + newline + "";
                } else if (newline.contains(" ERROR ")
                    || newline.contains("Failure in case")
                    || newline.contains("AssertionError")) {
                    newline = "" + newline + "";
                } else {
                    Matcher m = http.matcher(line);
                    if (m.find()) {
                        String url = m.group(1);
                        String a = String.format("%s", url, url);
                        newline = newline.replace(url, a);
                    }
                }
                pw.println(newline);
                for (File file : files) {
                    String path = file.getAbsolutePath();
                    String name = file.getName();
                    if (newline.contains(path)) {
                        if (name.endsWith(".png")) {
                            pw.printf("",
                                name, name);
                        }
                        String a = String.format("%s", name, name);
                        int len = newline.indexOf("    ");
                        pw.printf((len > 0 ? newline.substring(0, len + 5) : "") + a);
                        pw.println();
                        files.remove(file);
                        break;
                    }
                }
            }

            pw.println("");
            logFile.toFile().delete();
        } catch (IOException ex) {
            LOG.warn(ex.getMessage());
        }
    }

    Path addLog4jFileAppender(final Path path) throws IOException {
        org.apache.log4j.Logger rootLogger = org.apache.log4j.Logger.getRootLogger();

        String pattern = "%d{HH:mm:ss.SSS} %-5p %t %C{1}.%M:%L - %m%n";
        final String threadName = Thread.currentThread().getName();

        class ThreadFilter extends Filter {
            @Override
            public int decide(LoggingEvent event) {
                if (event.getThreadName().startsWith(threadName)) {
                    return Filter.ACCEPT;
                }
                return Filter.DENY;
            }
        }

        FileAppender fa = new FileAppender(new PatternLayout(pattern), path.toFile().getAbsolutePath());
        fa.addFilter(new ThreadFilter());
        fa.setThreshold(sysConfig.getCaseLogLevel());
        fa.setImmediateFlush(true);
        fa.setAppend(true);
        fa.setName(path.toFile().getAbsolutePath());

        fa.activateOptions();
        rootLogger.addAppender(fa);

        return path;
    }

    void removeLog4jAppender(Path path) {
        if (path == null) {
            LOG.warn("Appender name is null");
            return;
        }
        org.apache.log4j.Logger rootLogger = org.apache.log4j.Logger.getRootLogger();
        Appender appender = rootLogger.getAppender(path.toFile().getAbsolutePath());
        if (appender != null) {
            appender.close();
            rootLogger.removeAppender(appender);
        }
    }
}