/*******************************************************************************
 * <copyright>
 *
 * Copyright (c) 2005, 2013 SAP AG.
 * 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:
 *    SAP AG - initial API, implementation and documentation
 *    mwenz - Bug 355027: Move of connection decorators when zoom level != 100 behaves weird
 *    Felix Velasco (mwenz) - Bug 323351 - Enable to suppress/reactivate the speed buttons
 *    mwenz - Bug 341224: Allow to hide the selection and marquee tools in the palette
 *    Bug 336488 - DiagramEditor API
 *    mwenz - Bug 363796 - Make setting of selection width of connections public
 *    mwenz - Bug 358255 - Add Border/Background decorators
 *    Benjamin Schmeling - mwenz - Bug 367483 - Support composite connections
 *    mwenz - Bug 396793 - Text decorators
 *    pjpaulin - Bug 352120 - Now uses IDiagramContainerUI interface
 *
 * </copyright>
 *
 *******************************************************************************/
package org.eclipse.graphiti.bot.tests;

import static org.eclipse.swtbot.swt.finder.finders.UIThreadRunnable.syncExec;
import static org.hamcrest.CoreMatchers.instanceOf;
import static org.hamcrest.Matchers.greaterThan;
import static org.junit.Assert.assertThat;

import java.awt.AWTException;
import java.awt.Robot;
import java.awt.event.InputEvent;
import java.awt.event.KeyEvent;
import java.lang.reflect.Field;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;

import org.eclipse.draw2d.FigureCanvas;
import org.eclipse.draw2d.IFigure;
import org.eclipse.draw2d.Label;
import org.eclipse.draw2d.LineBorder;
import org.eclipse.draw2d.geometry.Rectangle;
import org.eclipse.gef.GraphicalEditPart;
import org.eclipse.gef.GraphicalViewer;
import org.eclipse.gef.LayerConstants;
import org.eclipse.gef.commands.Command;
import org.eclipse.gef.commands.CommandStack;
import org.eclipse.gef.editparts.ScalableFreeformRootEditPart;
import org.eclipse.gef.palette.PaletteEntry;
import org.eclipse.gef.palette.ToolEntry;
import org.eclipse.graphiti.DiagramScrollingBehavior;
import org.eclipse.graphiti.bot.tests.util.ITestConstants;
import org.eclipse.graphiti.dt.IDiagramTypeProvider;
import org.eclipse.graphiti.features.DefaultFeatureProviderWrapper;
import org.eclipse.graphiti.features.ICreateConnectionFeature;
import org.eclipse.graphiti.features.ICreateFeature;
import org.eclipse.graphiti.features.IFeatureProvider;
import org.eclipse.graphiti.features.context.ICreateContext;
import org.eclipse.graphiti.features.context.impl.CreateConnectionContext;
import org.eclipse.graphiti.features.context.impl.CustomContext;
import org.eclipse.graphiti.internal.command.CommandContainer;
import org.eclipse.graphiti.internal.command.GenericFeatureCommandWithContext;
import org.eclipse.graphiti.mm.algorithms.GraphicsAlgorithm;
import org.eclipse.graphiti.mm.algorithms.MultiText;
import org.eclipse.graphiti.mm.algorithms.RoundedRectangle;
import org.eclipse.graphiti.mm.pictograms.Anchor;
import org.eclipse.graphiti.mm.pictograms.ConnectionDecorator;
import org.eclipse.graphiti.mm.pictograms.ContainerShape;
import org.eclipse.graphiti.mm.pictograms.Diagram;
import org.eclipse.graphiti.mm.pictograms.PictogramElement;
import org.eclipse.graphiti.mm.pictograms.Shape;
import org.eclipse.graphiti.services.Graphiti;
import org.eclipse.graphiti.services.ICreateService;
import org.eclipse.graphiti.testtool.ecore.TestToolBehavior;
import org.eclipse.graphiti.testtool.sketch.SketchFeatureProvider;
import org.eclipse.graphiti.testtool.sketch.features.DisplayDecoratorFeature;
import org.eclipse.graphiti.testtool.sketch.features.DoubleClickFeature;
import org.eclipse.graphiti.testtool.sketch.features.ToggleDecorator;
import org.eclipse.graphiti.testtool.sketch.features.create.SketchCreateCompositeConnectionFeature;
import org.eclipse.graphiti.testtool.sketch.features.create.SketchCreateCurvedConnectionFeature;
import org.eclipse.graphiti.testtool.sketch.features.create.SketchCreateFreeformConnectionFeature;
import org.eclipse.graphiti.testtool.sketch.features.create.SketchCreateGaContainerFeature;
import org.eclipse.graphiti.testtool.sketch.features.create.SketchCreateGaShapeFeature;
import org.eclipse.graphiti.ui.editor.IDiagramContainerUI;
import org.eclipse.graphiti.ui.internal.command.CreateModelObjectCommand;
import org.eclipse.graphiti.ui.internal.command.GefCommandWrapper;
import org.eclipse.graphiti.ui.internal.contextbuttons.ContextButtonPad;
import org.eclipse.graphiti.ui.internal.editor.GFFigureCanvas;
import org.eclipse.graphiti.ui.internal.figures.DecoratorImageFigure;
import org.eclipse.graphiti.ui.internal.parts.ContainerShapeEditPart;
import org.eclipse.graphiti.ui.internal.parts.IPictogramElementDelegate;
import org.eclipse.graphiti.ui.internal.services.GraphitiUiInternal;
import org.eclipse.graphiti.util.IColorConstant;
import org.eclipse.swt.SWT;
import org.eclipse.swt.graphics.Color;
import org.eclipse.swt.graphics.Point;
import org.eclipse.swt.widgets.Control;
import org.eclipse.swt.widgets.Display;
import org.eclipse.swt.widgets.Event;
import org.eclipse.swtbot.eclipse.finder.SWTWorkbenchBot;
import org.eclipse.swtbot.eclipse.gef.finder.finders.PaletteFinder;
import org.eclipse.swtbot.eclipse.gef.finder.matchers.AbstractToolEntryMatcher;
import org.eclipse.swtbot.eclipse.gef.finder.widgets.SWTBotGefEditPart;
import org.eclipse.swtbot.swt.finder.exceptions.WidgetNotFoundException;
import org.eclipse.swtbot.swt.finder.results.VoidResult;
import org.eclipse.swtbot.swt.finder.widgets.SWTBotMenu;
import org.eclipse.ui.IEditorPart;
import org.eclipse.ui.IEditorReference;
import org.hamcrest.Description;
import org.junit.Test;

@SuppressWarnings("restriction")
public class GFInteractionComponentTests extends AbstractGFTests {

	private static final int SHORT_DELAY = 500;
	private static final int DELAY = 1000;
	/**
	 * 
	 */
	private static final String SHAPE_NAME = "Connection";
	/**
	 * 
	 */
	private static final String INPUT_STRING = "Graphiti Rulez";

	public GFInteractionComponentTests() {
		super();
	}

	@Test
	public void testSelectionTool() throws Exception {
		final int x = 100;
		final int y = 100;
		final IDiagramContainerUI diagramEditor = openDiagramEditor(ITestConstants.DIAGRAM_TYPE_ID_ECORE);
		createClassesAndConnection(x, y, diagramEditor.getDiagramTypeProvider(), null, SHAPE_NAME);
		Thread.sleep(DELAY);
		syncExec(new VoidResult() {
			public void run() {
				ed.getGefEditor().drag(SHAPE_NAME, 200, 50);
				ed.getGefEditor().drag("ConnectionDecorator", 200, 400);
			}
		});
		Thread.sleep(DELAY);
		page.shutdownEditor(diagramEditor);
	}

	@Test
	public void testHidePalette() throws Exception {
		boolean oldValue = TestToolBehavior.showFlyoutPalette;
		TestToolBehavior.setShowFlyoutPalette(true);
		IDiagramContainerUI diagramEditor = openDiagramEditor(ITestConstants.DIAGRAM_TYPE_ID_ECORE);
		assertNotNull(diagramEditor.getEditDomain().getPaletteViewer());
		page.shutdownEditor(diagramEditor);
		TestToolBehavior.setShowFlyoutPalette(false);
		diagramEditor = openDiagramEditor(ITestConstants.DIAGRAM_TYPE_ID_ECORE);
		assertNull(diagramEditor.getEditDomain().getPaletteViewer());
		page.shutdownEditor(diagramEditor);
		TestToolBehavior.setShowFlyoutPalette(oldValue);
	}

	@Test
	public void testLineSelectionWidth() throws Exception {
		final IDiagramContainerUI diagramEditor = openDiagramEditor(ITestConstants.DIAGRAM_TYPE_ID_ECORE);
		final ICreateService createService = Graphiti.getCreateService();
		executeInRecordingCommand(diagramEditor.getDiagramBehavior(), new Runnable() {
			public void run() {
				ContainerShape containerShape = createService.createContainerShape(diagramEditor
						.getDiagramTypeProvider().getDiagram(), true);
				createService.createPolyline(containerShape, new int[] { 0, 0, 10, 0 });
			}
		});
		diagramEditor.getDiagramBehavior().refresh();
		assertTrue(TestToolBehavior.lineSelectionWidthCalled);
		page.getGefEditor().saveAndClose();
	}

	@Test
	public void testMarqueeTool() throws Exception {
		final int x = 100;
		final int y = 100;
		final IDiagramContainerUI diagramEditor = openDiagramEditor(ITestConstants.DIAGRAM_TYPE_ID_ECORE);
		createClassesAndConnection(x, y, diagramEditor.getDiagramTypeProvider(), "Marquee", SHAPE_NAME);
		Thread.sleep(DELAY);
		// Select the newly added shapes with the marquee tool.
		syncExec(new VoidResult() {
			public void run() {
				ed.drag(x - 10, y - 10, x + SHORT_DELAY, y + SHORT_DELAY);
			}
		});
		Thread.sleep(DELAY);

		syncExec(new VoidResult() {
			public void run() {
				ed.getGefEditor().activateTool("Select");
				ed.getGefEditor().drag(x + 50, y + 50, x + 200, y + 50);
			}
		});
		Thread.sleep(DELAY);
		IFigure figure = ed.getFigureWithLabel(SHAPE_NAME);
		// Drag might not be accurate, add tolerance +-1
		assertTrue((x + 149 <= figure.getBounds().x) && (figure.getBounds().x <= x + 151));
		page.shutdownEditor(diagramEditor);
	}

