org.eclipse.ajdt.ui.visual.tests/src/org/eclipse/ajdt/ui/visual/tests/OpenDeclarationTest.java

/*******************************************************************************
 * Copyright (c) 2005 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
 *     Matt Chapman  - initial version
 *******************************************************************************/
package org.eclipse.ajdt.ui.visual.tests;

import org.eclipse.ajdt.core.AspectJPlugin;
import org.eclipse.ajdt.core.javaelements.AJCompilationUnitManager;
import org.eclipse.ajdt.ui.tests.AspectJTestPlugin;
import org.eclipse.core.resources.IFile;
import org.eclipse.core.resources.IProject;
import org.eclipse.jface.text.TextSelection;
import org.eclipse.jface.viewers.ISelection;
import org.eclipse.swt.SWT;
import org.eclipse.ui.IEditorPart;
import org.eclipse.ui.IWorkbenchPage;
import org.eclipse.ui.IWorkbenchWindow;

public class OpenDeclarationTest extends VisualTestCase {
	public void testOpenDeclaration() throws Exception {
		IProject project = createPredefinedProject("OpenDeclaration"); //$NON-NLS-1$
		assertNotNull("The OpenDeclaration project should have been created", //$NON-NLS-1$
				project);
		IFile getinfo = (IFile) project.findMember("src/tjp/GetInfo.aj"); //$NON-NLS-1$
		assertNotNull(
				"The OpenDeclaration project should contain a file called 'GetInfo.aj'", //$NON-NLS-1$
				getinfo);
		// make sure ajdt knows about the supertype - we probably shouldn't need
		// this
		AJCompilationUnitManager.INSTANCE.initCompilationUnits(AspectJPlugin
				.getWorkspace());
		openFileInDefaultEditor(getinfo, true);

		// F3 over the demoExecs pointcut used in around advice
		// perform the test at each valid cursor position
		for (int i = 20; i <= 29; i++) {
			gotoLine(28);
			moveCursorRight(i);
			postKey(SWT.F3);
			waitForJobsToComplete();
			checkSelection(25, "demoExecs"); //$NON-NLS-1$
		}
		// check that F3 either side of the pointcut doesnt work
		gotoLine(28);
		moveCursorRight(19);
		postKey(SWT.F3);
		waitForJobsToComplete();
		checkEmptySelection();
		gotoLine(28);
		moveCursorRight(30);
		postKey(SWT.F3);
		waitForJobsToComplete();
		checkEmptySelection();

		waitForJobsToComplete();
		
		// F3 over the goCut pointcut used in the demoExecs pointcut
		gotoLine(26);
		moveCursorRight(43);
		postKey(SWT.F3);
		waitForJobsToComplete();
		checkSelection(23, "goCut"); //$NON-NLS-1$

		// F3 over the executeGo pointcut inherited from the supertype
		gotoLine(24);
		moveCursorRight(53);
		postKey(SWT.F3);
		waitForJobsToComplete();
		checkSelection(3, "executeGo"); //$NON-NLS-1$

		// close the editor opened for the supertype, to return to GetInfo.aj
		IWorkbenchPage page = getWindow().getActivePage();
		IEditorPart activeEditor = page.getActiveEditor();
		page.closeEditor(activeEditor, false);

		// F3 over the demoExecs2 pointcut defined in the Demo type
		gotoLine(40);
		moveCursorRight(24);
		postKey(SWT.F3);
		waitForJobsToComplete();
		checkSelection(18, "demoExecs2"); //$NON-NLS-1$
	}

	private IWorkbenchWindow getWindow() {
		IWorkbenchWindow window = AspectJTestPlugin.getDefault().getWorkbench()
				.getActiveWorkbenchWindow();
		if (window == null) {
			fail("Couldn't get active workbench window"); //$NON-NLS-1$
		}
		return window;
	}

	private TextSelection getCurrentTextSelection() {
		ISelection sel = getWindow().getSelectionService().getSelection();
		if (!(sel instanceof TextSelection)) {
			fail("Expected ISelection to be a TextSelection"); //$NON-NLS-1$
		}
		return (TextSelection) sel;
	}

	private void checkEmptySelection() {
		TextSelection ts = getCurrentTextSelection();
		assertTrue("Current text selection should be empty: "+ts.getText(), //$NON-NLS-1$
				ts.getText().length() == 0);
	}

	private void checkSelection(int expectedLine, String expectedText) {
		TextSelection ts = getCurrentTextSelection();
		assertTrue("Pressing F3 should have selected the pointcut on line " //$NON-NLS-1$
				+ expectedLine + ". got start line: " + ts.getStartLine(), ts //$NON-NLS-1$
				.getStartLine() == expectedLine);
		assertTrue("Pressing F3 should have selected the pointcut on line " //$NON-NLS-1$
				+ expectedLine + ". got end line: " + ts.getEndLine(), ts //$NON-NLS-1$
				.getEndLine() == expectedLine);
		assertEquals("Pressing F3 should have selected the " + expectedText //$NON-NLS-1$
				+ " pointcut." + ts.getText(), expectedText, ts.getText()); //$NON-NLS-1$
	}
}