org.eclipse.jdt.ui.tests/ui/org/eclipse/jdt/junit/tests/JUnitWorkspaceTestSetup.java

/*******************************************************************************
 * Copyright (c) 2007, 2010 IBM Corporation and others.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *     IBM Corporation - initial API and implementation
 *******************************************************************************/
package org.eclipse.jdt.junit.tests;

import junit.extensions.TestSetup;
import junit.framework.Test;

import org.eclipse.jdt.junit.JUnitCore;
import org.eclipse.jdt.testplugin.JavaProjectHelper;
import org.eclipse.jdt.testplugin.JavaTestPlugin;

import org.eclipse.core.resources.IFolder;

import org.eclipse.jdt.core.IClasspathEntry;
import org.eclipse.jdt.core.IJavaProject;
import org.eclipse.jdt.core.IPackageFragmentRoot;
import org.eclipse.jdt.core.JavaCore;


public class JUnitWorkspaceTestSetup extends TestSetup {

	public static final String WORKSPACE_PATH= "testresources/JUnitWorkspace/";

	private static final String PROJECT_NAME_3= "JUnitTests";
	private static final String PROJECT_NAME_4= "JUnit4Tests";
	private static final String SRC_NAME= "src";

	private boolean fJUnit4;
	private static IJavaProject fgProject;
	private static IPackageFragmentRoot fgRoot;

	public JUnitWorkspaceTestSetup(Test test, boolean jUnit4) {
		super(test);
		fJUnit4= jUnit4;
	}

	public static IJavaProject getJavaProject() {
		return fgProject;
	}

	public static IPackageFragmentRoot getRoot() {
		return fgRoot;
	}

	public static String getProjectPath() {
		return WORKSPACE_PATH + fgProject.getElementName() + '/';
	}

	@Override
	protected void setUp() throws Exception {
		if (fJUnit4) {
			fgProject= JavaProjectHelper.createJavaProject(PROJECT_NAME_4, "bin");
			JavaProjectHelper.addRTJar(fgProject);
			IClasspathEntry cpe= JavaCore.newContainerEntry(JUnitCore.JUNIT4_CONTAINER_PATH);
			JavaProjectHelper.addToClasspath(fgProject, cpe);

		} else {
			fgProject= JavaProjectHelper.createJavaProject(PROJECT_NAME_3, "bin");
			JavaProjectHelper.addRTJar13(fgProject);
			IClasspathEntry cpe= JavaCore.newContainerEntry(JUnitCore.JUNIT3_CONTAINER_PATH);
			JavaProjectHelper.addToClasspath(fgProject, cpe);
		}
		fgRoot= JavaProjectHelper.addSourceContainer(fgProject, SRC_NAME);
		JavaProjectHelper.importResources((IFolder) fgRoot.getResource(), JavaTestPlugin.getDefault().getBundle(), getProjectPath() + SRC_NAME);
	}

	@Override
	protected void tearDown() throws Exception {
		JavaProjectHelper.delete(fgProject);
		fgProject= null;
		fgRoot= null;
	}

}