	@Test
	public void testHideSelectionAndMarqueeTools() throws Exception {
		openDiagramEditor(ITestConstants.DIAGRAM_TYPE_ID_WITH_HIDDEN_SELECTION_AND_MARQUEE_TOOL);
		try {
			ed.getGefEditor().activateTool("Select");
		} catch (WidgetNotFoundException eSelection) {
			// Expected this exception
			try {
				ed.getGefEditor().activateTool("Marquee");
			} catch (WidgetNotFoundException eMarquee) {
				// Expected this exception
				return;
			}
			fail("Marquee tool must not exist");
		}
		fail("Selection tool must not exist");
	}

	@Test
	public void testGFFigureCanvas() throws Exception {
		final int x = 100;
		final int y = 100;
		final IDiagramContainerUI diagramEditor = openDiagramEditor(ITestConstants.DIAGRAM_TYPE_ID_ECORE);
		syncExec(new VoidResult() {
			public void run() {
				// find diagram
				IDiagramTypeProvider diagramTypeProvider = diagramEditor.getDiagramTypeProvider();
				final IFeatureProvider fp = diagramTypeProvider.getFeatureProvider();
				final Diagram currentDiagram = diagramTypeProvider.getDiagram();
				executeInRecordingCommand(diagramEditor.getDiagramBehavior(), new Runnable() {
					public void run() {
						// add a class to the diagram
						addClassToDiagram(fp, currentDiagram, x, y, SHAPE_NAME);
					}
				});
			}
		});
		Thread.sleep(DELAY);

		syncExec(new VoidResult() {
			public void run() {
				// move class-shape to origin
				ed.getGefEditor().drag(SHAPE_NAME, -100, -100);
			}
		});
		Thread.sleep(SHORT_DELAY);

		syncExec(new VoidResult() {
			public void run() {
				// get instance of GFFigureCanvas
				GFFigureCanvas gfFigureCanvas = ed.getGFCanvas();

				// do some scrolling
				Event e = new Event();
				e.doit = true;
				e.stateMask = 0;

				e.detail = SWT.ARROW_UP;
				gfFigureCanvas.getHorizontalBar().notifyListeners(SWT.Selection, e);
				gfFigureCanvas.getVerticalBar().notifyListeners(SWT.Selection, e);

				e.detail = SWT.ARROW_DOWN;
				gfFigureCanvas.getHorizontalBar().notifyListeners(SWT.Selection, e);
				gfFigureCanvas.getVerticalBar().notifyListeners(SWT.Selection, e);

				e.detail = SWT.PAGE_UP;
				gfFigureCanvas.getHorizontalBar().notifyListeners(SWT.Selection, e);
				gfFigureCanvas.getVerticalBar().notifyListeners(SWT.Selection, e);

				e.detail = SWT.PAGE_DOWN;
				gfFigureCanvas.getHorizontalBar().notifyListeners(SWT.Selection, e);
				gfFigureCanvas.getVerticalBar().notifyListeners(SWT.Selection, e);
			}
		});
		Thread.sleep(SHORT_DELAY);

		syncExec(new VoidResult() {
			public void run() {
				Event event = new Event();
				event.type = SWT.KeyDown;
				event.keyCode = SWT.DEL;
				event.character = SWT.DEL;
				event.stateMask = 0; // SWT.SHIFT;
				Display.getDefault().post(event);
			}
		});
		Thread.sleep(SHORT_DELAY);

		syncExec(new VoidResult() {
			public void run() {
				Event event = new Event();
				event.type = SWT.KeyDown;
				event.keyCode = SWT.CR;
				event.character = SWT.CR;
				event.stateMask = 0; // SWT.SHIFT;
				Display.getDefault().post(event);
			}
		});
		Thread.sleep(SHORT_DELAY);

		syncExec(new VoidResult() {
			public void run() {
				GFFigureCanvas gfFigureCanvas = ed.getGFCanvas();
				// regain space
				gfFigureCanvas.regainSpace();
			}
		});
		Thread.sleep(SHORT_DELAY);
		page.shutdownEditor(diagramEditor);
	}

	@Test
	public void testScrolling() throws Exception {
		final int x = 100;
		final int y = 100;
		final int deltaX = 200;
		final int deltaY = 200;
		TestToolBehavior.scrollingBehavior = DiagramScrollingBehavior.GEF_DEFAULT;
		final IDiagramContainerUI diagramEditor = openDiagramEditor(ITestConstants.DIAGRAM_TYPE_ID_ECORE);

		IEditorReference reference = ed.getGefEditor().getReference();
		IEditorPart editor = reference.getEditor(true);
		GraphicalViewer graphicalViewer = (GraphicalViewer) editor.getAdapter(GraphicalViewer.class);
		final Control control = graphicalViewer.getControl();
		assertThat(control, instanceOf(FigureCanvas.class));
		final FigureCanvas canvas = (FigureCanvas) control;

		final org.eclipse.draw2d.geometry.Point[] position = new org.eclipse.draw2d.geometry.Point[1];

		syncExec(new VoidResult() {
			public void run() {
				// find diagram
				IDiagramTypeProvider diagramTypeProvider = diagramEditor.getDiagramTypeProvider();
				final IFeatureProvider fp = diagramTypeProvider.getFeatureProvider();
				final Diagram currentDiagram = diagramTypeProvider.getDiagram();
				executeInRecordingCommand(diagramEditor.getDiagramBehavior(), new Runnable() {
					public void run() {
						// add a lot of classes to the diagram
						for (int i = 0; i < 10; i++)
							for (int j = 0; j < 10; j++)
								addClassToDiagram(fp, currentDiagram, x + deltaX * i, y + deltaY * j, SHAPE_NAME + "_"
										+ i + "_" + j);
					}
				});
			}
		});
		Thread.sleep(DELAY);
		syncExec(new VoidResult() {
			public void run() {
				position[0] = canvas.getViewport().getViewLocation();
			}
		});
		org.eclipse.draw2d.geometry.Point initialPos = position[0];

		syncExec(new VoidResult() {
			public void run() {

				Robot r;
				try {
					r = new Robot();
					Point p = canvas.toDisplay(0, 0);
					r.mouseMove(p.x + 150, p.y + 150);
					r.mouseWheel(10);
				} catch (AWTException e) {
					fail(e.getMessage());
				}
			}
		});
		Thread.sleep(SHORT_DELAY);

		syncExec(new VoidResult() {
			public void run() {
				position[0] = canvas.getViewport().getViewLocation();
			}
		});
		org.eclipse.draw2d.geometry.Point currentPos = position[0];
		assertEquals(initialPos.x, currentPos.x);
		assertThat(currentPos.y, greaterThan(initialPos.y));
		initialPos = currentPos;

		syncExec(new VoidResult() {
			public void run() {

				Robot r;
				try {
					r = new Robot();
					r.keyPress(KeyEvent.VK_SHIFT);
					r.mouseWheel(-10);
					r.keyRelease(KeyEvent.VK_SHIFT);

				} catch (AWTException e) {
					fail(e.getMessage());
				}
			}
		});
		Thread.sleep(SHORT_DELAY);

		syncExec(new VoidResult() {
			public void run() {
				position[0] = canvas.getViewport().getViewLocation();
			}
		});
		currentPos = position[0];
		assertEquals(initialPos.y, currentPos.y);
		assertThat(currentPos.x, greaterThan(initialPos.x));

		// Revert scrolling behavior
		TestToolBehavior.scrollingBehavior = DiagramScrollingBehavior.SCROLLBARS_ALWAYS_VISIBLE;

		page.shutdownEditor(diagramEditor);
	}

	@Test
	public void testScrollingGFFigureCanvas() throws Exception {
		final int x = 100;
		final int y = 100;
		final int deltaX = 200;
		final int deltaY = 200;
		final IDiagramContainerUI diagramEditor = openDiagramEditor(ITestConstants.DIAGRAM_TYPE_ID_ECORE);

		final GFFigureCanvas gfFigureCanvas = ed.getGFCanvas();
		final org.eclipse.draw2d.geometry.Point[] position = new org.eclipse.draw2d.geometry.Point[1];

		syncExec(new VoidResult() {
			public void run() {
				// find diagram
				IDiagramTypeProvider diagramTypeProvider = diagramEditor.getDiagramTypeProvider();
				final IFeatureProvider fp = diagramTypeProvider.getFeatureProvider();
				final Diagram currentDiagram = diagramTypeProvider.getDiagram();
				executeInRecordingCommand(diagramEditor.getDiagramBehavior(), new Runnable() {
					public void run() {
						// add a lot of classes to the diagram
						for (int i = 0; i < 10; i++)
							for (int j = 0; j < 10; j++)
								addClassToDiagram(fp, currentDiagram, x + deltaX * i, y + deltaY * j, SHAPE_NAME + "_"
										+ i + "_" + j);
					}
				});
			}
		});
		Thread.sleep(DELAY);
		syncExec(new VoidResult() {
			public void run() {
				position[0] = gfFigureCanvas.getViewport().getViewLocation();
			}
		});
		org.eclipse.draw2d.geometry.Point initialPos = position[0];

		syncExec(new VoidResult() {
			public void run() {
				
				Robot r;
				try {
					r = new Robot();
					Point p = ed.getOrigin();
					r.mouseMove(p.x + 150, p.y + 150);
					r.mouseWheel(10);
				} catch (AWTException e) {
					fail(e.getMessage());
				}
			}
		});
		Thread.sleep(SHORT_DELAY);

		syncExec(new VoidResult() {
			public void run() {
				position[0] = gfFigureCanvas.getViewport().getViewLocation();
			}
		});
		org.eclipse.draw2d.geometry.Point currentPos = position[0];
		assertEquals(initialPos.x, currentPos.x);
		assertThat(currentPos.y, greaterThan(initialPos.y));
		initialPos = currentPos;

		syncExec(new VoidResult() {
			public void run() {

				Robot r;
				try {
					r = new Robot();
					r.keyPress(KeyEvent.VK_SHIFT);
					r.mouseWheel(-10);
					r.keyRelease(KeyEvent.VK_SHIFT);

				} catch (AWTException e) {
					fail(e.getMessage());
				}
			}
		});
		Thread.sleep(SHORT_DELAY);

		syncExec(new VoidResult() {
			public void run() {
				position[0] = gfFigureCanvas.getViewport().getViewLocation();
			}
		});
		currentPos = position[0];
		assertEquals(initialPos.y, currentPos.y);
		assertThat(currentPos.x, greaterThan(initialPos.x));

		page.shutdownEditor(diagramEditor);
	}

