tests/org.eclipse.jst.j2ee.tests/j2ee-tests/org/eclipse/jst/jee/model/web/tests/Web25MergedModelProviderTest.java

/***********************************************************************
 * Copyright (c) 2008 by SAP AG, Walldorf. 
 * 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
 * https://www.eclipse.org/legal/epl-2.0/
 *
 * Contributors:
 *     SAP AG - initial API and implementation
 ***********************************************************************/
package org.eclipse.jst.jee.model.web.tests;

import junit.framework.TestCase;
import junit.framework.TestSuite;

import org.eclipse.core.commands.ExecutionException;
import org.eclipse.core.resources.IFile;
import org.eclipse.core.resources.IFolder;
import org.eclipse.core.resources.IProject;
import org.eclipse.core.resources.ResourcesPlugin;
import org.eclipse.core.runtime.NullProgressMonitor;
import org.eclipse.jdt.core.IJavaProject;
import org.eclipse.jdt.core.IPackageFragment;
import org.eclipse.jdt.core.IPackageFragmentRoot;
import org.eclipse.jdt.core.JavaCore;
import org.eclipse.jst.j2ee.internal.web.operations.INewServletClassDataModelProperties;
import org.eclipse.jst.j2ee.internal.web.operations.NewServletClassDataModelProvider;
import org.eclipse.jst.j2ee.model.IModelProvider;
import org.eclipse.jst.j2ee.model.ModelProviderManager;
import org.eclipse.jst.javaee.web.WebApp;
import org.eclipse.jst.jee.model.tests.AbstractTest;
import org.eclipse.jst.jee.model.tests.SynchronousModelChangedListener;
import org.eclipse.jst.jee.model.tests.TestUtils;
import org.eclipse.wst.common.frameworks.datamodel.DataModelFactory;
import org.eclipse.wst.common.frameworks.datamodel.IDataModel;
import org.eclipse.wst.common.frameworks.datamodel.IDataModelOperation;
import org.eclipse.wst.common.project.facet.core.IFacetedProject;

/**
 * @author Kiril Mitov k.mitov@sap.com
 * 
 */
public class Web25MergedModelProviderTest extends TestCase {

	private static IFacetedProject facetedProject;

	private IModelProvider fixture;

	public static TestSuite suite() throws Exception {
		TestSuite suite = new TestSuite(Web25MergedModelProviderTest.class);
		return suite;
	}

	// @Before
	@Override
	protected void setUp() throws Exception {
		setUpProject();
		fixture = ModelProviderManager.getModelProvider(facetedProject.getProject());
	}

	protected void tearDown() throws Exception {
		deleteProjectAndWait(facetedProject.getProject().getName());
	}

	private void deleteProjectAndWait(String projectName) throws InterruptedException {
		SynchronousModelChangedListener listener = new SynchronousModelChangedListener(1);
		fixture.addListener(listener);
		AbstractTest.deleteProject(projectName);
		assertTrue(listener.waitForEvents());
		fixture.removeListener(listener);
	}

	// @BeforeClass
	public void setUpProject() throws Exception {
		IProject project = ResourcesPlugin.getWorkspace().getRoot().getProject(
				Web25MergedModelProviderTest.class.getSimpleName() + this.getName());
		facetedProject = AbstractTest.createWebProject(project.getName());
		createProjectContent();
	}

	private static void createProjectContent() throws Exception {
		IJavaProject javaProject = JavaCore.create(facetedProject.getProject());
		IFolder comFolder = javaProject.getProject().getFolder("src/com");
		comFolder.create(true, true, new NullProgressMonitor());
		IPackageFragmentRoot root = javaProject.getPackageFragmentRoot(comFolder);
		IPackageFragment fragment = root.createPackageFragment("sap", true, new NullProgressMonitor());
	}

