GradleUnitTestFinder.java

package org.netbeans.gradle.project.java.query;

import java.io.File;
import java.net.URL;
import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
import org.jtrim.utils.ExceptionHelper;
import org.netbeans.gradle.model.java.JavaSourceGroup;
import org.netbeans.gradle.model.java.JavaSourceSet;
import org.netbeans.gradle.project.java.JavaExtension;
import org.netbeans.gradle.project.java.model.NbJavaModel;
import org.netbeans.gradle.project.java.model.NbJavaModule;
import org.netbeans.gradle.project.util.UrlFactory;
import org.netbeans.spi.java.queries.MultipleRootsUnitTestForSourceQueryImplementation;
import org.openide.filesystems.FileObject;
import org.openide.filesystems.FileUtil;

public final class GradleUnitTestFinder implements MultipleRootsUnitTestForSourceQueryImplementation {
    private final JavaExtension javaExt;

    public GradleUnitTestFinder(JavaExtension javaExt) {
        ExceptionHelper.checkNotNullArgument(javaExt, "javaExt");
        this.javaExt = javaExt;
    }

    private static boolean hasSource(NbJavaModule module, FileObject source) {
        for (JavaSourceSet sourceSet: module.getNonTestSourceSets()) {
            for (JavaSourceGroup sourceGroup: sourceSet.getSourceGroups()) {
                for (File sourceRoot: sourceGroup.getSourceRoots()) {
                    FileObject sourceRootObj = FileUtil.toFileObject(sourceRoot);
                    if (sourceRootObj != null && FileUtil.getRelativePath(sourceRootObj, source) != null) {
                        return true;
                    }
                }
            }
        }

        return false;
    }

    private static URL[] urlsFromSourceSets(Collection<JavaSourceSet> sourceSets) {
        UrlFactory urlFactory = UrlFactory.getDefaultArchiveOrDirFactory();

        List<URL> result = new ArrayList<>();
        for (JavaSourceSet sourceSet: sourceSets) {
            for (JavaSourceGroup sourceGroup: sourceSet.getSourceGroups()) {
                for (File sourceRoot: sourceGroup.getSourceRoots()) {
                    URL url = urlFactory.toUrl(sourceRoot);
                    if (url != null) {
                        result.add(url);
                    }
                }
            }
        }

        return result.toArray(new URL[result.size()]);
    }

    private static URL[] getSourceRoots(NbJavaModule module) {
        return urlsFromSourceSets(module.getNonTestSourceSets());
    }

    private static URL[] getTestRoots(NbJavaModule module) {
        return urlsFromSourceSets(module.getTestSourceSets());
    }

    @Override
    public URL[] findUnitTests(FileObject source) {
        NbJavaModel projectModel = javaExt.getCurrentModel();

        NbJavaModule mainModule = projectModel.getMainModule();
        if (hasSource(mainModule, source)) {
            return getTestRoots(mainModule);
        }

        return null;
    }

    @Override
    public URL[] findSources(FileObject unitTest) {
        NbJavaModel projectModel = javaExt.getCurrentModel();

        NbJavaModule mainModule = projectModel.getMainModule();
        if (!hasSource(mainModule, unitTest)) {
            return getSourceRoots(mainModule);
        }

        return null;
    }
}