com.github.greengerong.NgProtractor.java Source code

Introduction

Here is the source code for com.github.greengerong.NgProtractor.java

Source

/******************************************
 *                                        *
 * Auth: green gerong                     *
 * Date: 2014-03-02                       *
 * blog: http://greengerong.github.io/    *
 * github: https://github.com/greengerong *
 *                                        *
 ******************************************/

package com.github.greengerong;

/*
 * Copyright 2001-2005 The Apache Software Foundation.
 *
 * 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.
 */

import org.apache.commons.io.IOUtil;
import org.apache.commons.lang.StringUtils;
import org.apache.maven.plugin.AbstractMojo;
import org.apache.maven.plugin.MojoExecutionException;
import org.apache.maven.plugin.logging.Log;
import org.apache.maven.plugins.annotations.LifecyclePhase;
import org.apache.maven.plugins.annotations.Mojo;
import org.apache.maven.plugins.annotations.Parameter;

import java.io.File;
import java.io.IOException;
import java.util.Arrays;
import java.util.List;

import static com.google.common.base.Preconditions.checkNotNull;

@Mojo(name = "run", defaultPhase = LifecyclePhase.TEST)
public class NgProtractor extends AbstractMojo {

    @Parameter(property = "protractor", defaultValue = "protractor")
    private String protractor;

    @Parameter(property = "configFile", defaultValue = "${basedir}/protractor.conf.js", required = true)
    private File configFile;

    @Parameter(required = false)
    private boolean debug;

    @Parameter(required = false)
    private boolean debugBrk;

    @Parameter(property = "skipTests", required = false, defaultValue = "false")
    private Boolean skipTests;

    @Parameter(property = "skipProtractor", required = false, defaultValue = "false")
    private Boolean skipProtractor;

    @Parameter(property = "arguments", required = false)
    private String arguments;

    @Parameter(property = "ignoreFailed", required = false)
    private boolean ignoreFailed;

    @Parameter(property = "beforeRunning", required = false)
    private String beforeRunning;

    @Parameter(property = "afterRunning", required = false)
    private String afterRunning;

    public void execute() throws MojoExecutionException {
        final Log log = getLog();

        log.info(String.format("protractor:%s", protractor));
        log.info(String.format("configFile:%s", configFile));

        if (skipProtractor || skipTests) {
            log.info("Skipping protractor test.");
            return;
        }

        if (StringUtils.isNotBlank(beforeRunning)) {
            execBeforeRunning();
        }

        try {
            checkNotNull(protractor, "Protractor should not be empty.");
            checkFileExists(configFile, "Protractor should be exists.");

            new ProtractorService(ignoreFailed, log)
                    .exec(new Command(protractor, configFile, debug, debugBrk, arguments));
        } catch (Exception e) {
            throw new MojoExecutionException("There were exceptions when run protractor test.", e);
        } finally {
            execAfterRunning();
        }
    }

    private void execBeforeRunning() {
        executeCommand(beforeRunning, "before");
    }

    private void execAfterRunning() {
        executeCommand(afterRunning, "after");
    }

    private void executeCommand(final String cmd, final String logAction) {
        final Log log = getLog();

        try {
            log.info(String.format("execute %s running: %s", logAction, cmd));
            final ProcessBuilder processBuilder = new ProcessBuilder(getCommandAccordingToOS(cmd));
            final Process process = processBuilder.start();
            final String runningInfo = IOUtil.toString(process.getInputStream());
            log.info(runningInfo);
        } catch (IOException e) {
            log.warn(String.format("execute %s running script error", logAction), e);
        }
    }

    private List<String> getCommandAccordingToOS(final String execCommand) {
        final List<String> commandWithArgs = Arrays.asList(execCommand.split(" "));
        if (OSUtils.isWindows()) {
            commandWithArgs.addAll(0, Arrays.asList("cmd.exe", "/C"));
        }
        return commandWithArgs;
    }

    private File checkFileExists(File configFile, String message) {
        if (!configFile.exists()) {
            throw new RuntimeException(message);
        }
        return configFile;
    }
}