com.smash.revolance.ui.materials.CmdlineHelper.java Source code

Introduction

Here is the source code for com.smash.revolance.ui.materials.CmdlineHelper.java

Source

package com.smash.revolance.ui.materials;

/*
 * ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
 * Revolance-UI-Materials-Commons
 * ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
 * Copyright (C) 2012 - 2013 RevoLance
 * ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as
 * published by the Free Software Foundation, either version 3 of the 
 * License, or (at your option) any later version.
 * 
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 * 
 * You should have received a copy of the GNU General Public 
 * License along with this program.  If not, see
 * <http://www.gnu.org/licenses/gpl-3.0.html>.
 * ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
 */

import org.apache.commons.io.FileUtils;

import java.io.File;
import java.io.IOException;
import java.util.UUID;

import static java.io.File.createTempFile;

/**
 * User: wsmash
 * Date: 10/10/13
 * Time: 22:14
 */
public class CmdlineHelper extends TestConstants {
    private final String id;
    private boolean sync = false;
    private File out;
    private String[] cmd;
    private File dir;
    private Process process;
    private File in;
    private File err;

    public CmdlineHelper() throws IOException {
        id = UUID.randomUUID().toString();

        err = createTempFile("err-" + id, ".txt");
        out = createTempFile("out-" + id, ".txt");
        in = createTempFile("in-" + id, ".txt");
    }

    public CmdlineHelper cmd(String... cmd) {
        this.cmd = cmd;
        return this;
    }

    public CmdlineHelper dir(File dir) {
        this.dir = dir;
        return this;
    }

    public CmdlineHelper dir(String target) {
        dir(new File(target));
        return this;
    }

    public CmdlineHelper sync() {
        sync = true;
        return this;
    }

    public CmdlineHelper exec() throws InterruptedException, IOException {
        ProcessBuilder pb = new ProcessBuilder();

        if (dir != null) {
            pb.directory(dir);
        }

        pb.command(cmd);

        pb.redirectError(ProcessBuilder.Redirect.to(err));
        pb.redirectOutput(ProcessBuilder.Redirect.to(out));
        pb.redirectInput(ProcessBuilder.Redirect.from(in));

        System.out.println("Executing cmd: " + cmd[0] + " from dir: " + dir);
        System.out.println("Redirecting out to: " + out.getAbsolutePath());
        System.out.println("Redirecting err to: " + err.getAbsolutePath());

        Process process = pb.start();
        if (sync) {
            process.waitFor();
        }

        this.process = process;

        return this;
    }

    public int exitValue() {
        return process.exitValue();
    }

    public String id() {
        return id;
    }

    public void write(String in) throws IOException {
        FileUtils.writeStringToFile(this.in, in);
    }

    public void waitInSec(int seconds) {
        long mark = System.currentTimeMillis();
        while ((System.currentTimeMillis() - mark) < seconds * 1000) {
            sleep(1);
        }
    }

    private void sleep(int seconds) {
        try {
            Thread.sleep(seconds);
        } catch (InterruptedException e) {
            // Ignore gently
        }
    }

    public CmdlineHelper awaitOut(String line, int timeout) throws Exception {
        long mark = System.currentTimeMillis();
        timeout = timeout * 1000;

        do {
            sleep(1000);
            if (outContains(line)) {
                return this;
            }
        } while ((System.currentTimeMillis() - mark) < timeout);

        if (!outContains(line)) {
            throw new Exception("Unable to find: '" + line + "' in out");
        }
        return this;
    }

    public boolean hasExited() {
        try {
            process.exitValue();
            return true;
        } catch (IllegalThreadStateException e) {
            return false;
        }
    }

    public CmdlineHelper kill() {
        process.destroy();
        return this;
    }

    public File getOut() {
        return out;
    }

    public File getErr() {
        return err;
    }

    public File getIn() {
        return in;
    }

    public String out() throws IOException {
        return FileUtils.readFileToString(getOut());
    }

    public String err() throws IOException {
        return FileUtils.readFileToString(getErr());
    }

    public CmdlineHelper withoutErrors() throws Exception {
        if (!err().isEmpty()) {
            throw new Exception("Process execution generated errors in file: " + err);
        }
        return this;
    }

    public CmdlineHelper awaitErr(String line, int timeout) throws Exception {
        long mark = System.currentTimeMillis();
        timeout = timeout * 1000;

        do {
            if (errContains(line)) {
                return this;
            } else {
                sleep(1000);
            }
        } while ((System.currentTimeMillis() - mark) < timeout);

        if (!errContains(line)) {
            throw new Exception("Unable to find: '" + line + "' in err");
        }
        return this;
    }

    public boolean errContains(String line) throws IOException {
        return err().contains(line);
    }

    public boolean outContains(String line) throws IOException {
        return out().contains(line);
    }

    public void await(int timeout) {
        sleep(timeout);
    }
}