com.github.junitrunner.unitils.UnitilsPlugin Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of junitrunner-unitils Show documentation

Fully functional Unitils plugin for JUnitRunner

The newest version!

package com.github.junitrunner.unitils;

import org.junit.runners.model.Statement;

import com.github.junitrunner.JUnitBasePlugin;
import com.github.junitrunner.JUnitRunner;
import com.github.junitrunner.JUnitTest;
import com.github.junitrunner.javamethod.JUnitJavaMethodTest;

public class UnitilsPlugin extends JUnitBasePlugin {

    public UnitilsPlugin(@SuppressWarnings("unused") Class testClass) {
    }

    @Override
    public void plug(JUnitRunner jUnitRunner) {

        jUnitRunner.registerListener(new UnitilsRunnerListener());
    }

    @Override
    public Statement enhanceTest(JUnitTest test, Object testObject, Statement statement) {
        if (test instanceof JUnitJavaMethodTest) {
            return new UnitilsTestInterceptor((JUnitJavaMethodTest) test, testObject, statement);
        } else {
            return statement;
        }
    }
}