testplugins/org.eclipse.objectteams.otdt.ui.tests/src/org/eclipse/objectteams/otdt/ui/tests/core/AddImportQuickFixTest.java - objectteams/org.eclipse.objectteams - Git at Google

 /**********************************************************************
  * This file is part of "Object Teams Development Tooling"-Software
  *
  * Copyright 2004, 2016 Fraunhofer Gesellschaft, Munich, Germany,
  * for its Fraunhofer Institute and Computer Architecture and Software
  * Technology (FIRST), Berlin, Germany and Technical University Berlin,
  * Germany.
  *
  * 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
  * $Id: AddImportQuickFixTest.java 23495 2010-02-05 23:15:16Z stephan $
  *
  * Please visit http://www.eclipse.org/objectteams for updates and contact.
  *
  * Contributors:
  * 	  Fraunhofer FIRST - Initial API and implementation
  * 	  Technical University Berlin - Initial API and implementation
  **********************************************************************/
 package org.eclipse.objectteams.otdt.ui.tests.core;

 import static org.junit.Assert.assertEquals;
 import static org.junit.Assert.assertNotNull;

 import java.util.ArrayList;

 import org.eclipse.core.runtime.CoreException;
 import org.eclipse.jdt.core.ICompilationUnit;
 import org.eclipse.jdt.core.IImportDeclaration;
 import org.eclipse.jdt.core.IPackageFragment;
 import org.eclipse.jdt.core.IPackageFragmentRoot;
 import org.eclipse.jdt.core.dom.CompilationUnit;
 import org.eclipse.jdt.internal.compiler.lookup.ExtraCompilerModifiers;
 import org.eclipse.jdt.ui.JavaUI;
 import org.eclipse.jdt.ui.tests.core.rules.ProjectTestSetup;
 import org.eclipse.jdt.ui.text.java.IJavaCompletionProposal;
 import org.eclipse.jdt.ui.text.java.correction.CUCorrectionProposal;
 import org.eclipse.jface.text.IDocument;
 import org.eclipse.objectteams.otdt.ui.tests.util.JavaProjectHelper;
 import org.eclipse.ui.IEditorPart;
 import org.eclipse.ui.internal.Workbench;
 import org.junit.Rule;
 import org.junit.Test;
 import org.junit.runner.RunWith;
 import org.junit.runners.JUnit4;

 @RunWith(JUnit4.class)
 public class AddImportQuickFixTest extends OTQuickFixTest {

 	@Rule
     public ProjectTestSetup projectsetup = new ProjectTestSetup();

 	@Test
 	public void testChangeFQNToBaseImport1() throws Exception {
 		// base:
 		IPackageFragment pack1base= fSourceFolder.createPackageFragment("test.base1", false, null);
 		StringBuffer buf= new StringBuffer();
 		buf = new StringBuffer();
 		buf.append("package test.base1;\n");
 		buf.append("public class Base1 {\n");
 		buf.append("}\n");
 		pack1base.createCompilationUnit("Base1.java", buf.toString(), false, null);

 		// team:
 		IPackageFragment pack1= fSourceFolder.createPackageFragment("test.import1", false, null);
 		buf = new StringBuffer();
 		buf.append("package test.import1;\n");
 		buf.append("public team class T {\n");
 		buf.append("	protected class R2 playedBy test.base1.Base1 {\n");
 		buf.append("	}\n");
 		buf.append("}\n");
 		ICompilationUnit cu= pack1.createCompilationUnit("T.java", buf.toString(), false, null);

 		CompilationUnit astRoot= getASTRoot(cu);
 		ArrayList<IJavaCompletionProposal> proposals= collectCorrections(cu, astRoot);
 		assertNumberOfProposals(proposals, 3);
 		assertCorrectLabels(proposals);

 		CUCorrectionProposal proposal= null;
 		for (Object prop : proposals)
 			if (prop instanceof CUCorrectionProposal && ((CUCorrectionProposal)prop).getRelevance() > 0)
 				proposal = (CUCorrectionProposal)prop;
 		assertNotNull("Need proposal with positive relevance", proposal);
 		assertEquals("Wrong quickfix label", "\"Change qualified reference to using 'import base test...Base1;'\"", proposal.getDisplayString());

 		String preview= getPreviewContent(proposal);

 		buf= new StringBuffer();
 		buf.append("package test.import1;\n");
 		buf.append("\n");
 		buf.append("import base test.base1.Base1;\n");
 		buf.append("\n");
 		buf.append("public team class T {\n");
 		buf.append("	protected class R2 playedBy Base1 {\n");
 		buf.append("	}\n");
 		buf.append("}\n");
 		assertEqualString(preview, buf.toString());
 	}

 	// ROFI variant:
 	@Test
 	public void testChangeFQNToBaseImport2() throws Exception {
 		// base:
 		IPackageFragment pack1base= fSourceFolder.createPackageFragment("test.base1", false, null);
 		StringBuffer buf= new StringBuffer();
 		buf = new StringBuffer();
 		buf.append("package test.base1;\n");
 		buf.append("public class Base1 {\n");
 		buf.append("}\n");
 		pack1base.createCompilationUnit("Base1.java", buf.toString(), false, null);

 		// team:
 		IPackageFragment pack1= fSourceFolder.createPackageFragment("test.import1", false, null);
 		buf = new StringBuffer();
 		buf.append("package test.import1;\n");
 		buf.append("public team class T {\n");
 		buf.append("}\n");
 		ICompilationUnit teamUnit = pack1.createCompilationUnit("T.java", buf.toString(), false, null);
 		// ROFI:
 		IPackageFragment teamPack = fSourceFolder.createPackageFragment("test.import1.T", true, null);
 		buf = new StringBuffer();
 		buf.append("team package test.import1.T;\n");
 		buf.append("protected class R2 playedBy test.base1.Base1 {\n");
 		buf.append("}\n");
 		ICompilationUnit cu= teamPack.createCompilationUnit("R2.java", buf.toString(), false, null);

 		CompilationUnit astRoot= getASTRoot(cu);
 		ArrayList<IJavaCompletionProposal> proposals= collectCorrections(cu, astRoot);
 		assertNumberOfProposals(proposals, 3);
 		assertCorrectLabels(proposals);

 		CUCorrectionProposal proposal= null;
 		for (Object prop : proposals)
 			if (prop instanceof CUCorrectionProposal && ((CUCorrectionProposal)prop).getRelevance() > 0)
 				proposal = (CUCorrectionProposal)prop;
 		assertNotNull("Need proposal with positive relevance", proposal);
 		assertEquals("Wrong quickfix label", "\"Change qualified reference to using 'import base test...Base1;'\"", proposal.getDisplayString());

 		String preview= getPreviewContent(proposal);

 		buf= new StringBuffer();
 		buf.append("team package test.import1.T;\n");
 		buf.append("protected class R2 playedBy Base1 {\n");
 		buf.append("}\n");
 		assertEqualString(preview, buf.toString());

 		// fetch and analyze team content which is not covered by the preview:
 		IEditorPart part= JavaUI.openInEditor(teamUnit);
 		assertNotNull("Failed to open team unit in editor", part);
 		IDocument document = JavaUI.getDocumentProvider().getDocument(part.getEditorInput());
 		proposal.apply(document);
 		// not needed: teamUnit.makeConsistent(null);
 		IImportDeclaration[] imports = teamUnit.getImports();
 		assertNotNull("Expected an import", imports);
 		assertEquals("Expected one import", 1, imports.length);
 		assertEquals("Unexpected import", "test.base1.Base1", imports[0].getElementName());
 		assertEquals("Expected base import", ExtraCompilerModifiers.AccBase, imports[0].getFlags());

 		// drop editors whose content has been modified by apply(),
 		// so that tearDown can delete files without asking the user:
 		Workbench.getInstance().getActiveWorkbenchWindow().getActivePage().closeAllEditors(false);
 	}

     // Bug 348076 - [assist][rewrite] changing import to import base not working when package name contains "base"
 	@Test
     public void testChangeImportToBaseImport1() throws Exception
     {
         IPackageFragmentRoot sourceFolder = JavaProjectHelper
                 .addSourceContainer(this.fJProject1, "src");

         IPackageFragment basePkg = sourceFolder.createPackageFragment(
                 "bug348076.base",
                 false,
                 null);
         StringBuffer buf = new StringBuffer();
         buf.append("package bug348076.base;\n");
         buf.append("public class B1 {\n");
         buf.append("}\n");
         basePkg.createCompilationUnit("B1.java", buf.toString(), false, null);

         IPackageFragment teamPkg = sourceFolder.createPackageFragment(
                 "teamPkg",
                 false,
                 null);
         buf = new StringBuffer();
         buf.append("package teamPkg;\n");
         buf.append("import bug348076.base.B1;\n");
         buf.append("public team class T1 {\n");
         buf.append("    public class R1 playedBy B1 {\n");
         buf.append("    }\n");
         buf.append("}\n");
         ICompilationUnit cu = teamPkg.createCompilationUnit("T1.java", buf
                 .toString(), false, null);

 		CompilationUnit astRoot= getASTRoot(cu);
 		ArrayList<IJavaCompletionProposal> proposals= collectCorrections(cu, astRoot);
 		assertNumberOfProposals(proposals, 3);
 		assertCorrectLabels(proposals);

 		CUCorrectionProposal proposal= (CUCorrectionProposal) proposals.get(2);
 		String preview= getPreviewContent(proposal);


         buf = new StringBuffer();
         buf.append("package teamPkg;\n");
         buf.append("import base bug348076.base.B1;\n");
         buf.append("public team class T1 {\n");
         buf.append("    public class R1 playedBy B1 {\n");
         buf.append("    }\n");
         buf.append("}\n");
         assertEqualString(preview, buf.toString());
     }

 	/* Adding a base import to a team on behalf of a role file. */
 	@Test
 	public void testAddBaseImportForRofi1() throws Exception {
 		// base:
 		IPackageFragment pack1base= fSourceFolder.createPackageFragment("test.base", false, null);
 		StringBuffer buf= new StringBuffer();
 		buf = new StringBuffer();
 		buf.append("package test.base;\n");
 		buf.append("protected class Base1 {\n");
 		buf.append("}\n");
 		pack1base.createCompilationUnit("Base1.java", buf.toString(), false, null);

 		// team:
 		IPackageFragment pack1= fSourceFolder.createPackageFragment("test.import1", false, null);
 		buf = new StringBuffer();
 		buf.append("package test.import1;\n");
 		buf.append("public team class T {\n");
 		buf.append("}\n");
 		pack1.createCompilationUnit("T.java", buf.toString(), false, null);

 		// role:
 		IPackageFragment pack1T= fSourceFolder.createPackageFragment("test.import1.T", false, null);
 		buf = new StringBuffer();
 		buf.append("team package test.import1.T;\n");
 		buf.append("protected class R2 playedBy Base1 {\n");
 		buf.append("}\n");
 		ICompilationUnit cu= pack1T.createCompilationUnit("R2.java", buf.toString(), false, null);


 		CompilationUnit astRoot= getASTRoot(cu);
 		ArrayList<IJavaCompletionProposal> proposals= collectCorrections(cu, astRoot);
 		assertNumberOfProposals(proposals, 6);
 		assertCorrectLabels(proposals);

 		CUCorrectionProposal proposal= (CUCorrectionProposal) proposals.get(0);
 		String preview= getPreviewContent(proposal);

 		buf= new StringBuffer();
 		buf.append("package test.import1;\n");
 		buf.append("\n");
 		buf.append("import base test.base.Base1;\n");
 		buf.append("\n");
 		buf.append("public team class T {\n");
 		buf.append("}\n");
 		assertEqualString(preview, buf.toString());
 	}

 	// this time files already have content
 	@Test
 	public void testAddBaseImportForRofi2() throws Exception {
 		StringBuffer buf;
 		// base:
 		IPackageFragment pack1base= fSourceFolder.createPackageFragment("test.base", false, null);
 		buf = new StringBuffer();
 		buf.append("package test.base;\n");
 		buf.append("protected class Base1 {\n");
 		buf.append("}\n");
 		pack1base.createCompilationUnit("Base1.java", buf.toString(), false, null);

 		// team:
 		IPackageFragment pack1= fSourceFolder.createPackageFragment("test.import1", false, null);
 		buf = new StringBuffer();
 		buf.append("package test.import1;\n");
 		buf.append("\n");
 		buf.append("import java.util.List;\n");
 		buf.append("\n");
 		buf.append("import base java.awt.Window;\n");
 		buf.append("\n");
 		buf.append("public team class T {\n");
 		buf.append("    protected class R playedBy Window;\n");
 		buf.append("    \n");
 		buf.append("    List l;\n");
 		buf.append("    \n");
 		buf.append("}\n");
 		pack1.createCompilationUnit("T.java", buf.toString(), false, null);

 		// role:
 		IPackageFragment pack1T= fSourceFolder.createPackageFragment("test.import1.T", false, null);
 		buf = new StringBuffer();
 		buf.append("team package test.import1.T;\n");
 		buf.append("protected class R2 playedBy Base1 {\n");
 		buf.append("	void foo() {}\n");
 		buf.append("}\n");
 		ICompilationUnit cu= pack1T.createCompilationUnit("R2.java", buf.toString(), false, null);


 		CompilationUnit astRoot= getASTRoot(cu);
 		ArrayList<IJavaCompletionProposal> proposals= collectCorrections(cu, astRoot);
 		assertNumberOfProposals(proposals, 6);
 		assertCorrectLabels(proposals);

 		CUCorrectionProposal proposal= (CUCorrectionProposal) proposals.get(0);
 		String preview= getPreviewContent(proposal);

 		buf= new StringBuffer();
 		buf.append("package test.import1;\n");
 		buf.append("\n");
 		buf.append("import java.util.List;\n");
 		buf.append("\n");
 		buf.append("import base java.awt.Window;\n");
 		buf.append("import base test.base.Base1;\n");
 		buf.append("\n");
 		buf.append("public team class T {\n");
 		buf.append("    protected class R playedBy Window;\n");
 		buf.append("    \n");
 		buf.append("    List l;\n");
 		buf.append("    \n");
 		buf.append("}\n");
 		assertEqualString(preview, buf.toString());
 	}

 	// for role types rather than proposing an import change type to anchored:
 	@Test
 	public void testChangeTypeToAnchored1() throws CoreException {
 		StringBuffer buf;
 		// team:
 		IPackageFragment pack1= fSourceFolder.createPackageFragment("test.import1", false, null);
 		buf = new StringBuffer();
 		buf.append("package test.import1;\n");
 		buf.append("\n");
 		buf.append("public team class T {\n");
 		buf.append("    public class R {}\n");
 		buf.append("}\n");
 		pack1.createCompilationUnit("T.java", buf.toString(), false, null);

 		// client
 		IPackageFragment pack2= fSourceFolder.createPackageFragment("test.import2", false, null);
 		buf = new StringBuffer();
 		buf.append("package test.import2;\n");
 		buf.append("\n");
 		buf.append("import test.import1.T;\n");
 		buf.append("\n");
 		buf.append("public class C {\n");
 		buf.append("    final T t = new T();\n");
 		buf.append("    void test() {\n");
 		buf.append("    	R r = t.new R();\n");
 		buf.append("    }\n");
 		buf.append("}\n");
 		ICompilationUnit cu = pack2.createCompilationUnit("C.java", buf.toString(), false, null);

 		CompilationUnit astRoot= getASTRoot(cu);
 		ArrayList<IJavaCompletionProposal> proposals= collectCorrections(cu, astRoot);
 		assertNumberOfProposals(proposals, 7);
 		assertCorrectLabels(proposals);

 		CUCorrectionProposal proposal= (CUCorrectionProposal) proposals.get(0);
 		String preview= getPreviewContent(proposal);

 		buf= new StringBuffer();
 		buf.append("package test.import2;\n");
 		buf.append("\n");
 		buf.append("import test.import1.T;\n");
 		buf.append("\n");
 		buf.append("public class C {\n");
 		buf.append("    final T t = new T();\n");
 		buf.append("    void test() {\n");
 		buf.append("    	R<@t> r = t.new R();\n");
 		buf.append("    }\n");
 		buf.append("}\n");
 		assertEqualString(preview, buf.toString());
 	}

 	// for role types rather than proposing an import change type to anchored (role file):
 	@Test
 	public void testChangeTypeToAnchored2() throws CoreException {
 		StringBuffer buf;
 		// team:
 		IPackageFragment pack1= fSourceFolder.createPackageFragment("test.import1", false, null);
 		buf = new StringBuffer();
 		buf.append("package test.import1;\n");
 		buf.append("\n");
 		buf.append("/** @role R */\n");
 		buf.append("public team class T {\n");
 		buf.append("}\n");
 		pack1.createCompilationUnit("T.java", buf.toString(), false, null);

 		IPackageFragment teampack= fSourceFolder.createPackageFragment("test.import1.T", false, null);
 		buf = new StringBuffer();
 		buf.append("team package test.import1.T;\n");
 		buf.append("\n");
 		buf.append("public class R {}\n");
 		teampack.createCompilationUnit("R.java", buf.toString(), false, null);

 		// client
 		IPackageFragment pack2= fSourceFolder.createPackageFragment("test.import2", false, null);
 		buf = new StringBuffer();
 		buf.append("package test.import2;\n");
 		buf.append("\n");
 		buf.append("import test.import1.T;\n");
 		buf.append("\n");
 		buf.append("public class C {\n");
 		buf.append("    final T t = new T();\n");
 		buf.append("    void test() {\n");
 		buf.append("    	R r = t.new R();\n");
 		buf.append("    }\n");
 		buf.append("}\n");
 		ICompilationUnit cu = pack2.createCompilationUnit("C.java", buf.toString(), false, null);

 		CompilationUnit astRoot= getASTRoot(cu);
 		ArrayList<IJavaCompletionProposal> proposals= collectCorrections(cu, astRoot);
 		assertNumberOfProposals(proposals, 7);
 		assertCorrectLabels(proposals);

 		CUCorrectionProposal proposal= (CUCorrectionProposal) proposals.get(0);
 		String preview= getPreviewContent(proposal);

 		buf= new StringBuffer();
 		buf.append("package test.import2;\n");
 		buf.append("\n");
 		buf.append("import test.import1.T;\n");
 		buf.append("\n");
 		buf.append("public class C {\n");
 		buf.append("    final T t = new T();\n");
 		buf.append("    void test() {\n");
 		buf.append("    	R<@t> r = t.new R();\n");
 		buf.append("    }\n");
 		buf.append("}\n");
 		assertEqualString(preview, buf.toString());
 	}
 }
	/**********************************************************************
	* This file is part of "Object Teams Development Tooling"-Software
	*
	* Copyright 2004, 2016 Fraunhofer Gesellschaft, Munich, Germany,
	* for its Fraunhofer Institute and Computer Architecture and Software
	* Technology (FIRST), Berlin, Germany and Technical University Berlin,
	* Germany.
	*
	* 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
	* $Id: AddImportQuickFixTest.java 23495 2010-02-05 23:15:16Z stephan $
	*
	* Please visit http://www.eclipse.org/objectteams for updates and contact.
	*
	* Contributors:
	* Fraunhofer FIRST - Initial API and implementation
	* Technical University Berlin - Initial API and implementation
	**********************************************************************/
	package org.eclipse.objectteams.otdt.ui.tests.core;

	import static org.junit.Assert.assertEquals;
	import static org.junit.Assert.assertNotNull;

	import java.util.ArrayList;

	import org.eclipse.core.runtime.CoreException;
	import org.eclipse.jdt.core.ICompilationUnit;
	import org.eclipse.jdt.core.IImportDeclaration;
	import org.eclipse.jdt.core.IPackageFragment;
	import org.eclipse.jdt.core.IPackageFragmentRoot;
	import org.eclipse.jdt.core.dom.CompilationUnit;
	import org.eclipse.jdt.internal.compiler.lookup.ExtraCompilerModifiers;
	import org.eclipse.jdt.ui.JavaUI;
	import org.eclipse.jdt.ui.tests.core.rules.ProjectTestSetup;
	import org.eclipse.jdt.ui.text.java.IJavaCompletionProposal;
	import org.eclipse.jdt.ui.text.java.correction.CUCorrectionProposal;
	import org.eclipse.jface.text.IDocument;
	import org.eclipse.objectteams.otdt.ui.tests.util.JavaProjectHelper;
	import org.eclipse.ui.IEditorPart;
	import org.eclipse.ui.internal.Workbench;
	import org.junit.Rule;
	import org.junit.Test;
	import org.junit.runner.RunWith;
	import org.junit.runners.JUnit4;

	@RunWith(JUnit4.class)
	public class AddImportQuickFixTest extends OTQuickFixTest {

	@Rule
	public ProjectTestSetup projectsetup = new ProjectTestSetup();

	@Test
	public void testChangeFQNToBaseImport1() throws Exception {
	// base:
	IPackageFragment pack1base= fSourceFolder.createPackageFragment("test.base1", false, null);
	StringBuffer buf= new StringBuffer();
	buf = new StringBuffer();
	buf.append("package test.base1;\n");
	buf.append("public class Base1 {\n");
	buf.append("}\n");
	pack1base.createCompilationUnit("Base1.java", buf.toString(), false, null);

	// team:
	IPackageFragment pack1= fSourceFolder.createPackageFragment("test.import1", false, null);
	buf = new StringBuffer();
	buf.append("package test.import1;\n");
	buf.append("public team class T {\n");
	buf.append(" protected class R2 playedBy test.base1.Base1 {\n");
	buf.append(" }\n");
	buf.append("}\n");
	ICompilationUnit cu= pack1.createCompilationUnit("T.java", buf.toString(), false, null);

	CompilationUnit astRoot= getASTRoot(cu);
	ArrayList<IJavaCompletionProposal> proposals= collectCorrections(cu, astRoot);
	assertNumberOfProposals(proposals, 3);
	assertCorrectLabels(proposals);

	CUCorrectionProposal proposal= null;
	for (Object prop : proposals)
	if (prop instanceof CUCorrectionProposal && ((CUCorrectionProposal)prop).getRelevance() > 0)
	proposal = (CUCorrectionProposal)prop;
	assertNotNull("Need proposal with positive relevance", proposal);
	assertEquals("Wrong quickfix label", "\"Change qualified reference to using 'import base test...Base1;'\"", proposal.getDisplayString());

	String preview= getPreviewContent(proposal);

	buf= new StringBuffer();
	buf.append("package test.import1;\n");
	buf.append("\n");
	buf.append("import base test.base1.Base1;\n");
	buf.append("\n");
	buf.append("public team class T {\n");
	buf.append(" protected class R2 playedBy Base1 {\n");
	buf.append(" }\n");
	buf.append("}\n");
	assertEqualString(preview, buf.toString());
	}

	// ROFI variant:
	@Test
	public void testChangeFQNToBaseImport2() throws Exception {
	// base:
	IPackageFragment pack1base= fSourceFolder.createPackageFragment("test.base1", false, null);
	StringBuffer buf= new StringBuffer();
	buf = new StringBuffer();
	buf.append("package test.base1;\n");
	buf.append("public class Base1 {\n");
	buf.append("}\n");
	pack1base.createCompilationUnit("Base1.java", buf.toString(), false, null);

	// team:
	IPackageFragment pack1= fSourceFolder.createPackageFragment("test.import1", false, null);
	buf = new StringBuffer();
	buf.append("package test.import1;\n");
	buf.append("public team class T {\n");
	buf.append("}\n");
	ICompilationUnit teamUnit = pack1.createCompilationUnit("T.java", buf.toString(), false, null);
	// ROFI:
	IPackageFragment teamPack = fSourceFolder.createPackageFragment("test.import1.T", true, null);
	buf = new StringBuffer();
	buf.append("team package test.import1.T;\n");
	buf.append("protected class R2 playedBy test.base1.Base1 {\n");
	buf.append("}\n");
	ICompilationUnit cu= teamPack.createCompilationUnit("R2.java", buf.toString(), false, null);

	CompilationUnit astRoot= getASTRoot(cu);
	ArrayList<IJavaCompletionProposal> proposals= collectCorrections(cu, astRoot);
	assertNumberOfProposals(proposals, 3);
	assertCorrectLabels(proposals);

	CUCorrectionProposal proposal= null;
	for (Object prop : proposals)
	if (prop instanceof CUCorrectionProposal && ((CUCorrectionProposal)prop).getRelevance() > 0)
	proposal = (CUCorrectionProposal)prop;
	assertNotNull("Need proposal with positive relevance", proposal);
	assertEquals("Wrong quickfix label", "\"Change qualified reference to using 'import base test...Base1;'\"", proposal.getDisplayString());

	String preview= getPreviewContent(proposal);

	buf= new StringBuffer();
	buf.append("team package test.import1.T;\n");
	buf.append("protected class R2 playedBy Base1 {\n");
	buf.append("}\n");
	assertEqualString(preview, buf.toString());

	// fetch and analyze team content which is not covered by the preview:
	IEditorPart part= JavaUI.openInEditor(teamUnit);
	assertNotNull("Failed to open team unit in editor", part);
	IDocument document = JavaUI.getDocumentProvider().getDocument(part.getEditorInput());
	proposal.apply(document);
	// not needed: teamUnit.makeConsistent(null);
	IImportDeclaration[] imports = teamUnit.getImports();
	assertNotNull("Expected an import", imports);
	assertEquals("Expected one import", 1, imports.length);
	assertEquals("Unexpected import", "test.base1.Base1", imports[0].getElementName());
	assertEquals("Expected base import", ExtraCompilerModifiers.AccBase, imports[0].getFlags());

	// drop editors whose content has been modified by apply(),
	// so that tearDown can delete files without asking the user:
	Workbench.getInstance().getActiveWorkbenchWindow().getActivePage().closeAllEditors(false);
	}

	// Bug 348076 - [assist][rewrite] changing import to import base not working when package name contains "base"
	@Test
	public void testChangeImportToBaseImport1() throws Exception
	{
	IPackageFragmentRoot sourceFolder = JavaProjectHelper
	.addSourceContainer(this.fJProject1, "src");

	IPackageFragment basePkg = sourceFolder.createPackageFragment(
	"bug348076.base",
	false,
	null);
	StringBuffer buf = new StringBuffer();
	buf.append("package bug348076.base;\n");
	buf.append("public class B1 {\n");
	buf.append("}\n");
	basePkg.createCompilationUnit("B1.java", buf.toString(), false, null);

	IPackageFragment teamPkg = sourceFolder.createPackageFragment(
	"teamPkg",
	false,
	null);
	buf = new StringBuffer();
	buf.append("package teamPkg;\n");
	buf.append("import bug348076.base.B1;\n");
	buf.append("public team class T1 {\n");
	buf.append(" public class R1 playedBy B1 {\n");
	buf.append(" }\n");
	buf.append("}\n");
	ICompilationUnit cu = teamPkg.createCompilationUnit("T1.java", buf
	.toString(), false, null);

	CompilationUnit astRoot= getASTRoot(cu);
	ArrayList<IJavaCompletionProposal> proposals= collectCorrections(cu, astRoot);
	assertNumberOfProposals(proposals, 3);
	assertCorrectLabels(proposals);

	CUCorrectionProposal proposal= (CUCorrectionProposal) proposals.get(2);
	String preview= getPreviewContent(proposal);


	buf = new StringBuffer();
	buf.append("package teamPkg;\n");
	buf.append("import base bug348076.base.B1;\n");
	buf.append("public team class T1 {\n");
	buf.append(" public class R1 playedBy B1 {\n");
	buf.append(" }\n");
	buf.append("}\n");
	assertEqualString(preview, buf.toString());
	}

	/* Adding a base import to a team on behalf of a role file. */
	@Test
	public void testAddBaseImportForRofi1() throws Exception {
	// base:
	IPackageFragment pack1base= fSourceFolder.createPackageFragment("test.base", false, null);
	StringBuffer buf= new StringBuffer();
	buf = new StringBuffer();
	buf.append("package test.base;\n");
	buf.append("protected class Base1 {\n");
	buf.append("}\n");
	pack1base.createCompilationUnit("Base1.java", buf.toString(), false, null);

	// team:
	IPackageFragment pack1= fSourceFolder.createPackageFragment("test.import1", false, null);
	buf = new StringBuffer();
	buf.append("package test.import1;\n");
	buf.append("public team class T {\n");
	buf.append("}\n");
	pack1.createCompilationUnit("T.java", buf.toString(), false, null);

	// role:
	IPackageFragment pack1T= fSourceFolder.createPackageFragment("test.import1.T", false, null);
	buf = new StringBuffer();
	buf.append("team package test.import1.T;\n");
	buf.append("protected class R2 playedBy Base1 {\n");
	buf.append("}\n");
	ICompilationUnit cu= pack1T.createCompilationUnit("R2.java", buf.toString(), false, null);


	CompilationUnit astRoot= getASTRoot(cu);
	ArrayList<IJavaCompletionProposal> proposals= collectCorrections(cu, astRoot);
	assertNumberOfProposals(proposals, 6);
	assertCorrectLabels(proposals);

	CUCorrectionProposal proposal= (CUCorrectionProposal) proposals.get(0);
	String preview= getPreviewContent(proposal);

	buf= new StringBuffer();
	buf.append("package test.import1;\n");
	buf.append("\n");
	buf.append("import base test.base.Base1;\n");
	buf.append("\n");
	buf.append("public team class T {\n");
	buf.append("}\n");
	assertEqualString(preview, buf.toString());
	}

	// this time files already have content
	@Test
	public void testAddBaseImportForRofi2() throws Exception {
	StringBuffer buf;
	// base:
	IPackageFragment pack1base= fSourceFolder.createPackageFragment("test.base", false, null);
	buf = new StringBuffer();
	buf.append("package test.base;\n");
	buf.append("protected class Base1 {\n");
	buf.append("}\n");
	pack1base.createCompilationUnit("Base1.java", buf.toString(), false, null);

	// team:
	IPackageFragment pack1= fSourceFolder.createPackageFragment("test.import1", false, null);
	buf = new StringBuffer();
	buf.append("package test.import1;\n");
	buf.append("\n");
	buf.append("import java.util.List;\n");
	buf.append("\n");
	buf.append("import base java.awt.Window;\n");
	buf.append("\n");
	buf.append("public team class T {\n");
	buf.append(" protected class R playedBy Window;\n");
	buf.append(" \n");
	buf.append(" List l;\n");
	buf.append(" \n");
	buf.append("}\n");
	pack1.createCompilationUnit("T.java", buf.toString(), false, null);

	// role:
	IPackageFragment pack1T= fSourceFolder.createPackageFragment("test.import1.T", false, null);
	buf = new StringBuffer();
	buf.append("team package test.import1.T;\n");
	buf.append("protected class R2 playedBy Base1 {\n");
	buf.append(" void foo() {}\n");
	buf.append("}\n");
	ICompilationUnit cu= pack1T.createCompilationUnit("R2.java", buf.toString(), false, null);


	CompilationUnit astRoot= getASTRoot(cu);
	ArrayList<IJavaCompletionProposal> proposals= collectCorrections(cu, astRoot);
	assertNumberOfProposals(proposals, 6);
	assertCorrectLabels(proposals);

	CUCorrectionProposal proposal= (CUCorrectionProposal) proposals.get(0);
	String preview= getPreviewContent(proposal);

	buf= new StringBuffer();
	buf.append("package test.import1;\n");
	buf.append("\n");
	buf.append("import java.util.List;\n");
	buf.append("\n");
	buf.append("import base java.awt.Window;\n");
	buf.append("import base test.base.Base1;\n");
	buf.append("\n");
	buf.append("public team class T {\n");
	buf.append(" protected class R playedBy Window;\n");
	buf.append(" \n");
	buf.append(" List l;\n");
	buf.append(" \n");
	buf.append("}\n");
	assertEqualString(preview, buf.toString());
	}

	// for role types rather than proposing an import change type to anchored:
	@Test
	public void testChangeTypeToAnchored1() throws CoreException {
	StringBuffer buf;
	// team:
	IPackageFragment pack1= fSourceFolder.createPackageFragment("test.import1", false, null);
	buf = new StringBuffer();
	buf.append("package test.import1;\n");
	buf.append("\n");
	buf.append("public team class T {\n");
	buf.append(" public class R {}\n");
	buf.append("}\n");
	pack1.createCompilationUnit("T.java", buf.toString(), false, null);

	// client
	IPackageFragment pack2= fSourceFolder.createPackageFragment("test.import2", false, null);
	buf = new StringBuffer();
	buf.append("package test.import2;\n");
	buf.append("\n");
	buf.append("import test.import1.T;\n");
	buf.append("\n");
	buf.append("public class C {\n");
	buf.append(" final T t = new T();\n");
	buf.append(" void test() {\n");
	buf.append(" R r = t.new R();\n");
	buf.append(" }\n");
	buf.append("}\n");
	ICompilationUnit cu = pack2.createCompilationUnit("C.java", buf.toString(), false, null);

	CompilationUnit astRoot= getASTRoot(cu);
	ArrayList<IJavaCompletionProposal> proposals= collectCorrections(cu, astRoot);
	assertNumberOfProposals(proposals, 7);
	assertCorrectLabels(proposals);

	CUCorrectionProposal proposal= (CUCorrectionProposal) proposals.get(0);
	String preview= getPreviewContent(proposal);

	buf= new StringBuffer();
	buf.append("package test.import2;\n");
	buf.append("\n");
	buf.append("import test.import1.T;\n");
	buf.append("\n");
	buf.append("public class C {\n");
	buf.append(" final T t = new T();\n");
	buf.append(" void test() {\n");
	buf.append(" R<@t> r = t.new R();\n");
	buf.append(" }\n");
	buf.append("}\n");
	assertEqualString(preview, buf.toString());
	}

	// for role types rather than proposing an import change type to anchored (role file):
	@Test
	public void testChangeTypeToAnchored2() throws CoreException {
	StringBuffer buf;
	// team:
	IPackageFragment pack1= fSourceFolder.createPackageFragment("test.import1", false, null);
	buf = new StringBuffer();
	buf.append("package test.import1;\n");
	buf.append("\n");
	buf.append("/** @role R */\n");
	buf.append("public team class T {\n");
	buf.append("}\n");
	pack1.createCompilationUnit("T.java", buf.toString(), false, null);

	IPackageFragment teampack= fSourceFolder.createPackageFragment("test.import1.T", false, null);
	buf = new StringBuffer();
	buf.append("team package test.import1.T;\n");
	buf.append("\n");
	buf.append("public class R {}\n");
	teampack.createCompilationUnit("R.java", buf.toString(), false, null);

	// client
	IPackageFragment pack2= fSourceFolder.createPackageFragment("test.import2", false, null);
	buf = new StringBuffer();
	buf.append("package test.import2;\n");
	buf.append("\n");
	buf.append("import test.import1.T;\n");
	buf.append("\n");
	buf.append("public class C {\n");
	buf.append(" final T t = new T();\n");
	buf.append(" void test() {\n");
	buf.append(" R r = t.new R();\n");
	buf.append(" }\n");
	buf.append("}\n");
	ICompilationUnit cu = pack2.createCompilationUnit("C.java", buf.toString(), false, null);

	CompilationUnit astRoot= getASTRoot(cu);
	ArrayList<IJavaCompletionProposal> proposals= collectCorrections(cu, astRoot);
	assertNumberOfProposals(proposals, 7);
	assertCorrectLabels(proposals);

	CUCorrectionProposal proposal= (CUCorrectionProposal) proposals.get(0);
	String preview= getPreviewContent(proposal);

	buf= new StringBuffer();
	buf.append("package test.import2;\n");
	buf.append("\n");
	buf.append("import test.import1.T;\n");
	buf.append("\n");
	buf.append("public class C {\n");
	buf.append(" final T t = new T();\n");
	buf.append(" void test() {\n");
	buf.append(" R<@t> r = t.new R();\n");
	buf.append(" }\n");
	buf.append("}\n");
	assertEqualString(preview, buf.toString());
	}
	}