	@Test
	public void testMouseLocation() throws Exception {
		/*
		 * regression test for CSN 0120031469 0003790113 2008
		 */
		final int x = 100;
		final int y = 100;
		final IDiagramContainerUI diagramEditor = openDiagramEditor(ITestConstants.DIAGRAM_TYPE_ID_ECORE);

		syncExec(new VoidResult() {
			public void run() {
				// find diagram
				IDiagramTypeProvider diagramTypeProvider = diagramEditor.getDiagramTypeProvider();
				final IFeatureProvider fp = diagramTypeProvider.getFeatureProvider();
				final Diagram currentDiagram = diagramTypeProvider.getDiagram();
				executeInRecordingCommand(diagramEditor.getDiagramBehavior(), new Runnable() {
					public void run() {
						// add a class to the diagram
						addClassToDiagram(fp, currentDiagram, x, y, SHAPE_NAME);
					}
				});
			}
		});
		Thread.sleep(DELAY);
		// move class-shape to the origin (0,0)
		ed.drag(SHAPE_NAME, 0, 0);
		Thread.sleep(DELAY);
		syncExec(new VoidResult() {
			public void run() {
				// get instance of GFFigureCanvas
				GFFigureCanvas gfFigureCanvas = ed.getGFCanvas();
				if (gfFigureCanvas == null)
					return;
				// scroll shape into visible region
				Event e = new Event();
				e.doit = true;
				e.stateMask = 0;

				e.detail = SWT.ARROW_UP;
				gfFigureCanvas.getHorizontalBar().notifyListeners(SWT.Selection, e);
				gfFigureCanvas.getVerticalBar().notifyListeners(SWT.Selection, e);
			}
		});
		Thread.sleep(DELAY);

		syncExec(new VoidResult() {
			public void run() {
				Point p = ed.getOrigin();
				Display display = GraphitiUiInternal.getWorkbenchService().getShell().getDisplay();
				Event event = createMouseEvent(p.x + 35, p.y + 35, 0, 0, 0);
				event.type = SWT.MouseMove;
				event.widget = diagramEditor.getGraphicalViewer().getControl();
				event.display = display;
				display.post(event);
			}
		});
		Thread.sleep(DELAY);

		PictogramElement[] selectedPictogramElements = diagramEditor.getSelectedPictogramElements();
		for (PictogramElement pictogramElement : selectedPictogramElements) {
			if (pictogramElement instanceof ContainerShape) {
				GraphicsAlgorithm containerShapeGa = pictogramElement.getGraphicsAlgorithm();
				Rectangle rectangle = new Rectangle(containerShapeGa.getX(), containerShapeGa.getY(),
						containerShapeGa.getWidth(), containerShapeGa.getHeight());
				org.eclipse.draw2d.geometry.Point mouseLocation = diagramEditor.getDiagramBehavior()
						.calculateRealMouseLocation(diagramEditor.getDiagramBehavior().getMouseLocation());
				boolean mouseIsInsideShape = rectangle.contains(mouseLocation);
				assertEquals(" Wrong mouse coordinates :  ", true, mouseIsInsideShape);
				break;
			}
		}
		Thread.sleep(SHORT_DELAY);
		page.shutdownEditor(diagramEditor);
	}

	// /*
	// * Test for Bugzilla 323351
	// */
	// @Test
	// @Ignore("Unstable, see Bugzilla 418398")
	// public void testContextPadHiding() throws Exception {
	// final int x = 100;
	// final int y = 100;
	// final IDiagramContainerUI diagramEditor =
	// openDiagramEditor(ITestConstants.DIAGRAM_TYPE_ID_ECORE);
	//
	// syncExec(new VoidResult() {
	// public void run() {
	// IDiagramTypeProvider diagramTypeProvider =
	// diagramEditor.getDiagramTypeProvider();
	// final IFeatureProvider fp = diagramTypeProvider.getFeatureProvider();
	// final Diagram currentDiagram = diagramTypeProvider.getDiagram();
	// executeInRecordingCommand(diagramEditor.getDiagramBehavior(), new
	// Runnable() {
	// public void run() {
	// addClassToDiagram(fp, currentDiagram, x, y, SHAPE_NAME);
	// }
	// });
	// }
	// });
	// Thread.sleep(DELAY);
	//
	// syncExec(new VoidResult() {
	// public void run() {
	// Robot r;
	// try {
	// r = new Robot();
	// Point p = ed.getOrigin();
	// r.mouseMove(p.x + 150, p.y + 150);
	// } catch (AWTException e) {
	// fail(e.getMessage());
	// }
	// }
	// });
	// Thread.sleep(DELAY);
	//
	// assertNotNull(findContextButtonPad());
	//
	// syncExec(new VoidResult() {
	// public void run() {
	// Robot r;
	// try {
	// r = new Robot();
	// Point p = ed.getOrigin();
	// r.mouseMove(p.x + 50, p.y + 50);
	// } catch (AWTException e) {
	// fail(e.getMessage());
	// }
	// }
	// });
	// Thread.sleep(DELAY);
	//
	// assertNull(findContextButtonPad());
	//
	// syncExec(new VoidResult() {
	// public void run() {
	// Robot r;
	// try {
	// r = new Robot();
	// r.setAutoDelay(20);
	// r.keyPress(KeyEvent.VK_CONTROL);
	// Point p = ed.getOrigin();
	// r.mouseMove(p.x + 150, p.y + 150);
	// } catch (AWTException e) {
	// fail(e.getMessage());
	// }
	// }
	// });
	// Thread.sleep(DELAY);
	//
	// assertNull(findContextButtonPad());
	//
	// syncExec(new VoidResult() {
	// public void run() {
	// Robot r;
	// try {
	// r = new Robot();
	// r.setAutoDelay(20);
	// r.keyRelease(KeyEvent.VK_CONTROL);
	// } catch (AWTException e) {
	// fail(e.getMessage());
	// }
	// }
	// });
	// Thread.sleep(DELAY);
	//
	// syncExec(new VoidResult() {
	// public void run() {
	// try {
	// final Robot robot = new Robot();
	// robot.setAutoDelay(10);
	// try {
	// robot.mousePress(InputEvent.BUTTON1_MASK);
	// robot.mouseRelease(InputEvent.BUTTON1_MASK);
	// robot.keyPress(KeyEvent.VK_DELETE);
	// } catch (RuntimeException e) {
	// fail(e.getMessage());
	// } finally {
	// robot.keyRelease(KeyEvent.VK_DELETE);
	// }
	// } catch (Exception e) {
	// fail(e.getMessage());
	// }
	// }
	// });
	//
	// Thread.sleep(DELAY);
	//
	// syncExec(new VoidResult() {
	// public void run() {
	// try {
	// final Robot robot = new Robot();
	// robot.setAutoDelay(1);
	//
	// try {
	// robot.keyPress(KeyEvent.VK_ENTER);
	// } catch (RuntimeException e) {
	// fail(e.getMessage());
	// } finally {
	// robot.keyRelease(KeyEvent.VK_ENTER);
	// }
	// } catch (Exception e) {
	// fail(e.getMessage());
	// }
	// }
	//
	// });
	// Thread.sleep(DELAY);
	// page.shutdownEditor(diagramEditor);
	// }

