com.github.mahnkong.testutils.byteman.BytemanRuleSubmitter 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.runner.Description;
import org.junit.runners.model.Statement;

import java.io.File;

/**
 * Created by mahnkong on 21.03.15.
 */
public class BytemanRuleSubmitter extends AbstractBytemanTestRule {

    private BytemanRuleSubmitter(Builder builder) {
        super(builder.bytemanHome);
    }

    public Statement apply(Statement base, Description description) {
        BytemanRuleFile bytemanRuleFileClassAnnotation = description.getTestClass().getAnnotation(BytemanRuleFile.class);
        BytemanRuleFile bytemanRuleFileMethodAnnotation = description.getAnnotation(BytemanRuleFile.class);
        IgnoreBytemanClassRuleFile ignoreBytemanClassRuleFile = description.getAnnotation(IgnoreBytemanClassRuleFile.class);

        if (bytemanRuleFileMethodAnnotation != null) {
            base = processBytemanRuleFile(base, bytemanRuleFileMethodAnnotation);
        }
        if (ignoreBytemanClassRuleFile == null) {
            return processBytemanRuleFile(base, bytemanRuleFileClassAnnotation);
        }
        return base;
    }

    private Statement processBytemanRuleFile(final Statement statement, final BytemanRuleFile bytemanRuleFile) {

        return new Statement() {
            @Override
            public void evaluate() throws Throwable {
                submitBytemanRule(bytemanRuleFile);
                try {
                    statement.evaluate();
                } finally {
                    removeBytemanRule(bytemanRuleFile);
                }
            }
        };
    }

    private void submitBytemanRule(final BytemanRuleFile bytemanRuleFile) throws Exception {
        if (bytemanRuleFile != null) {
            String command = bmsubmit + " " + new File(bytemanRuleFile.filepath()).getPath();
            execute(command, bytemanRuleFile.verbose());
            if (bytemanRuleFile.verbose()) {
                execute(bmsubmit, bytemanRuleFile.verbose());
            }
        }
    }

    private void removeBytemanRule(final BytemanRuleFile bytemanRuleFile) throws Exception {
        if (bytemanRuleFile != null) {
            String command = bmsubmit + " -u " + new File(bytemanRuleFile.filepath()).getPath();
            execute(command, bytemanRuleFile.verbose());
        }
    }

    public static class Builder {
        private String bytemanHome;


        public Builder() {
            this.bytemanHome = System.getenv("BYTEMAN_HOME");
        }

        public Builder bytemanHome(final String bytemanHome) {
            this.bytemanHome = bytemanHome;
            return this;
        }

        public BytemanRuleSubmitter build() {
            return new BytemanRuleSubmitter(this);
        }
    }
}