org.babyfish.lang.instrument.TestInstrumentMojo Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of babyfish-instrument-maven-plugin Show documentation
This tool project supports some maven plugins of BabyFish, it shouldn't used by the user directly.
The newest version!
/*
 * BabyFish, Object Model Framework for Java and JPA.
 * https://github.com/babyfish-ct/babyfish
 *
 * Copyright (c) 2008-2016, Tao Chen
 *
 * This copyrighted material is made available to anyone wishing to use, modify,
 * copy, or redistribute it subject to the terms and conditions of the GNU
 * Lesser General Public License, as published by the Free Software Foundation.
 *
 * Please visit "http://opensource.org/licenses/LGPL-3.0" to know more.
 *
 * 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 Lesser General Public License
 * for more details.
 */
package org.babyfish.lang.instrument;

import java.io.File;
import java.net.URL;
import java.net.URLClassLoader;
import java.util.ArrayList;
import java.util.List;
import java.util.Set;

import org.apache.maven.artifact.Artifact;
import org.apache.maven.plugin.MojoExecutionException;
import org.apache.maven.plugin.MojoFailureException;
import org.apache.maven.plugins.annotations.Execute;
import org.apache.maven.plugins.annotations.LifecyclePhase;
import org.apache.maven.plugins.annotations.Mojo;
import org.apache.maven.plugins.annotations.Parameter;
import org.apache.maven.plugins.annotations.ResolutionScope;

/**
 * @author Tao Chen
 */
@Mojo(
        name = "instrument-test",
        defaultPhase = LifecyclePhase.PROCESS_TEST_CLASSES,
        requiresProject = false,
        threadSafe = true,
        executionStrategy = "always",
        requiresDirectInvocation = false,
        requiresDependencyResolution = ResolutionScope.TEST
)
@Execute(
        goal = "instrument-test", 
        phase = LifecyclePhase.PROCESS_TEST_CLASSES
)
public class TestInstrumentMojo extends AbstractInstrumentMojo {

    @Parameter(property = "maven.test.skip")
    private boolean skip;
    
    @Parameter(defaultValue = "${project.build.testOutputDirectory}", required = true)
    private File testClassesDirectory;
    
    @Override
    public void execute() throws MojoExecutionException, MojoFailureException {
        if (this.skip) {
            this.getLog().info("Skipping the instrumentation of the test classes");
        } else {
            super.doExecute();
        }
    }

    @Override
    protected File getInstrumentDirectory() {
        return this.testClassesDirectory;
    }

    @Override
    protected String getType() {
        return "test-classes";
    }
    
    @Override
    protected ClassLoader createClassLoader() {
        URL[] urls = generateTestClasspaths().stream().map(
                AbstractInstrumentMojo::fileNameToURL
        ).toArray(len -> new URL[len]);
        return new URLClassLoader(urls);
    }

    private List generateTestClasspaths() {
        Set artifacts = this.project.getArtifacts();
        List classpaths = new ArrayList<>(2 + artifacts.size());
        classpaths.add(this.testClassesDirectory.getAbsolutePath());
        classpaths.add(this.classesDirectory.getAbsolutePath());
        Set classpathArtifacts = artifacts;
        for (Artifact artifact : classpathArtifacts) {
            if (artifact.getArtifactHandler().isAddedToClasspath()) {
                File file = artifact.getFile();
                if (file != null) {
                    classpaths.add(file.getPath());
                }
            }
        }
        return classpaths;
    }
}