	// /*
	// * Test for Bugzilla 323351
	// */
	// @Test
	// @Ignore("Unstable, see Bugzilla 418398")
	// public void testContextPadHidingViaToolbar() throws Exception {
	// /*
	// * TODO: Workaround for not displayed toolbar starting with Eclipse 4.3
	// * M1
	// */
	// syncExec(new VoidResult() {
	// public void run() {
	// SWTWorkbenchBot swtWorkbenchBot = new SWTWorkbenchBot();
	// swtWorkbenchBot.menu("&Window").menu("Hide &Toolbar").click();
	// swtWorkbenchBot.menu("&Window").menu("Show &Toolbar").click();
	// }
	// });
	// Thread.sleep(DELAY);
	// /* End workaround */
	//
	// final int x = 100;
	// final int y = 100;
	// final IDiagramContainerUI diagramEditor =
	// openDiagramEditor(ITestConstants.DIAGRAM_TYPE_ID_ECORE);
	//
	// syncExec(new VoidResult() {
	// public void run() {
	// IDiagramTypeProvider diagramTypeProvider =
	// diagramEditor.getDiagramTypeProvider();
	// final IFeatureProvider fp = diagramTypeProvider.getFeatureProvider();
	// final Diagram currentDiagram = diagramTypeProvider.getDiagram();
	// executeInRecordingCommand(diagramEditor.getDiagramBehavior(), new
	// Runnable() {
	// public void run() {
	// addClassToDiagram(fp, currentDiagram, x, y, SHAPE_NAME);
	// }
	// });
	// }
	// });
	// Thread.sleep(DELAY);
	//
	// syncExec(new VoidResult() {
	// public void run() {
	// Robot r;
	// try {
	// r = new Robot();
	// Point p = ed.getOrigin();
	// r.mouseMove(p.x + 10, p.y + 10);
	// } catch (AWTException e) {
	// fail(e.getMessage());
	// }
	// }
	// });
	// Thread.sleep(SHORT_DELAY);
	//
	// syncExec(new VoidResult() {
	// public void run() {
	// Robot r;
	// try {
	// r = new Robot();
	// Point p = ed.getOrigin();
	// r.mouseMove(p.x + 150, p.y + 150);
	// } catch (AWTException e) {
	// fail(e.getMessage());
	// }
	// }
	// });
	// Thread.sleep(SHORT_DELAY);
	//
	// assertNotNull(findContextButtonPad());
	//
	// syncExec(new VoidResult() {
	// public void run() {
	// Robot r;
	// try {
	// r = new Robot();
	// Point p = ed.getOrigin();
	// r.mouseMove(p.x + 10, p.y + 10);
	// } catch (AWTException e) {
	// fail(e.getMessage());
	// }
	// }
	// });
	// Thread.sleep(DELAY);
	//
	// assertNull(findContextButtonPad());
	//
	// syncExec(new VoidResult() {
	// public void run() {
	// Robot r;
	// try {
	// r = new Robot();
	// r.setAutoDelay(20);
	// new
	// SWTWorkbenchBot().toolbarToggleButtonWithTooltip(Messages.ToggleContextButtonPadAction_0_xmsg)
	// .click();
	// Point p = ed.getOrigin();
	// r.mouseMove(p.x + 150, p.y + 150);
	// } catch (AWTException e) {
	// fail(e.getMessage());
	// }
	// }
	// });
	// Thread.sleep(DELAY);
	//
	// assertNull(findContextButtonPad());
	//
	// syncExec(new VoidResult() {
	// public void run() {
	// try {
	// final Robot robot = new Robot();
	// robot.setAutoDelay(1);
	// try {
	// robot.mousePress(InputEvent.BUTTON1_MASK);
	// robot.mouseRelease(InputEvent.BUTTON1_MASK);
	// robot.keyPress(KeyEvent.VK_DELETE);
	// } catch (RuntimeException e) {
	// fail(e.getMessage());
	// } finally {
	// robot.keyRelease(KeyEvent.VK_DELETE);
	// }
	// } catch (Exception e) {
	// fail(e.getMessage());
	// }
	// }
	// });
	//
	// syncExec(new VoidResult() {
	// public void run() {
	// try {
	// final Robot robot = new Robot();
	// robot.setAutoDelay(1);
	//
	// try {
	// robot.keyPress(KeyEvent.VK_ENTER);
	// } catch (RuntimeException e) {
	// fail(e.getMessage());
	// } finally {
	// robot.keyRelease(KeyEvent.VK_ENTER);
	// }
	// } catch (Exception e) {
	// fail(e.getMessage());
	// }
	// }
	//
	// });
	// Thread.sleep(DELAY);
	// page.shutdownEditor(diagramEditor);
	// }

	private ContextButtonPad findContextButtonPad() {
		SWTBotGefEditPart rootEditPart = ed.getGefEditor().rootEditPart();
		IFigure feedbackLayer = ((ScalableFreeformRootEditPart) rootEditPart.part())
				.getLayer(LayerConstants.HANDLE_LAYER);
		ContextButtonPad cbp = null;
		for (Object obj : feedbackLayer.getChildren()) {
			if (obj instanceof ContextButtonPad) {
				cbp = (ContextButtonPad) obj;
				break;
			}
		}
		return cbp;
	}

	@Test
	public void testContextButtons() throws Exception {
		final int x = 100;
		final int y = 100;
		final IDiagramContainerUI diagramEditor = openDiagramEditor(ITestConstants.DIAGRAM_TYPE_ID_ECORE);

		syncExec(new VoidResult() {
			public void run() {
				IDiagramTypeProvider diagramTypeProvider = diagramEditor.getDiagramTypeProvider();
				final IFeatureProvider fp = diagramTypeProvider.getFeatureProvider();
				final Diagram currentDiagram = diagramTypeProvider.getDiagram();
				executeInRecordingCommand(diagramEditor.getDiagramBehavior(), new Runnable() {
					public void run() {
						addClassToDiagram(fp, currentDiagram, x, y, SHAPE_NAME);
					}
				});
			}
		});
		Thread.sleep(DELAY);

		syncExec(new VoidResult() {
			public void run() {
				Robot r;
				try {
					r = new Robot();
					Point p = ed.getOrigin();
					r.mouseMove(p.x + 150, p.y + 150);
				} catch (AWTException e) {
					fail(e.getMessage());
				}
			}
		});
		Thread.sleep(DELAY);

		syncExec(new VoidResult() {
			public void run() {
				try {
					final Robot robot = new Robot();
					robot.setAutoDelay(1);
					try {
						robot.mousePress(InputEvent.BUTTON3_MASK);
					} catch (RuntimeException e) {
						fail(e.getMessage());
					} finally {
						robot.mouseRelease(InputEvent.BUTTON3_MASK);
					}
				} catch (Exception e) {
					fail(e.getMessage());
				}
			}
		});
		Thread.sleep(DELAY);

		syncExec(new VoidResult() {
			public void run() {
				try {
					final Robot robot = new Robot();
					robot.setAutoDelay(1);
					try {
						robot.keyPress(KeyEvent.VK_ESCAPE);
						robot.keyRelease(KeyEvent.VK_ESCAPE);
						robot.keyPress(KeyEvent.VK_DELETE);
					} catch (RuntimeException e) {
						fail(e.getMessage());
					} finally {
						robot.keyRelease(KeyEvent.VK_DELETE);
					}
				} catch (Exception e) {
					fail(e.getMessage());
				}
			}
		});

		syncExec(new VoidResult() {
			public void run() {
				try {
					final Robot robot = new Robot();
					robot.setAutoDelay(1);

					try {
						robot.keyPress(KeyEvent.VK_ENTER);
					} catch (RuntimeException e) {
						fail(e.getMessage());
					} finally {
						robot.keyRelease(KeyEvent.VK_ENTER);
					}
				} catch (Exception e) {
					fail(e.getMessage());
				}
			}

		});
		Thread.sleep(DELAY);
		page.shutdownEditor(diagramEditor);
	}

	@Test
	public void testDropOnConnection() throws Exception {
		final int xOfShape1 = 100;
		final int yOfShape1 = 100;
		final int xOfShape2 = 100;
		final int yOfShape2 = 400;
		final int xOfShape3 = 400;
		final int yOfShape3 = 100;
		final int xOfShape4 = 400;
		final int yOfShape4 = 400;
		final int width = 60;
		final int height = 60;
		final int DIL = 30;

		final IDiagramContainerUI diagramEditor = openDiagramEditor(ITestConstants.DIAGRAM_TYPE_ID_SKETCH);

		syncExec(new VoidResult() {
			public void run() {
				IDiagramTypeProvider dtp = diagramEditor.getDiagramTypeProvider();
				IFeatureProvider fp = dtp.getFeatureProvider();

				CommandStack commandStack = diagramEditor.getEditDomain().getCommandStack();

				ICreateFeature[] createFeatures = fp.getCreateFeatures();
				for (ICreateFeature createFeature : createFeatures) {
					if (!("Rectangle".equals(createFeature.getName())))
						continue;

					Rectangle rectangle = new Rectangle(xOfShape1, yOfShape1, width, height);
					ICreateContext createContext = createCreateContext(dtp.getDiagram(), rectangle);
					Command createCommand = new CreateModelObjectCommand(getConfigProviderMock(dtp, diagramEditor),
							createFeature, createContext);
					commandStack.execute(createCommand);

					rectangle = new Rectangle(xOfShape2, yOfShape2, width, height);
					createContext = createCreateContext(dtp.getDiagram(), rectangle);
					createCommand = new CreateModelObjectCommand(getConfigProviderMock(dtp, diagramEditor),
							createFeature, createContext);
					commandStack.execute(createCommand);

					rectangle = new Rectangle(xOfShape3, yOfShape3, width, height);
					createContext = createCreateContext(dtp.getDiagram(), rectangle);
					createCommand = new CreateModelObjectCommand(getConfigProviderMock(dtp, diagramEditor),
							createFeature, createContext);
					commandStack.execute(createCommand);

					rectangle = new Rectangle(xOfShape4, yOfShape4, width, height);
					createContext = createCreateContext(dtp.getDiagram(), rectangle);
					createCommand = new CreateModelObjectCommand(getConfigProviderMock(dtp, diagramEditor),
							createFeature, createContext);
					commandStack.execute(createCommand);
				}
			}
		});
		Thread.sleep(DELAY);

		syncExec(new VoidResult() {
			public void run() {
				ed.getGefEditor().activateTool("free");
				ed.getGefEditor().drag(xOfShape1 + DIL, yOfShape1 + DIL, xOfShape3 + DIL, yOfShape3 + DIL);
				ed.getGefEditor().click(xOfShape3 + DIL, yOfShape3 + DIL);
				ed.getGefEditor().activateTool("free");
				ed.getGefEditor().drag(xOfShape1 + DIL, yOfShape1 + DIL, xOfShape2 + DIL, yOfShape2 + DIL);
				ed.getGefEditor().click(xOfShape2 + DIL, yOfShape2 + DIL);
			}
		});
		Thread.sleep(DELAY);

		syncExec(new VoidResult() {
			public void run() {
				ed.getGefEditor().activateTool("Rectangle");
				// click on connection to insert rectangle between two other
				// rectangles
				ed.getGefEditor().click((xOfShape1 + 60 + xOfShape3) / 2, (yOfShape1 + 60 + yOfShape3) / 2);
			}
		});
		Thread.sleep(DELAY);
		ed.drag(xOfShape4 + DIL, yOfShape4 + DIL, xOfShape1 + DIL, (yOfShape2 + yOfShape1 + 2 * DIL) / 2);
		Thread.sleep(DELAY);
		ed.drag(xOfShape1 + DIL, yOfShape1 + DIL, (xOfShape1 + 2 * DIL + xOfShape3) / 2,
				(yOfShape2 + yOfShape1 + 2 * DIL) / 2);
		Thread.sleep(DELAY);
		page.shutdownEditor(diagramEditor);
	}

