org.eclipse.gmf.tests.runtime.diagram.ui/src/org/eclipse/gmf/tests/runtime/diagram/ui/commands/SemanticCreateCommandTest.java

/******************************************************************************
 * Copyright (c) 2004, 2006 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.gmf.tests.runtime.diagram.ui.commands;

import junit.framework.Test;
import junit.framework.TestCase;
import junit.framework.TestSuite;
import junit.textui.TestRunner;

import org.eclipse.core.commands.ExecutionException;
import org.eclipse.core.commands.operations.IUndoContext;
import org.eclipse.core.commands.operations.UndoContext;
import org.eclipse.core.runtime.IAdaptable;
import org.eclipse.core.runtime.IProgressMonitor;
import org.eclipse.core.runtime.NullProgressMonitor;
import org.eclipse.emf.transaction.TransactionalEditingDomain;
import org.eclipse.gef.commands.Command;
import org.eclipse.gef.commands.CompoundCommand;
import org.eclipse.gmf.runtime.common.core.command.AbstractCommand;
import org.eclipse.gmf.runtime.common.core.command.CommandResult;
import org.eclipse.gmf.runtime.common.core.command.ICommand;
import org.eclipse.gmf.runtime.diagram.core.edithelpers.CreateElementRequestAdapter;
import org.eclipse.gmf.runtime.diagram.ui.commands.ICommandProxy;
import org.eclipse.gmf.runtime.diagram.ui.commands.SemanticCreateCommand;
import org.eclipse.gmf.runtime.emf.commands.core.command.CompositeTransactionalCommand;
import org.eclipse.gmf.runtime.emf.type.core.requests.CreateElementRequest;

/**
 * @author ldamus
 */
public class SemanticCreateCommandTest
	extends TestCase {

	private SemanticCreateCommand fixture;

	public SemanticCreateCommandTest(String name) {
		super(name);
	}

	public static void main(String[] args) {
		TestRunner.run(suite());
	}

	public static Test suite() {
		return new TestSuite(SemanticCreateCommandTest.class);
	}

	protected SemanticCreateCommand getFixture() {
		return fixture;
	}

	protected void setFixture(SemanticCreateCommand fixture) {
		this.fixture = fixture;
	}

	public void test_wrapCompositeModelCommand() {

		String commandLabel = "test_wrapCompositeModelCommand"; //$NON-NLS-1$

		// Create:
		// ICommandProxy(CompositeModelCommand(AbstractCommand2))
		ICommand command = new AbstractCommand(commandLabel, null) {

			protected CommandResult doExecuteWithResult(
                    IProgressMonitor progressMonitor, IAdaptable info)
                throws ExecutionException {

                return CommandResult.newOKCommandResult();
            };
            
            protected CommandResult doRedoWithResult(IProgressMonitor progressMonitor, IAdaptable info)
                throws ExecutionException {

                return null;
            }
            
            protected CommandResult doUndoWithResult(IProgressMonitor progressMonitor, IAdaptable info)
                throws ExecutionException {

                return null;
            }
		};

        TransactionalEditingDomain editingDomain = TransactionalEditingDomain.Factory.INSTANCE
            .createEditingDomain();
		CompositeTransactionalCommand compositeModelCommand = new CompositeTransactionalCommand(editingDomain, 
			commandLabel);
        
		compositeModelCommand.compose(command);
		ICommandProxy proxyCommand = new ICommandProxy(
			compositeModelCommand);

		// Now wrap this in a compound command
		CompoundCommand compoundCommand = new CompoundCommand();
		compoundCommand.add(proxyCommand);

		// Create the test fixture
		CreateElementRequest createRequest = new CreateElementRequest(
            editingDomain, null, null);
		CreateElementRequestAdapter requestAdapter = new CreateElementRequestAdapter(
			createRequest);
		
		setFixture(new SemanticCreateCommand(requestAdapter, compoundCommand));
		
		// Execute the test fixture
        try {
            getFixture().execute(new NullProgressMonitor(), null);
        } catch (ExecutionException e) {
            fail(e.getLocalizedMessage());
        }
		
		CommandResult result = getFixture().getCommandResult();
		assertTrue(result.getStatus().isOK());
		
		// Should return the request adapter
		assertSame(requestAdapter, result.getReturnValue());
	}
	
	/**
	 * Verifies that contexts in the real semantic command are propagated to the
	 * SemanticCreateCommand wrapper when it is created.
	 */
	public void test_contextPropagation_141122() {

		final IUndoContext contextA = new UndoContext();
		final IUndoContext contextB = new UndoContext();
		final IUndoContext contextC = new UndoContext();
		
		// create an ICommand
		ICommand iCommand = new AbstractCommand(
				"test_contextPropagation_141122") { //$NON-NLS-1$
			protected CommandResult doExecuteWithResult(
					IProgressMonitor progressMonitor, IAdaptable info)
					throws ExecutionException {
				
				// change my contexts
				removeContext(contextB);
				addContext(contextC);
				
				return CommandResult.newOKCommandResult();
			}

			protected CommandResult doRedoWithResult(
					IProgressMonitor progressMonitor, IAdaptable info)
					throws ExecutionException {
				return CommandResult.newOKCommandResult();
			}

			protected CommandResult doUndoWithResult(
					IProgressMonitor progressMonitor, IAdaptable info)
					throws ExecutionException {
				return CommandResult.newOKCommandResult();
			}
		};

		// add two contexts to the ICommand
		iCommand.addContext(contextA);
		iCommand.addContext(contextB);

		// wrap the ICommand in an ICommandProxy
		Command command = new ICommandProxy(iCommand);

		// Create the test fixture
		TransactionalEditingDomain editingDomain = TransactionalEditingDomain.Factory.INSTANCE
				.createEditingDomain();

		CreateElementRequest createRequest = new CreateElementRequest(
				editingDomain, null, null);

		CreateElementRequestAdapter requestAdapter = new CreateElementRequestAdapter(
				createRequest);

		SemanticCreateCommand semanticCreateCommand = new SemanticCreateCommand(
				requestAdapter, command);

		// verify that both contexts have been propagated to the semanticCreateCommand fixture
		assertTrue(semanticCreateCommand.hasContext(contextA));
		assertTrue(semanticCreateCommand.hasContext(contextB));
		
		// execute removes contextB and adds contextC
		try {
			semanticCreateCommand.execute(new NullProgressMonitor(), null);
        } catch (ExecutionException e) {
            fail(e.getLocalizedMessage());
        }
        
        assertTrue(semanticCreateCommand.hasContext(contextA));
		assertFalse(semanticCreateCommand.hasContext(contextB));
		assertTrue(semanticCreateCommand.hasContext(contextC));
	}
}