	// @Test
	public void testAddServletWithOperation() throws Exception {
		IDataModel dataModel = createAddServletModel("testAddServletWithOperation");
		executeAndWait(dataModel.getDefaultOperation());

		WebApp app = (WebApp) fixture.getModelObject();
		assertNotNull(TestUtils.findServletByName(app, "testAddServletWithOperation"));
	}

	public void testAddServletFromJSP() throws Exception {
		AbstractTest.saveFile(facetedProject.getProject().getFile("WebContent/index.jsp"), "");
		IDataModel dataModel = createAddServletFromJSPModel("index.jsp", "index");
		executeAndWait(dataModel.getDefaultOperation());

		WebApp app = (WebApp) fixture.getModelObject();
		assertNotNull(TestUtils.findServletByName(app, "index"));
	}

	// public void testPreserveListeners() throws Exception {
	// IDataModel dataModel = createAddServletModel("testPreserveListeners");
	//
	// SynchronousModelChangedListener preserveListener = new
	// SynchronousModelChangedListener(2);
	// fixture.addListener(preserveListener);
	// IFile webXml =
	// facetedProject.getProject().getFile("WebContent/WEB-INF/web.xml");
	// String content = TestUtils.getFileContent(webXml);
	//
	// saveFileAndUpdate(webXml, content);
	//
	// int oldEventSize = preserveListener.getReceivedEvents().size();
	// executeAndWait(dataModel.getDefaultOperation());
	// WebApp app = (WebApp) fixture.getModelObject();
	// assertNotNull(TestUtils.findServletByName(app, "testPreserveListeners"));
	// assertTrue(preserveListener.waitForEvents());
	// assertTrue(oldEventSize < preserveListener.getReceivedEvents().size());
	// fixture.removeListener(preserveListener);
	// }

	private void saveFileAndUpdate(IFile webXml, String content) throws InterruptedException, Exception {
		SynchronousModelChangedListener listener = new SynchronousModelChangedListener(1);
		fixture.addListener(listener);
		AbstractTest.saveFile(webXml, content);
		assertTrue(listener.waitForEvents());
		fixture.removeListener(listener);
	}

	private IDataModel createAddServletFromJSPModel(String className, String dispayName) {
		IDataModel dataModel;
		dataModel = DataModelFactory.createDataModel(new NewServletClassDataModelProvider());
		dataModel.setStringProperty(INewServletClassDataModelProperties.CLASS_NAME, className);
		dataModel.setStringProperty(INewServletClassDataModelProperties.DISPLAY_NAME, dispayName);
		dataModel.setStringProperty(INewServletClassDataModelProperties.PROJECT_NAME, facetedProject.getProject()
				.getName());
		dataModel.setBooleanProperty(INewServletClassDataModelProperties.USE_EXISTING_CLASS, Boolean.TRUE);
		dataModel.setBooleanProperty(INewServletClassDataModelProperties.IS_SERVLET_TYPE, Boolean.FALSE);
		return dataModel;
	}

	private IDataModel createAddServletModel(String className) {
		IDataModel dataModel = DataModelFactory.createDataModel(new NewServletClassDataModelProvider());
		dataModel.setStringProperty(INewServletClassDataModelProperties.CLASS_NAME, className);
		dataModel.setStringProperty(INewServletClassDataModelProperties.PROJECT_NAME, facetedProject.getProject()
				.getName());
		dataModel.setBooleanProperty(INewServletClassDataModelProperties.IS_SERVLET_TYPE, Boolean.TRUE);
		dataModel.setStringProperty(INewServletClassDataModelProperties.JAVA_PACKAGE, "org.eclipse");
		return dataModel;
	}

	private void executeAndWait(IDataModelOperation dataModelOperation) throws InterruptedException, ExecutionException {
		SynchronousModelChangedListener listener = new SynchronousModelChangedListener(1);
		fixture.addListener(listener);
		dataModelOperation.execute(null, null);
		listener.waitForEvents();
		fixture.removeListener(listener);
	}

}