	@Test
	public void testDirectEditingMultiText() throws Exception {
		final int x = 100;
		final int y = 100;
		final IDiagramContainerUI diagramEditor = openDiagramEditor(ITestConstants.DIAGRAM_TYPE_ID_SKETCH);

		syncExec(new VoidResult() {
			public void run() {
				IDiagramTypeProvider dtp = diagramEditor.getDiagramTypeProvider();
				IFeatureProvider fp = dtp.getFeatureProvider();

				CommandStack commandStack = diagramEditor.getEditDomain().getCommandStack();

				ICreateFeature[] createFeatures = fp.getCreateFeatures();
				for (ICreateFeature createFeature : createFeatures) {
					if (!(SketchFeatureProvider.CF_RECTANGLE_MULTI_TEXT.equals(createFeature.getName())))
						continue;

					Rectangle rectangle = new Rectangle(x, y, 100, 100);
					ICreateContext createContext = createCreateContext(dtp.getDiagram(), rectangle);
					Command createCommand = new CreateModelObjectCommand(getConfigProviderMock(dtp, diagramEditor),
							createFeature, createContext);
					commandStack.execute(createCommand);
				}
			}
		});
		Thread.sleep(DELAY);

		syncExec(new VoidResult() {
			public void run() {
				diagramEditor.getDiagramBehavior().getMouseLocation().x = 150;
				diagramEditor.getDiagramBehavior().getMouseLocation().y = 150;
				SWTBotGefEditPart ep = ed.getGefEditor().mainEditPart().children().get(0);
				ep.activateDirectEdit();
			}
		});
		Thread.sleep(DELAY);
		ed.getGefEditor().directEditType(INPUT_STRING);
		// Check results
		SWTBotGefEditPart ep = ed.getGefEditor().mainEditPart().children().get(0);
		Shape shape = (Shape) ep.part().getModel();
		String value = ((MultiText) shape.getGraphicsAlgorithm().getGraphicsAlgorithmChildren().get(0)).getValue();
		assertEquals(INPUT_STRING, value);
		page.shutdownEditor(diagramEditor);
	}

	@Test
	public void testDirectEditingSingleText() throws Exception {
		final int x = 100;
		final int y = 100;
		final IDiagramContainerUI diagramEditor = openDiagramEditor(ITestConstants.DIAGRAM_TYPE_ID_SKETCH);

		syncExec(new VoidResult() {
			public void run() {
				IDiagramTypeProvider dtp = diagramEditor.getDiagramTypeProvider();
				IFeatureProvider fp = dtp.getFeatureProvider();

				CommandStack commandStack = diagramEditor.getEditDomain().getCommandStack();

				ICreateFeature[] createFeatures = fp.getCreateFeatures();
				for (ICreateFeature createFeature : createFeatures) {
					if (SketchFeatureProvider.CF_RECTANGLE_SINGLE_TEXT.equals(createFeature.getName())) {
						Rectangle rectangle = new Rectangle(x, y, 100, 100);
						ICreateContext createContext = createCreateContext(dtp.getDiagram(), rectangle);
						Command createCommand = new CreateModelObjectCommand(getConfigProviderMock(dtp, diagramEditor),
								createFeature, createContext);
						commandStack.execute(createCommand);
						break;
					}
				}
			}
		});
		Thread.sleep(DELAY);

		syncExec(new VoidResult() {
			public void run() {
				diagramEditor.getDiagramBehavior().getMouseLocation().x = 150;
				diagramEditor.getDiagramBehavior().getMouseLocation().y = 150;
				SWTBotGefEditPart ep = ed.getGefEditor().mainEditPart().children().get(0);
				ep.activateDirectEdit();
			}
		});
		Thread.sleep(DELAY);

		final Robot robot = new Robot();
		robot.setAutoDelay(50);

		syncExec(new VoidResult() {
			public void run() {
				robot.keyPress(KeyEvent.VK_CONTROL);
				robot.keyPress(KeyEvent.VK_SPACE);
				robot.keyRelease(KeyEvent.VK_SPACE);
				robot.keyRelease(KeyEvent.VK_CONTROL);
				robot.keyPress(KeyEvent.VK_ENTER);
				robot.keyRelease(KeyEvent.VK_ENTER);
			}
		});
		Thread.sleep(DELAY);
		page.shutdownEditor(diagramEditor);
	}

	@Test
	public void testDropIntoContainer() throws Exception {
		final int x = 100;
		final int y = 100;
		final IDiagramContainerUI diagramEditor = openDiagramEditor(ITestConstants.DIAGRAM_TYPE_ID_SKETCH);

		syncExec(new VoidResult() {
			public void run() {
				IDiagramTypeProvider dtp = diagramEditor.getDiagramTypeProvider();
				IFeatureProvider fp = dtp.getFeatureProvider();

				CommandStack commandStack = diagramEditor.getEditDomain().getCommandStack();

				ICreateFeature[] createFeatures = fp.getCreateFeatures();
				for (ICreateFeature createFeature : createFeatures) {
					if ("Rectangle".equals(createFeature.getName())) {
						Rectangle rectangle = new Rectangle(x, y, 100, 100);
						ICreateContext createContext = createCreateContext(dtp.getDiagram(), rectangle);
						Command createCommand = new CreateModelObjectCommand(getConfigProviderMock(dtp, diagramEditor),
								createFeature, createContext);
						commandStack.execute(createCommand);
					} else if ("Rectangle Container".equals(createFeature.getName())) {
						Rectangle rectangle = new Rectangle(x + 300, y - 50, 200, 200);
						ICreateContext createContext = createCreateContext(dtp.getDiagram(), rectangle);
						Command createCommand = new CreateModelObjectCommand(getConfigProviderMock(dtp, diagramEditor),
								createFeature, createContext);
						commandStack.execute(createCommand);
					}
				}
			}

		});
		Thread.sleep(DELAY);
		ed.drag(x + 50, y + 50, x + 300 + 100, y + 50);
		Thread.sleep(DELAY);
		assertTrue(ed.getGefEditor().mainEditPart().children().size() == 1);
		page.shutdownEditor(diagramEditor);
	}

	@Test
	public void testConnectionReassign() throws Exception {
		final int xContainer1 = 100;
		final int yContainer1 = 100;
		final int xContainer2 = 300;
		final int yContainer2 = 100;
		final int xContainer3 = 100;
		final int yContainer3 = 300;
		final int containerSize = 100;
		final int rectangleSize = 60;
		final IDiagramContainerUI diagramEditor = openDiagramEditor(ITestConstants.DIAGRAM_TYPE_ID_SKETCH);

		syncExec(new VoidResult() {
			public void run() {
				IDiagramTypeProvider dtp = diagramEditor.getDiagramTypeProvider();
				IFeatureProvider fp = dtp.getFeatureProvider();

				CommandStack commandStack = diagramEditor.getEditDomain().getCommandStack();

				ICreateFeature[] createFeatures = fp.getCreateFeatures();
				for (ICreateFeature createFeature : createFeatures) {
					if ("Rectangle".equals(createFeature.getName())) {
						Rectangle rectangle = new Rectangle(xContainer1 + containerSize, yContainer1 + containerSize,
								rectangleSize, rectangleSize);
						ICreateContext createContext = createCreateContext(dtp.getDiagram(), rectangle);
						Command createCommand = new CreateModelObjectCommand(getConfigProviderMock(dtp, diagramEditor),
								createFeature, createContext);
						commandStack.execute(createCommand);

						rectangle = new Rectangle(xContainer2 + containerSize, yContainer2 + containerSize,
								rectangleSize, rectangleSize);
						createContext = createCreateContext(dtp.getDiagram(), rectangle);
						createCommand = new CreateModelObjectCommand(getConfigProviderMock(dtp, diagramEditor),
								createFeature, createContext);
						commandStack.execute(createCommand);
					} else if ("Rectangle Container".equals(createFeature.getName())) {
						Rectangle rectangle = new Rectangle(xContainer1, yContainer1, containerSize, containerSize);
						ICreateContext createContext = createCreateContext(dtp.getDiagram(), rectangle);
						Command createCommand = new CreateModelObjectCommand(getConfigProviderMock(dtp, diagramEditor),
								createFeature, createContext);
						commandStack.execute(createCommand);

						rectangle = new Rectangle(xContainer2, yContainer2, containerSize, containerSize);
						createContext = createCreateContext(dtp.getDiagram(), rectangle);
						createCommand = new CreateModelObjectCommand(getConfigProviderMock(dtp, diagramEditor),
								createFeature, createContext);
						commandStack.execute(createCommand);

						rectangle = new Rectangle(xContainer3, yContainer3, containerSize, containerSize);
						createContext = createCreateContext(dtp.getDiagram(), rectangle);
						createCommand = new CreateModelObjectCommand(getConfigProviderMock(dtp, diagramEditor),
								createFeature, createContext);
						commandStack.execute(createCommand);
					}
				}
			}

		});
		Thread.sleep(DELAY);

		syncExec(new VoidResult() {
			public void run() {
				int currX = xContainer1 + containerSize + rectangleSize / 2;
				int currY = yContainer1 + containerSize + rectangleSize / 2;
				ed.drag(currX, currY, currX - (containerSize + rectangleSize) / 2, currY
						- (containerSize + rectangleSize) / 2);
				currX = xContainer2 + containerSize + rectangleSize / 2;
				currY = yContainer2 + containerSize + rectangleSize / 2;
				ed.drag(currX, currY, currX - (containerSize + rectangleSize) / 2, currY
						- (containerSize + rectangleSize) / 2);
			}
		});
		Thread.sleep(DELAY);

		syncExec(new VoidResult() {
			public void run() {
				ed.getGefEditor().activateTool("free");
				ed.getGefEditor().click(100 + containerSize / 2, 100 + containerSize / 2);
				ed.getGefEditor().click(300 + containerSize / 2, 100 + containerSize / 2);
			}
		});
		Thread.sleep(DELAY);

		syncExec(new VoidResult() {
			public void run() {
				ed.getGefEditor().activateDefaultTool();
				// move to connection start point
				ed.getGefEditor().click(300 + containerSize / 2 - rectangleSize / 2 - 5, 100 + containerSize / 2); // middle
			}
		});
		Thread.sleep(DELAY);
		ed.drag(300 + containerSize / 2 - rectangleSize / 2 - 1, 100 + containerSize / 2, 300 + containerSize / 2
				- rectangleSize / 2 - 10, 100 + containerSize / 2);
		ed.drag(100 + containerSize / 2, 100 + containerSize / 2, 100 + containerSize / 2, 300 + containerSize / 2);
		Thread.sleep(DELAY);
		page.shutdownEditor(diagramEditor);
	}

