tests/org.eclipse.wst.xsl.ui.tests/src/org/eclipse/wst/xsl/ui/tests/editor/XSLCompletionTest.java

/*******************************************************************************
 * Copyright (c) Standards for Technology in Automotive Retail 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:
 *     David Carver - STAR - intial API and implementation
 *******************************************************************************/

package org.eclipse.wst.xsl.ui.tests.editor;

import java.io.File;
import java.io.IOException;

import junit.framework.Assert;

import org.eclipse.core.resources.IFile;
import org.eclipse.core.resources.IProject;
import org.eclipse.core.resources.IProjectDescription;
import org.eclipse.core.resources.ResourcesPlugin;
import org.eclipse.core.runtime.CoreException;
import org.eclipse.core.runtime.NullProgressMonitor;
import org.eclipse.core.runtime.Path;
import org.eclipse.jface.text.contentassist.ICompletionProposal;
import org.eclipse.swt.SWT;
import org.eclipse.swt.widgets.Composite;
import org.eclipse.swt.widgets.Display;
import org.eclipse.swt.widgets.Shell;
import org.eclipse.ui.IEditorPart;
import org.eclipse.ui.PlatformUI;
import org.eclipse.ui.texteditor.ITextEditor;
import org.eclipse.wst.sse.core.StructuredModelManager;
import org.eclipse.wst.sse.core.internal.provisional.IModelManager;
import org.eclipse.wst.sse.core.internal.provisional.IStructuredModel;
import org.eclipse.wst.sse.core.internal.provisional.exceptions.ResourceAlreadyExists;
import org.eclipse.wst.sse.core.internal.provisional.exceptions.ResourceInUse;
import org.eclipse.wst.sse.core.internal.provisional.text.IStructuredDocument;
import org.eclipse.wst.sse.ui.internal.StructuredTextViewer;
import org.eclipse.wst.sse.ui.internal.contentassist.ContentAssistUtils;
import org.eclipse.wst.xml.core.internal.encoding.XMLDocumentLoader;
import org.eclipse.wst.xml.core.internal.provisional.document.IDOMNode;
import org.eclipse.wst.xsl.ui.internal.StructuredTextViewerConfigurationXSL;
import org.eclipse.wst.xsl.ui.internal.contentassist.XSLContentAssistProcessor;
import org.eclipse.wst.xsl.ui.tests.AbstractXSLUITest;

/**
 * Tests everything about code completion and code assistance.
 * 
 */
public class XSLCompletionTest extends AbstractXSLUITest {

	protected String projectName = null;
	protected String fileName = null;
	protected IFile file = null;
	protected IEditorPart textEditorPart = null;
	protected ITextEditor editor = null;

	protected XMLDocumentLoader xmlDocumentLoader = null;
	protected IStructuredDocument document = null;
	protected StructuredTextViewer sourceViewer = null;
	
	public XSLCompletionTest() {
		// TODO Auto-generated constructor stub
	}
	
	/**
	 * Setup the necessary projects, files, and source viewer for the
	 * tests.
	 */
	@Override
	protected void setUp() throws Exception {
		// TODO Auto-generated method stub
		super.setUp();
		projectName = "xsltestfiles";
		fileName = "utils.xsl";
		
        // Setup the Project and File to be used during the test.
		String xslFilePath = setupProject();
		file = ResourcesPlugin.getWorkspace().getRoot().getFile(new Path(xslFilePath));
		if (file != null && !file.exists()) {
			Assert.fail("Unable to locate " + fileName + " stylesheet.");
		}
		
		loadXSLFile();
		
		initializeSourceViewer();
	}

	protected void initializeSourceViewer() {
		// some test environments might not have a "real" display
		if(Display.getCurrent() != null) {
			
			Shell shell = null;
			Composite parent = null;
			
			if(PlatformUI.isWorkbenchRunning()) {
				shell = PlatformUI.getWorkbench().getActiveWorkbenchWindow().getShell();
			}
			else {	
				shell = new Shell(Display.getCurrent());
			}
			parent = new Composite(shell, SWT.NONE);
			
			// dummy viewer
			sourceViewer = new StructuredTextViewer(parent, null, null, false, SWT.NONE);
		}
		else {
			Assert.fail("Unable to run the test as a display must be available.");
		}
		
		configureSourceViewer();
	}

