tests/org.eclipse.graphiti.bot.tests/src/org/eclipse/graphiti/bot/tests/GFDialogTests.java

/*******************************************************************************
 * <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
 *    pjpaulin - Bug 352120 - Changed to create action with actual workbench part
 *    pjpaulin - Bug 352120 - Now uses IDiagramContainerUI interface
 *    mwenz - Bug 370888 - API Access to export and print
 *
 * </copyright>
 *
 *******************************************************************************/
package org.eclipse.graphiti.bot.tests;

import static org.eclipse.swtbot.swt.finder.finders.UIThreadRunnable.asyncExec;

import org.eclipse.graphiti.bot.tests.util.ITestConstants;
import org.eclipse.graphiti.dt.IDiagramTypeProvider;
import org.eclipse.graphiti.ui.editor.IDiagramContainerUI;
import org.eclipse.graphiti.ui.internal.Messages;
import org.eclipse.graphiti.ui.internal.action.SaveImageAction;
import org.eclipse.jface.action.IAction;
import org.eclipse.swtbot.swt.finder.results.VoidResult;
import org.eclipse.swtbot.swt.finder.waits.Conditions;
import org.eclipse.swtbot.swt.finder.widgets.SWTBotShell;
import org.junit.Test;

public class GFDialogTests extends AbstractGFTests {

	public GFDialogTests() {
		super();
	}

	// @Test
	// public void testPrintDialog() throws Exception {
	// final IDiagramContainerUI diagramEditor =
	// openDiagramEditor(ITestConstants.DIAGRAM_TYPE_ID_SKETCH);
	// final boolean[] enabled = new boolean[1];
	// final CountDownLatch signal = new CountDownLatch(1);
	// asyncExec(new VoidResult() {
	// public void run() {
	// IDiagramTypeProvider dtp = diagramEditor.getDiagramTypeProvider();
	// IAction printGraphicalViewerAction = new
	// PrintGraphicalViewerAction(diagramEditor.getDiagramBehavior(),
	// getConfigProviderMock(dtp, diagramEditor));
	// // check if default printer is configured, otherwise SWT throws
	// // a "no more handles" error in Printer.checkNull(..)
	// enabled[0] = printGraphicalViewerAction.isEnabled();
	// if (enabled[0])
	// printGraphicalViewerAction.run();
	// signal.countDown();
	// }
	// });
	// signal.await(5, TimeUnit.SECONDS);
	// // If the action is not enabled since there is no printer available on
	// // the system, we simply return.
	// if (!enabled[0]) {
	// return;
	// }
	// bot.waitUntil(Conditions.shellIsActive(Messages.PrintFigureDialog_3_xfld));
	// SWTBotShell shell = bot.shell(Messages.PrintFigureDialog_3_xfld);
	// Thread.sleep(1000);
	// shell.bot().button("Cancel").click();
	// Thread.sleep(300);
	// bot.waitUntil(Conditions.shellCloses(shell));
	// page.closeActiveEditor();
	//
	// }

	@Test
	public void testSaveDialog() throws Exception {
		final IDiagramContainerUI diagramEditor = openDiagramEditor(ITestConstants.DIAGRAM_TYPE_ID_SKETCH);
		final boolean[] enabled = new boolean[1];

		asyncExec(new VoidResult() {
			public void run() {
				IDiagramTypeProvider dtp = diagramEditor.getDiagramTypeProvider();
				IAction saveImageAction = new SaveImageAction(diagramEditor.getDiagramBehavior(),
						getConfigProviderMock(dtp, diagramEditor));
				// check if default printer is configured, otherwise SWT throws
				// a "no more handles" error in Printer.checkNull(..)
				enabled[0] = saveImageAction.isEnabled();
				if (enabled[0])
					saveImageAction.run();
			}
		});

		bot.waitUntil(Conditions.shellIsActive(Messages.SaveFigureAsImageDialog_5_xtxt));
		SWTBotShell shell = bot.shell(Messages.SaveFigureAsImageDialog_5_xtxt);
		Thread.sleep(2000);
		shell.bot().button("Cancel").click();
		Thread.sleep(300);
		bot.waitUntil(Conditions.shellCloses(shell));
		page.closeActiveEditor();

	}
}