	@Test
	public void testCreateConnection() throws Exception {
		final int x = 100;
		final int y = 100;
		final IDiagramContainerUI diagramEditor = openDiagramEditor(ITestConstants.DIAGRAM_TYPE_ID_SKETCH);

		syncExec(new VoidResult() {
			public void run() {
				IDiagramTypeProvider dtp = diagramEditor.getDiagramTypeProvider();
				IFeatureProvider fp = dtp.getFeatureProvider();
				CommandStack commandStack = diagramEditor.getEditDomain().getCommandStack();
				ICreateFeature[] createFeatures = fp.getCreateFeatures();
				for (ICreateFeature createFeature : createFeatures) {
					if ("Rectangle".equals(createFeature.getName())) {
						Rectangle rectangle = new Rectangle(x, y, 100, 100);
						ICreateContext createContext = createCreateContext(dtp.getDiagram(), rectangle);
						Command createCommand = new CreateModelObjectCommand(getConfigProviderMock(dtp, diagramEditor),
								createFeature, createContext);
						commandStack.execute(createCommand);
						rectangle = new Rectangle(x + 200, y, 100, 100);
						createContext = createCreateContext(dtp.getDiagram(), rectangle);
						createCommand = new CreateModelObjectCommand(getConfigProviderMock(dtp, diagramEditor),
								createFeature, createContext);
						commandStack.execute(createCommand);
					}
				}
			}

		});
		Thread.sleep(DELAY);

		syncExec(new VoidResult() {
			public void run() {
				// activate create-freeform-connection tool
				ed.getGefEditor().activateTool("free");
				ed.getGefEditor().drag(150, 150, 350, 150);
				ed.getGefEditor().click(350, 150);
			}
		});
		Thread.sleep(DELAY);

		syncExec(new VoidResult() {
			public void run() {
				// activate selection tool
				ed.getGefEditor().activateDefaultTool();
				// move to line-center and select line
				ed.getGefEditor().click(250, 150); // middle
			}
		});
		Thread.sleep(SHORT_DELAY);
		ed.drag(250, 150, 250, 100);
		Thread.sleep(SHORT_DELAY);
		ed.drag(250, 100, 250, 200);
		Thread.sleep(SHORT_DELAY);
		ed.drag(250, 200, 250, 150);
		Thread.sleep(SHORT_DELAY);
		ed.drag(250, 150, 250, 195);
		Thread.sleep(SHORT_DELAY);

		// remove the connection via context menu
		syncExec(new VoidResult() {
			public void run() {
				ed.getGefEditor().click(250, 195);
				ed.getGefEditor().clickContextMenu("Remove");
			}
		});
		Thread.sleep(DELAY);
		page.shutdownEditor(diagramEditor);
	}

	@Test
	public void testCancelConnection() throws Exception {
		final int x = 100;
		final int y = 100;
		final IDiagramContainerUI diagramEditor = openDiagramEditor(ITestConstants.DIAGRAM_TYPE_ID_SKETCH);

		syncExec(new VoidResult() {
			public void run() {
				IDiagramTypeProvider dtp = diagramEditor.getDiagramTypeProvider();
				IFeatureProvider fp = dtp.getFeatureProvider();
				CommandStack commandStack = diagramEditor.getEditDomain().getCommandStack();
				ICreateFeature[] createFeatures = fp.getCreateFeatures();
				for (ICreateFeature createFeature : createFeatures) {
					if ("Rectangle".equals(createFeature.getName())) {
						Rectangle rectangle = new Rectangle(x, y, 100, 100);
						ICreateContext createContext = createCreateContext(dtp.getDiagram(), rectangle);
						Command createCommand = new CreateModelObjectCommand(getConfigProviderMock(dtp, diagramEditor),
								createFeature, createContext);
						commandStack.execute(createCommand);
						rectangle = new Rectangle(x + 200, y, 100, 100);
						createContext = createCreateContext(dtp.getDiagram(), rectangle);
						createCommand = new CreateModelObjectCommand(getConfigProviderMock(dtp, diagramEditor),
								createFeature, createContext);
						commandStack.execute(createCommand);
					}
				}
			}

		});
		Thread.sleep(DELAY);

		Diagram diagram = diagramEditor.getDiagramTypeProvider().getDiagram();
		assertTrue(diagram.getConnections().isEmpty());

		syncExec(new VoidResult() {
			public void run() {
				// activate create-freeform-connection tool
				try {
					SketchCreateFreeformConnectionFeature.setCancelling(true);
					ed.getGefEditor().activateTool("free");
					ed.getGefEditor().drag(150, 150, 350, 150);
					ed.getGefEditor().click(350, 150);
				} finally {
					SketchCreateFreeformConnectionFeature.setCancelling(false);
				}
			}
		});
		Thread.sleep(DELAY);

		assertTrue(diagram.getConnections().isEmpty());

		CommandStack commandStack = diagramEditor.getEditDomain().getCommandStack();
		assertFalse(commandStack.canRedo());
		page.shutdownEditor(diagramEditor);
	}

	@Test
	public void testResizeAndPosition() throws Exception {
		final int x = 100;
		final int y = 100;
		final IDiagramContainerUI diagramEditor = openDiagramEditor(ITestConstants.DIAGRAM_TYPE_ID_ECORE);

		syncExec(new VoidResult() {
			public void run() {
				// find diagram
				IDiagramTypeProvider diagramTypeProvider = diagramEditor.getDiagramTypeProvider();
				final IFeatureProvider fp = diagramTypeProvider.getFeatureProvider();
				final Diagram currentDiagram = diagramTypeProvider.getDiagram();
				executeInRecordingCommand(diagramEditor.getDiagramBehavior(), new Runnable() {
					public void run() {
						addClassToDiagram(fp, currentDiagram, x, y, SHAPE_NAME);
					}
				});
			}
		});
		Thread.sleep(DELAY);
		// select shape
		syncExec(new VoidResult() {
			public void run() {
				ed.getGefEditor().select(SHAPE_NAME);
			}
		});
		Thread.sleep(DELAY);
		// resize shape
		ed.drag(100, 100, 120, 120);
		Thread.sleep(DELAY);
		Diagram diagram = diagramEditor.getDiagramTypeProvider().getDiagram();
		Shape shape = diagram.getChildren().get(0);
		assertEquals(120, shape.getGraphicsAlgorithm().getX());
		assertEquals(120, shape.getGraphicsAlgorithm().getX());
		page.shutdownEditor(diagramEditor);
	}

	@Test
	public void testSketchAll() throws Exception {
		final IDiagramContainerUI diagramEditor = openDiagramEditor(ITestConstants.DIAGRAM_TYPE_ID_SKETCH);
		PaletteFinder paletteFinder = new PaletteFinder(diagramEditor.getEditDomain());
		List<PaletteEntry> objectCreationTools = paletteFinder.findEntries(new AbstractToolEntryMatcher() {

			public void describeTo(Description description) {
				// Do nothing
			}

			@Override
			protected boolean matches(ToolEntry toolEntry) {
				if (toolEntry.getParent().getLabel() == "Objects")
					return true;
				return false;
			}
		});
		final int x = 10;
		final int y = 6 * objectCreationTools.size();
		final int[] counter = new int[] { 0 };
		for (PaletteEntry o : objectCreationTools) {
			if (o instanceof ToolEntry) {
				final ToolEntry toolEntry = (ToolEntry) o;
				syncExec(new VoidResult() {
					public void run() {
						ed.getGefEditor().activateTool(toolEntry.getLabel());
						int currX = x + (counter[0] * 5);
						int currY = y - (5 * counter[0]);
						ed.getGefEditor().drag(currX, currY, currX + 10, currY + 10);
					}
				});
				// Thread.sleep(300);
				counter[0]++;
			}
		}
		Thread.sleep(DELAY);
		assertEquals(objectCreationTools.size(), ed.getGefEditor().mainEditPart().children().size());
		page.shutdownEditor(diagramEditor);
	}