	protected void configureSourceViewer() {
		sourceViewer.configure(new StructuredTextViewerConfigurationXSL());
		
        sourceViewer.setDocument(document);
	}

	protected void loadXSLFile() throws ResourceAlreadyExists, ResourceInUse,
			IOException, CoreException {
		IModelManager modelManager = StructuredModelManager.getModelManager();
		IStructuredModel model = modelManager.getNewModelForEdit(file, true);
		document = model.getStructuredDocument();
	}

	protected String setupProject() {
		String xslFilePath = projectName + File.separator + fileName;
		IProjectDescription description = ResourcesPlugin.getWorkspace().newProjectDescription(projectName);

		IProject project = ResourcesPlugin.getWorkspace().getRoot().getProject(projectName);
		try {
			project.create(description, new NullProgressMonitor());
			project.open(new NullProgressMonitor());
		}
		catch (CoreException e) {
			
		}
		return xslFilePath;
	}
	
	@Override
	protected void tearDown() throws Exception {
		super.tearDown();
	}
	
	/**
	 * Get the content completion proposals at <code>lineNumber</code>, <code>columnNumber</code>.
	 * @param lineNumber
	 * @param columnNumber
	 * @return
	 * @throws Exception
	 */
	private ICompletionProposal[] getProposals(int offset) throws Exception {
    	return new XSLContentAssistProcessor().computeCompletionProposals(sourceViewer, offset); 
	}
		
	
	public void testGetNodeAtLine15() throws Exception {
		IDOMNode node = (IDOMNode) ContentAssistUtils.getNodeAt(sourceViewer, 631);
		assertEquals("Wrong node name returned:", "xsl:stylesheet", node.getNodeName());
		
	}
	
	public void testGetNodeAtLine16() throws Exception {
		IDOMNode node = (IDOMNode) ContentAssistUtils.getNodeAt(sourceViewer, 712);
		assertEquals("Wrong node name returned:", "xsl:template", node.getNodeName());
	}
	
	public void testGetNodeAtLine17() throws Exception {
		IDOMNode node = (IDOMNode) ContentAssistUtils.getNodeAt(sourceViewer, 748);
		assertEquals("Wrong node name returned:", "xsl:param", node.getNodeName());
	}
	
	public void testAttributeNotValueAvailable() throws Exception {
    	ICompletionProposal[] proposals = getProposals(838);
    	
    	assertTrue(proposals.length > 1);
    	ICompletionProposal proposal = proposals[0];
    	assertFalse("Found \"number(substring($date, 6, 2))\".", proposal.getDisplayString().equals("\"number(substring($date, 6, 2))\""));
	}
    
    public void testSelectAttributeProposalsAvailable() throws Exception {
    	int offset = sourceViewer.getDocument().getLineOffset(18) + 44;
    	String s = sourceViewer.getDocument().get(offset-1, 6 );
    	assertEquals("number",s);

    	
    	ICompletionProposal[] proposals = getProposals(838);
    	
    	assertTrue(proposals.length > 1);
    	ICompletionProposal proposal = proposals[3];
    	assertEquals("Wrong select item returned: ", "..", proposal.getDisplayString());
    }
 
    /**
     * Bug 240170
     * @throws Exception
     */
    public void testSelectAttributeProposalsNarrow() throws Exception {
    	int offset = sourceViewer.getDocument().getLineOffset(18) + 44;
    	String s = sourceViewer.getDocument().get(offset-9,9);
    	assertEquals("select=\"n",s);
    	
    	ICompletionProposal[] proposals = getProposals(offset);
    	assertEquals("Wrong xpath item returned: ", "name(node-set)", proposals[0].getDisplayString());
    	assertEquals("Wrong Number of items returned: ", 6, proposals.length);
    }

    public void testTestAttributeProposalsAvailable() throws Exception {
    	ICompletionProposal[] proposals = getProposals(1753);
    	assertTrue(proposals.length >= 1);
    	ICompletionProposal proposal = proposals[0];
    	assertTrue("Wrong attribute proposal returned:", proposal.getDisplayString().contains("disable-output-escaping"));
    }
    
    public void testXSLElementProposalsAvailable() throws Exception {
    	ICompletionProposal[] proposals = getProposals(1569);
    	assertTrue(proposals.length >= 2);
    	
    	ICompletionProposal proposal = proposals[1];
    	assertTrue("Can't find XSL element proposals.", proposal.getDisplayString().equals("xsl:otherwise"));
    }
    
    
}