eu.stamp.project.assertfixer.Main Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of assert-fixer Show documentation
This tool aims at repairing the assertion inside JUnit tests.
There is a newer version: 1.0.10
package eu.stamp.project.assertfixer;

import eu.stamp.project.assertfixer.asserts.AssertFixer;
import eu.stamp.project.assertfixer.test.TestRunner;
import eu.stamp.project.assertfixer.util.Util;
import eu.stamp.project.testrunner.EntryPoint;
import eu.stamp.project.testrunner.runner.test.Failure;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import spoon.Launcher;
import spoon.SpoonModelBuilder;

import java.io.File;

/**
 * Created by Benjamin DANGLOT
 * [email protected]
 * on 20/03/18
 */
public class Main {

    static final Logger LOGGER = LoggerFactory.getLogger(Main.class);

    public static Configuration configuration;

    public static void main(String[] args) {
        System.exit(run(args));
    }

    public static int run(String [] args) {
        Main.configuration = Configuration.get(args);
        EntryPoint.verbose = Main.configuration.verbose;

        Launcher launcher = new Launcher();
        launcher.getEnvironment().setAutoImports(true);
        launcher.getEnvironment().setNoClasspath(true);
        launcher.getEnvironment().setSourceClasspath(configuration.classpath.split(Util.PATH_SEPARATOR));
        launcher.addInputResource(configuration.pathToSourceFolder);
        launcher.addInputResource(configuration.pathToTestFolder);
        launcher.getEnvironment().setShouldCompile(true);
        launcher.run();

        final Boolean result = Main.configuration.failingTestMethods.stream()
                .map(failingTestMethod -> fixGivenTest(launcher, failingTestMethod))
                .reduce(Boolean.TRUE, Boolean::logicalAnd);
        if (result) {
            return 1;
        } else {
            return 0;
        }
    }

    private static boolean fixGivenTest(Launcher launcher, String failingTestMethod) {
        Failure failure = TestRunner.runTest(launcher, failingTestMethod).getFailingTests().get(0);
        LOGGER.info("Fixing: {}", failure.messageOfFailure);
        try {
            AssertFixer.fixAssert(
                    launcher,
                    configuration.fullQualifiedFailingTestClass,
                    failingTestMethod,
                    failure,
                    configuration.classpath
            );
            final SpoonModelBuilder compiler = launcher.createCompiler();
            compiler.setBinaryOutputDirectory(new File(Main.configuration.output));
            return EntryPoint.runTests(
                    Main.configuration.getBinaryOutputDirectory() +
                            Util.PATH_SEPARATOR + configuration.classpath,
                    configuration.fullQualifiedFailingTestClass,
                    failingTestMethod
            ).getFailingTests().isEmpty();
        } catch (Exception e) {
            return false;
        }
    }
}