	@Test
	public void testMoveConnectionDecorator() throws InterruptedException {
		// Test for Bug 355027: Move of connection decorators when zoom level !=
		// 100 behaves weird
		page.closeAllEditors();
		final IDiagramContainerUI diagramEditor = openDiagramEditor(ITestConstants.DIAGRAM_TYPE_ID_SKETCH);
		final IDiagramTypeProvider dtp = diagramEditor.getDiagramTypeProvider();
		final IFeatureProvider fp = ((DefaultFeatureProviderWrapper) dtp.getFeatureProvider())
				.getInnerFeatureProvider();
		final CommandStack commandStack = diagramEditor.getEditDomain().getCommandStack();

		// Preparation work
		syncExec(new VoidResult() {
			public void run() {

				// Create an outer container
				ICreateFeature createFeature = new SketchCreateGaContainerFeature(fp, "Rounded Rectangle Container",
						"draw rounded rectangle", RoundedRectangle.class);
				Rectangle rectangle = new Rectangle(0, 0, 300, 200);
				ICreateContext createContext = createCreateContext(dtp.getDiagram(), rectangle);
				Command createCommand = new CreateModelObjectCommand(getConfigProviderMock(dtp, diagramEditor),
						createFeature, createContext);
				commandStack.execute(createCommand);
				ContainerShape outerShape = (ContainerShape) dtp.getDiagram().getChildren().get(0);

				// Create two children in the container
				createFeature = new SketchCreateGaShapeFeature(fp, SketchFeatureProvider.CF_RECTANGLE_SINGLE_TEXT,
						"draw rectangle with a single line text", org.eclipse.graphiti.mm.algorithms.Rectangle.class);
				rectangle = new Rectangle(50, 50, 51, 51);
				createContext = createCreateContext(outerShape, rectangle);
				createCommand = new CreateModelObjectCommand(getConfigProviderMock(dtp, diagramEditor), createFeature,
						createContext);
				commandStack.execute(createCommand);
				Shape leftShape = outerShape.getChildren().get(0);

				createFeature = new SketchCreateGaShapeFeature(fp, SketchFeatureProvider.CF_RECTANGLE_SINGLE_TEXT,
						"draw rectangle with a single line text", org.eclipse.graphiti.mm.algorithms.Rectangle.class);
				rectangle = new Rectangle(150, 50, 51, 51);
				createContext = createCreateContext(outerShape, rectangle);
				createCommand = new CreateModelObjectCommand(getConfigProviderMock(dtp, diagramEditor), createFeature,
						createContext);
				commandStack.execute(createCommand);
				Shape rightShape = outerShape.getChildren().get(1);

				// Create a connection between the child shapes
				final ICreateConnectionFeature[] ccfs = new ICreateConnectionFeature[] { new SketchCreateFreeformConnectionFeature(
						fp, "free", "freeform connection") };
				Anchor sourceAnchor = getPeService().getChopboxAnchor(leftShape);
				Anchor targetAnchor = getPeService().getChopboxAnchor(rightShape);
				final CreateConnectionContext ccc = new CreateConnectionContext();
				ccc.setSourceAnchor(sourceAnchor);
				ccc.setTargetAnchor(targetAnchor);
				executeInRecordingCommand(diagramEditor.getDiagramBehavior(), new Runnable() {
					public void run() {
						for (ICreateConnectionFeature ccf : ccfs) {
							if (ccf.canCreate(ccc)) {
								ccf.execute(ccc);
							}
						}
					}
				});
				// Show the connection decorators
				ToggleDecorator feature = new ToggleDecorator(fp);
				CustomContext context = new CustomContext();
				context.setPictogramElements(new PictogramElement[] { dtp.getDiagram() });
				GenericFeatureCommandWithContext gfcwc = new GenericFeatureCommandWithContext(feature, context);
				CommandContainer commandContainer = new CommandContainer(fp);
				commandContainer.add(gfcwc);
				GefCommandWrapper gefCommandWrapper = new GefCommandWrapper(commandContainer,
						ed.getTransactionalEditingDomain());
				commandStack.execute(gefCommandWrapper);

			}
		});
		Thread.sleep(SHORT_DELAY);

		// Do the move of the connection decorator with standard zoom level
		// (100%)
		ed.drag(115, 80, 135, 110);
		ConnectionDecorator connectionDecorator = dtp.getDiagram().getConnections().get(0).getConnectionDecorators()
				.get(0);
		/*
		 * Strange numbers below are ok: they are "calculated" using the
		 * previous offset of the decorator, the new move point and the position
		 * the drag operation starts
		 */
		assertEquals(17, connectionDecorator.getGraphicsAlgorithm().getX());
		assertEquals(5, connectionDecorator.getGraphicsAlgorithm().getY());
		Thread.sleep(SHORT_DELAY);

		// Set zoom level to next zoom level (150%)
		syncExec(new VoidResult() {
			public void run() {
				SWTBotMenu viewMenu = new SWTWorkbenchBot().menu("View");
				viewMenu.click();
				SWTBotMenu zoomInItem = viewMenu.menu("Zoom In");
				zoomInItem.click();
			}
		});
		Thread.sleep(SHORT_DELAY);

		// Do the move of the connection decorator with zoom level as set before
		ed.drag(198, 150, 220, 180);
		/*
		 * Strange numbers below are ok: they are "calculated" using the
		 * previous offset of the decorator, the new move point, the position
		 * the drag operation starts and the zoom factor
		 */
		connectionDecorator = dtp.getDiagram().getConnections().get(0).getConnectionDecorators().get(0);
		assertEquals(33, connectionDecorator.getGraphicsAlgorithm().getX());
		assertEquals(25, connectionDecorator.getGraphicsAlgorithm().getY());
		Thread.sleep(SHORT_DELAY);
		page.shutdownEditor(diagramEditor);
	}

	@Test
	public void testCurvedConnection() throws Exception {
		page.closeAllEditors();
		final IDiagramContainerUI diagramEditor = openDiagramEditor(ITestConstants.DIAGRAM_TYPE_ID_SKETCH);
		final IDiagramTypeProvider dtp = diagramEditor.getDiagramTypeProvider();
		final IFeatureProvider fp = ((DefaultFeatureProviderWrapper) dtp.getFeatureProvider())
				.getInnerFeatureProvider();
		final CommandStack commandStack = diagramEditor.getEditDomain().getCommandStack();

		// Create 2 rectangles and a curved connection
		syncExec(new VoidResult() {
			public void run() {
				// One
				{
					ICreateFeature createFeature = new SketchCreateGaContainerFeature(fp,
							"Rounded Rectangle Container", "draw rounded rectangle", RoundedRectangle.class);
					Rectangle rectangle = new Rectangle(100, 100, 100, 50);
					ICreateContext createContext = createCreateContext(dtp.getDiagram(), rectangle);
					Command createCommand = new CreateModelObjectCommand(getConfigProviderMock(dtp, diagramEditor),
							createFeature, createContext);
					commandStack.execute(createCommand);
				}
				ContainerShape shape1 = (ContainerShape) dtp.getDiagram().getChildren().get(0);
				assertNotNull(shape1);

				// Two
				{
					ICreateFeature createFeature = new SketchCreateGaContainerFeature(fp,
							"Rounded Rectangle Container", "draw rounded rectangle", RoundedRectangle.class);
					Rectangle rectangle = new Rectangle(500, 200, 100, 50);
					ICreateContext createContext = createCreateContext(dtp.getDiagram(), rectangle);
					Command createCommand = new CreateModelObjectCommand(getConfigProviderMock(dtp, diagramEditor),
							createFeature, createContext);
					commandStack.execute(createCommand);
				}
				ContainerShape shape2 = (ContainerShape) dtp.getDiagram().getChildren().get(1);
				assertNotNull(shape2);

				// Create a connection between the child shapes
				final ICreateConnectionFeature[] ccfs = new ICreateConnectionFeature[] { new SketchCreateCurvedConnectionFeature(
						fp, "curved", "curved connection") };
				Anchor sourceAnchor = getPeService().getChopboxAnchor(shape1);
				Anchor targetAnchor = getPeService().getChopboxAnchor(shape2);
				final CreateConnectionContext ccc = new CreateConnectionContext();
				ccc.setSourceAnchor(sourceAnchor);
				ccc.setTargetAnchor(targetAnchor);
				executeInRecordingCommand(diagramEditor.getDiagramBehavior(), new Runnable() {
					public void run() {
						for (ICreateConnectionFeature ccf : ccfs) {
							if (ccf.canCreate(ccc)) {
								ccf.execute(ccc);
							}
						}
					}
				});
			}
		});
		Thread.sleep(SHORT_DELAY);
		page.shutdownEditor(diagramEditor);
	}

