org.eclipse.ajdt.ui.visual.tests/src/org/eclipse/ajdt/ui/visual/tests/OrganiseImportsTest2.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
 *     Sian January  - initial version
 *******************************************************************************/
package org.eclipse.ajdt.ui.visual.tests;

import org.eclipse.ajdt.core.javaelements.AJCompilationUnitManager;
import org.eclipse.core.resources.IFile;
import org.eclipse.core.resources.IMarker;
import org.eclipse.core.resources.IProject;
import org.eclipse.core.resources.IResource;
import org.eclipse.core.runtime.CoreException;
import org.eclipse.jdt.core.ICompilationUnit;
import org.eclipse.jdt.core.JavaModelException;
import org.eclipse.swt.SWT;

/**
 * A second visual test case for organise imports in a .aj file
 */
public class OrganiseImportsTest2 extends VisualTestCase {

	/**
	 * Test organize imports when there's nothing to add or remove
	 * in an aspect that extends an abstract aspect and has more 
	 * than one variable.
	 * @throws Exception
	 */
	public void testOrganiseImports2() throws Exception {
		IProject project = createPredefinedProject("Spacewar Example"); //$NON-NLS-1$
		IFile gameSynchronizationFile = (IFile)project.findMember("src/spacewar/GameSynchronization.aj"); //$NON-NLS-1$
		assertTrue("The Spacewar Example project should contain a file called 'GameSynchronization.aj'", gameSynchronizationFile != null ); //$NON-NLS-1$
		openFileInDefaultEditor(gameSynchronizationFile, true);
		final ICompilationUnit cUnit = AJCompilationUnitManager.INSTANCE.getAJCompilationUnit(gameSynchronizationFile);
		assertTrue("GameSynchronization.aj should start with one import", cUnit.getImports().length == 1); //$NON-NLS-1$
		waitForJobsToComplete();			
		
//			 Post Ctrl+Shift+O to organise imports
		postKeyDown(SWT.CTRL);
		postKeyDown(SWT.SHIFT);
		postKey('o');
		postKeyUp(SWT.SHIFT);
		postKeyUp(SWT.CTRL);

//			 Wait for an error condition to make sure it's caught if it does occur
		new DisplayHelper() {
			protected boolean condition() {
				try {
					return  cUnit.getImports().length == 0;
				} catch (JavaModelException e) {
				}
				return false;
			}
		
		}.waitForCondition(display, 3000);
		assertTrue("GameSynchronization.aj should now have one imports, has " + cUnit.getImports().length, cUnit.getImports().length == 1); //$NON-NLS-1$
		
		// Post Ctrl+S to save the file
		postKeyDown(SWT.CTRL);
		postKey('s');
		postKeyUp(SWT.CTRL);

		// Wait for an error condition to make sure it's caught if it does occur
		new DisplayHelper() {
			protected boolean condition() {
				try {					
					IMarker[] problemMarkers = cUnit.getResource().findMarkers(IMarker.PROBLEM, true, IResource.DEPTH_ONE);
					return problemMarkers.length == 1;
				} catch (CoreException e) {
				}
				return false;
			}
		
		}.waitForCondition(display, 3000);
		IMarker[] problemMarkers = cUnit.getResource().findMarkers(IMarker.PROBLEM, true, IResource.DEPTH_ONE);
		assertTrue("BoundPoint.aj should not have any problems", problemMarkers.length == 0); //$NON-NLS-1$

	}

	
}