ExtendedJUnitLaunchConfigurationCreator.java

package org.projectusus.autotestsuite.core.internal.config;

import static java.util.Arrays.asList;
import static org.eclipse.jdt.ui.JavaElementLabels.ALL_FULLY_QUALIFIED;
import static org.projectusus.autotestsuite.core.internal.config.ExtendedJUnitLaunchConfigurationConstants.getLaunchConfigurationType;

import java.util.Collection;

import org.eclipse.core.runtime.CoreException;
import org.eclipse.debug.core.ILaunchConfiguration;
import org.eclipse.debug.core.ILaunchConfigurationWorkingCopy;
import org.eclipse.jdt.core.IJavaProject;
import org.eclipse.jdt.internal.junit.launcher.AssertionVMArg;
import org.eclipse.jdt.internal.junit.launcher.JUnitMigrationDelegate;
import org.eclipse.jdt.ui.JavaElementLabels;
import org.projectusus.autotestsuite.core.internal.AllJavaProjectsInWorkspace;

@SuppressWarnings( "restriction" )
public class ExtendedJUnitLaunchConfigurationCreator {

    public ILaunchConfiguration createAndSaveNewConfig( IJavaProject project ) throws CoreException {
        return createNewConfig( project ).doSave();
    }

    public ILaunchConfigurationWorkingCopy createNewConfig( IJavaProject root ) throws CoreException {
        return createNewConfig( root, asList( new AllJavaProjectsInWorkspace().find() ) );
    }

    public ILaunchConfigurationWorkingCopy createNewConfig( IJavaProject root, Collection<IJavaProject> projects ) throws CoreException {
        String name = JavaElementLabels.getTextLabel( root, ALL_FULLY_QUALIFIED );
        // TODO fix needs 3.6
        String uniqueName = ExtendedJUnitLaunchConfigurationConstants.getLaunchManager().generateUniqueLaunchConfigurationNameFrom( name );
        ILaunchConfigurationWorkingCopy config = getLaunchConfigurationType().newInstance( null, uniqueName );
        return fillAttributes( root, config, projects );
    }

    private ILaunchConfigurationWorkingCopy fillAttributes( IJavaProject project, ILaunchConfigurationWorkingCopy config, Collection<IJavaProject> projects ) throws CoreException {
        ExtendedJUnitLaunchConfigurationWriter writer = new ExtendedJUnitLaunchConfigurationWriter( config );
        writer.setUnusedAttributesToDefaults();
        writer.setProjectName( project.getElementName() );
        writer.setDefaultTestKind();
        writer.setCheckedProjects( projects );
        writer.setKeepRunning( false );
        writer.setTestContainer( project.getElementName() );
        JUnitMigrationDelegate.mapResources( config );
        AssertionVMArg.setArgDefault( config );
        return config;
    }
}