	@Test
	public void testRenderingDecorators() throws Exception {
		page.closeAllEditors();
		final IDiagramContainerUI diagramEditor = openDiagramEditor(ITestConstants.DIAGRAM_TYPE_ID_SKETCH);
		final IDiagramTypeProvider dtp = diagramEditor.getDiagramTypeProvider();
		final IFeatureProvider fp = ((DefaultFeatureProviderWrapper) dtp.getFeatureProvider())
				.getInnerFeatureProvider();
		final CommandStack commandStack = diagramEditor.getEditDomain().getCommandStack();

		// Create 3 rectangles
		syncExec(new VoidResult() {
			public void run() {
				// One
				{
					ICreateFeature createFeature = new SketchCreateGaContainerFeature(fp,
							"Rounded Rectangle Container", "draw rounded rectangle", RoundedRectangle.class);
					Rectangle rectangle = new Rectangle(50, 50, 100, 50);
					ICreateContext createContext = createCreateContext(dtp.getDiagram(), rectangle);
					Command createCommand = new CreateModelObjectCommand(getConfigProviderMock(dtp, diagramEditor),
							createFeature, createContext);
					commandStack.execute(createCommand);
				}
				ContainerShape shape1 = (ContainerShape) dtp.getDiagram().getChildren().get(0);
				assertNotNull(shape1);

				// Two
				{
					ICreateFeature createFeature = new SketchCreateGaContainerFeature(fp,
							"Rounded Rectangle Container", "draw rounded rectangle", RoundedRectangle.class);
					Rectangle rectangle = new Rectangle(200, 50, 100, 50);
					ICreateContext createContext = createCreateContext(dtp.getDiagram(), rectangle);
					Command createCommand = new CreateModelObjectCommand(getConfigProviderMock(dtp, diagramEditor),
							createFeature, createContext);
					commandStack.execute(createCommand);
				}
				ContainerShape shape2 = (ContainerShape) dtp.getDiagram().getChildren().get(1);
				assertNotNull(shape2);

				// Three
				{
					ICreateFeature createFeature = new SketchCreateGaContainerFeature(fp,
							"Rounded Rectangle Container", "draw rounded rectangle", RoundedRectangle.class);
					Rectangle rectangle = new Rectangle(50, 150, 100, 50);
					ICreateContext createContext = createCreateContext(dtp.getDiagram(), rectangle);
					Command createCommand = new CreateModelObjectCommand(getConfigProviderMock(dtp, diagramEditor),
							createFeature, createContext);
					commandStack.execute(createCommand);
				}
				ContainerShape shape3 = (ContainerShape) dtp.getDiagram().getChildren().get(2);
				assertNotNull(shape3);

				// Four
				{
					ICreateFeature createFeature = new SketchCreateGaContainerFeature(fp,
							"Rounded Rectangle Container", "draw rounded rectangle", RoundedRectangle.class);
					Rectangle rectangle = new Rectangle(200, 150, 100, 50);
					ICreateContext createContext = createCreateContext(dtp.getDiagram(), rectangle);
					Command createCommand = new CreateModelObjectCommand(getConfigProviderMock(dtp, diagramEditor),
							createFeature, createContext);
					commandStack.execute(createCommand);
				}
				ContainerShape shape4 = (ContainerShape) dtp.getDiagram().getChildren().get(3);
				assertNotNull(shape4);

				// Add image decorator to one
				{
					CustomContext context = new CustomContext(new PictogramElement[] { shape1 });
					DisplayDecoratorFeature feature = new DisplayDecoratorFeature(fp, context,
							DisplayDecoratorFeature.TYPE_IMAGE);
					diagramEditor.getDiagramBehavior().executeFeature(feature, context);
				}

				// Add border decorator to two
				{
					CustomContext context = new CustomContext(new PictogramElement[] { shape2 });
					DisplayDecoratorFeature feature = new DisplayDecoratorFeature(fp, context,
							DisplayDecoratorFeature.TYPE_BORDER);
					diagramEditor.getDiagramBehavior().executeFeature(feature, context);
				}

				// Add color decorator to three
				{
					CustomContext context = new CustomContext(new PictogramElement[] { shape3 });
					DisplayDecoratorFeature feature = new DisplayDecoratorFeature(fp, context,
							DisplayDecoratorFeature.TYPE_COLOR);
					diagramEditor.getDiagramBehavior().executeFeature(feature, context);
				}

				// Add text decorator to four
				{
					CustomContext context = new CustomContext(new PictogramElement[] { shape4 });
					DisplayDecoratorFeature feature = new DisplayDecoratorFeature(fp, context,
							DisplayDecoratorFeature.TYPE_TEXT);
					diagramEditor.getDiagramBehavior().executeFeature(feature, context);
				}

				// Check if image is shown
				{
					GraphicalEditPart shape1EditPart = diagramEditor.getDiagramBehavior()
							.getEditPartForPictogramElement(shape1);
					IPictogramElementDelegate delegate = ((ContainerShapeEditPart) shape1EditPart)
							.getPictogramElementDelegate();
					try {
						Class<? extends IPictogramElementDelegate> class1 = delegate.getClass();
						Field field = class1.getSuperclass().getDeclaredField("decoratorMap");
						field.setAccessible(true);
						@SuppressWarnings("rawtypes")
						HashMap map = (HashMap) field.get(delegate);
						@SuppressWarnings("rawtypes")
						DecoratorImageFigure figure = (DecoratorImageFigure) ((ArrayList) map.get(shape1EditPart
								.getFigure())).get(0);
						assertNotNull(figure);
					} catch (Exception e) {
						fail(e.getMessage());
					}
				}

				// Check if border is shown
				{
					GraphicalEditPart shape2EditPart = diagramEditor.getDiagramBehavior()
							.getEditPartForPictogramElement(shape2);
					Color color = ((LineBorder) shape2EditPart.getFigure().getBorder()).getColor();
					assertEquals(IColorConstant.ORANGE.getBlue(), color.getBlue());
					assertEquals(IColorConstant.ORANGE.getGreen(), color.getGreen());
					assertEquals(IColorConstant.ORANGE.getRed(), color.getRed());
				}

				// Check if color is shown
				{
					GraphicalEditPart shape3EditPart = diagramEditor.getDiagramBehavior()
							.getEditPartForPictogramElement(shape3);
					Color foregroundColor = shape3EditPart.getFigure().getForegroundColor();
					assertEquals(IColorConstant.DARK_GREEN.getBlue(), foregroundColor.getBlue());
					assertEquals(IColorConstant.DARK_GREEN.getGreen(), foregroundColor.getGreen());
					assertEquals(IColorConstant.DARK_GREEN.getRed(), foregroundColor.getRed());
					Color backgroundColor = shape3EditPart.getFigure().getBackgroundColor();
					assertEquals(IColorConstant.LIGHT_ORANGE.getBlue(), backgroundColor.getBlue());
					assertEquals(IColorConstant.LIGHT_ORANGE.getGreen(), backgroundColor.getGreen());
					assertEquals(IColorConstant.LIGHT_ORANGE.getRed(), backgroundColor.getRed());
				}

				// Check if text is shown
				{
					GraphicalEditPart shape4EditPart = diagramEditor.getDiagramBehavior()
							.getEditPartForPictogramElement(shape4);
					IPictogramElementDelegate delegate = ((ContainerShapeEditPart) shape4EditPart)
							.getPictogramElementDelegate();
					try {
						Class<? extends IPictogramElementDelegate> class4 = delegate.getClass();
						Field field = class4.getSuperclass().getDeclaredField("decoratorMap");
						field.setAccessible(true);
						@SuppressWarnings("rawtypes")
						HashMap map = (HashMap) field.get(delegate);
						@SuppressWarnings("rawtypes")
						Label figure = (Label) ((ArrayList) map.get(shape4EditPart
								.getFigure())).get(0);
						assertNotNull(figure);
						assertEquals("Decorated with some example text", figure.getText());
						Color whiteColor = new Color(Display.getCurrent(), IColorConstant.WHITE.getRed(),
								IColorConstant.WHITE.getGreen(), IColorConstant.WHITE.getBlue());
						assertEquals(whiteColor, figure.getForegroundColor());
						whiteColor.dispose();
						Color orangeColor = new Color(Display.getCurrent(), IColorConstant.ORANGE.getRed(),
								IColorConstant.ORANGE.getGreen(), IColorConstant.ORANGE.getBlue());
						assertEquals(orangeColor, figure.getBackgroundColor());
						orangeColor.dispose();
					} catch (Exception e) {
						fail(e.getMessage());
					}
				}
			}
		});

		Thread.sleep(SHORT_DELAY);
		page.shutdownEditor(diagramEditor);
	}

	public void testCompositeConnection() throws Exception {
		page.closeAllEditors();
		final IDiagramContainerUI diagramEditor = openDiagramEditor(ITestConstants.DIAGRAM_TYPE_ID_SKETCH);
		final IDiagramTypeProvider dtp = diagramEditor.getDiagramTypeProvider();
		final IFeatureProvider fp = ((DefaultFeatureProviderWrapper) dtp.getFeatureProvider())
				.getInnerFeatureProvider();
		final CommandStack commandStack = diagramEditor.getEditDomain().getCommandStack();

		// Create 2 rectangles and a curved connection
		syncExec(new VoidResult() {
			public void run() {
				// One
				{
					ICreateFeature createFeature = new SketchCreateGaContainerFeature(fp,
							"Rounded Rectangle Container", "draw rounded rectangle", RoundedRectangle.class);
					Rectangle rectangle = new Rectangle(100, 100, 100, 50);
					ICreateContext createContext = createCreateContext(dtp.getDiagram(), rectangle);
					Command createCommand = new CreateModelObjectCommand(getConfigProviderMock(dtp, diagramEditor),
							createFeature, createContext);
					commandStack.execute(createCommand);
				}
				ContainerShape shape1 = (ContainerShape) dtp.getDiagram().getChildren().get(0);
				assertNotNull(shape1);

				// Two
				{
					ICreateFeature createFeature = new SketchCreateGaContainerFeature(fp,
							"Rounded Rectangle Container", "draw rounded rectangle", RoundedRectangle.class);
					Rectangle rectangle = new Rectangle(500, 200, 100, 50);
					ICreateContext createContext = createCreateContext(dtp.getDiagram(), rectangle);
					Command createCommand = new CreateModelObjectCommand(getConfigProviderMock(dtp, diagramEditor),
							createFeature, createContext);
					commandStack.execute(createCommand);
				}
				ContainerShape shape2 = (ContainerShape) dtp.getDiagram().getChildren().get(1);
				assertNotNull(shape2);

				// Create a connection between the child shapes
				final ICreateConnectionFeature[] ccfs = new ICreateConnectionFeature[] { new SketchCreateCompositeConnectionFeature(
						fp, "curved", "curved connection") };
				Anchor sourceAnchor = getPeService().getChopboxAnchor(shape1);
				Anchor targetAnchor = getPeService().getChopboxAnchor(shape2);
				final CreateConnectionContext ccc = new CreateConnectionContext();
				ccc.setSourceAnchor(sourceAnchor);
				ccc.setTargetAnchor(targetAnchor);
				executeInRecordingCommand(diagramEditor.getDiagramBehavior(), new Runnable() {
					public void run() {
						for (ICreateConnectionFeature ccf : ccfs) {
							if (ccf.canCreate(ccc)) {
								ccf.execute(ccc);
							}
						}
					}
				});
			}
		});

		Thread.sleep(SHORT_DELAY);
		page.shutdownEditor(diagramEditor);
	}

	@Test
	public void testDoubleClickOnDiagram() throws Exception {
		page.closeAllEditors();
		final IDiagramContainerUI diagramEditor = openDiagramEditor(ITestConstants.DIAGRAM_TYPE_ID_SKETCH);

		DoubleClickFeature.executed = false;
		Thread.sleep(DELAY);

		syncExec(new VoidResult() {
			public void run() {
				ed.getGefEditor().doubleClick(50, 50);
			}
		});
		Thread.sleep(SHORT_DELAY);
		assertTrue("Double click was not executed on the diagram", DoubleClickFeature.executed);

		page.shutdownEditor(diagramEditor);
	}
}