com.github.mahnkong.testutils.byteman.AbstractBytemanTestRule Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of byteman-junit-rules Show documentation

This project contains functionality allowing the usage of byteman in JUnit tests by providing JUnit rules (install byteman agent, load/unload of the rules) and some annotations (byteman setup for the tests).

There is a newer version: 1.0.3

Show newest version

package com.github.mahnkong.testutils.byteman;

import org.junit.rules.TestRule;

import java.io.BufferedReader;
import java.io.File;
import java.io.InputStreamReader;

/**
 * Created by mahnkong on 19.04.15.
 */
public abstract class AbstractBytemanTestRule implements TestRule {
    private String bytemanHome;
    String bindAddress;
    int bindPort;

    private Runtime runtime = Runtime.getRuntime();

    private boolean isWindows = false;
    protected String bmsubmit = "";
    protected String bminstall = "";

    public AbstractBytemanTestRule(String bytemanHome, String bindAddress, int bindPort) {
        this.bytemanHome = bytemanHome;
        this.bindAddress = bindAddress;
        this.bindPort = bindPort;

        if (System.getProperty ("os.name").toLowerCase().contains("windows")) {
            isWindows = true;
        }
        String prefix = this.bytemanHome + File.separator + "bin" + File.separator;
        String suffix = (isWindows ? ".bat" : ".sh");

        bmsubmit = prefix + "bmsubmit" + suffix;
        bminstall = prefix + "bminstall" + suffix;
    }

    protected void execute(String command, boolean verbose) throws Exception {
        Process p = runtime.exec(command);
        p.waitFor();

        if (verbose) {
            BufferedReader reader =
                    new BufferedReader(new InputStreamReader(p.getInputStream()));

            String line;
            while ((line = reader.readLine())!= null) {
                System.out.println(line);
            }
        }

        if (p.exitValue() != 0) {
            throw new IllegalStateException("Execution of command '" + command + "' failed with exit code '" + p.exitValue() + "'");
        }
    }

    public String getBytemanHome() {
        return bytemanHome;
    }
}