| /******************************************************************************* |
| * Copyright (c) 2000, 2008 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 |
| *******************************************************************************/ |
| package org.eclipse.jdt.core.tests.rewrite.describing; |
| |
| import java.io.IOException; |
| |
| import junit.framework.Test; |
| import junit.framework.TestSuite; |
| |
| import org.eclipse.core.runtime.CoreException; |
| import org.eclipse.jdt.core.BindingKey; |
| import org.eclipse.jdt.core.ICompilationUnit; |
| import org.eclipse.jdt.core.IJavaElement; |
| import org.eclipse.jdt.core.IJavaProject; |
| import org.eclipse.jdt.core.IPackageFragment; |
| import org.eclipse.jdt.core.IPackageFragmentRoot; |
| import org.eclipse.jdt.core.IType; |
| import org.eclipse.jdt.core.JavaCore; |
| import org.eclipse.jdt.core.JavaModelException; |
| import org.eclipse.jdt.core.dom.AST; |
| import org.eclipse.jdt.core.dom.rewrite.ImportRewrite; |
| import org.eclipse.jdt.core.formatter.DefaultCodeFormatterConstants; |
| import org.eclipse.jdt.core.tests.model.AbstractJavaModelTests; |
| import org.eclipse.jface.text.BadLocationException; |
| import org.eclipse.jface.text.Document; |
| import org.eclipse.text.edits.MalformedTreeException; |
| import org.eclipse.text.edits.TextEdit; |
| import org.osgi.service.prefs.BackingStoreException; |
| |
| public class ImportRewriteTest extends AbstractJavaModelTests { |
| |
| private static final Class THIS= ImportRewriteTest.class; |
| |
| protected IPackageFragmentRoot sourceFolder; |
| |
| public ImportRewriteTest(String name) { |
| super(name); |
| } |
| |
| public static Test allTests() { |
| return new Suite(THIS); |
| } |
| |
| public static Test setUpTest(Test someTest) { |
| TestSuite suite= new Suite("one test"); |
| suite.addTest(someTest); |
| return suite; |
| } |
| |
| public static Test suite() { |
| return allTests(); |
| } |
| |
| protected void setUp() throws Exception { |
| super.setUp(); |
| |
| IJavaProject proj= createJavaProject("P", new String[] {"src"}, new String[] {"JCL_LIB"}, "bin"); |
| proj.setOption(DefaultCodeFormatterConstants.FORMATTER_TAB_CHAR, JavaCore.SPACE); |
| proj.setOption(DefaultCodeFormatterConstants.FORMATTER_TAB_SIZE, "4"); |
| proj.setOption(JavaCore.COMPILER_COMPLIANCE, JavaCore.VERSION_1_5); |
| proj.setOption(JavaCore.COMPILER_PB_ASSERT_IDENTIFIER, JavaCore.ERROR); |
| proj.setOption(JavaCore.COMPILER_SOURCE, JavaCore.VERSION_1_5); |
| proj.setOption(JavaCore.COMPILER_CODEGEN_TARGET_PLATFORM, JavaCore.VERSION_1_5); |
| proj.setOption(DefaultCodeFormatterConstants.FORMATTER_NUMBER_OF_EMPTY_LINES_TO_PRESERVE, String.valueOf(99)); |
| |
| proj.setOption(DefaultCodeFormatterConstants.FORMATTER_BLANK_LINES_BETWEEN_IMPORT_GROUPS, String.valueOf(1)); |
| |
| |
| this.sourceFolder = getPackageFragmentRoot("P", "src"); |
| |
| waitUntilIndexesReady(); |
| } |
| |
| protected void tearDown() throws Exception { |
| deleteProject("P"); |
| super.tearDown(); |
| } |
| |
| |
| public void testAddImports1() throws Exception { |
| |
| IPackageFragment pack1= this.sourceFolder.createPackageFragment("pack1", false, null); |
| StringBuffer buf= new StringBuffer(); |
| buf.append("package pack1;\n"); |
| buf.append("\n"); |
| buf.append("import java.util.Set;\n"); |
| buf.append("import java.util.Vector;\n"); |
| buf.append("import java.util.Map;\n"); |
| buf.append("\n"); |
| buf.append("import pack.List;\n"); |
| buf.append("import pack.List2;\n"); |
| buf.append("\n"); |
| buf.append("public class C {\n"); |
| buf.append("}\n"); |
| ICompilationUnit cu= pack1.createCompilationUnit("C.java", buf.toString(), false, null); |
| |
| String[] order= new String[] { "java", "com", "pack" }; |
| |
| ImportRewrite imports= newImportsRewrite(cu, order, 2, 2, true); |
| imports.addImport("java.net.Socket"); |
| imports.addImport("p.A"); |
| imports.addImport("com.something.Foo"); |
| |
| apply(imports); |
| |
| buf= new StringBuffer(); |
| buf.append("package pack1;\n"); |
| buf.append("\n"); |
| buf.append("import java.net.Socket;\n"); |
| buf.append("import java.util.Set;\n"); |
| buf.append("import java.util.Vector;\n"); |
| buf.append("import java.util.Map;\n"); |
| buf.append("\n"); |
| buf.append("import com.something.Foo;\n"); |
| buf.append("\n"); |
| buf.append("import p.A;\n"); |
| buf.append("\n"); |
| buf.append("import pack.List;\n"); |
| buf.append("import pack.List2;\n"); |
| buf.append("\n"); |
| buf.append("public class C {\n"); |
| buf.append("}\n"); |
| assertEqualString(cu.getSource(), buf.toString()); |
| } |
| |
| public void testAddImportsNoEmptyLines() throws Exception { |
| |
| this.sourceFolder.getJavaProject().setOption(DefaultCodeFormatterConstants.FORMATTER_BLANK_LINES_BETWEEN_IMPORT_GROUPS, String.valueOf(0)); |
| |
| IPackageFragment pack1= this.sourceFolder.createPackageFragment("pack1", false, null); |
| StringBuffer buf= new StringBuffer(); |
| buf.append("package pack1;\n"); |
| buf.append("\n"); |
| buf.append("import java.util.Set;\n"); |
| buf.append("\n"); |
| buf.append("public class C {\n"); |
| buf.append("}\n"); |
| ICompilationUnit cu= pack1.createCompilationUnit("C.java", buf.toString(), false, null); |
| |
| String[] order= new String[] { "java.util", "java.new", "p" }; |
| |
| ImportRewrite imports= newImportsRewrite(cu, order, 2, 2, true); |
| |
| imports.addImport("java.net.Socket"); |
| imports.addImport("p.A"); |
| |
| apply(imports); |
| |
| buf= new StringBuffer(); |
| buf.append("package pack1;\n"); |
| buf.append("\n"); |
| buf.append("import java.util.Set;\n"); |
| buf.append("import java.net.Socket;\n"); |
| buf.append("import p.A;\n"); |
| buf.append("\n"); |
| buf.append("public class C {\n"); |
| buf.append("}\n"); |
| assertEqualString(cu.getSource(), buf.toString()); |
| } |
| |
| public void testAddImportsMoreEmptyLines() throws Exception { |
| |
| this.sourceFolder.getJavaProject().setOption(DefaultCodeFormatterConstants.FORMATTER_BLANK_LINES_BETWEEN_IMPORT_GROUPS, String.valueOf(2)); |
| |
| IPackageFragment pack1= this.sourceFolder.createPackageFragment("pack1", false, null); |
| StringBuffer buf= new StringBuffer(); |
| buf.append("package pack1;\n"); |
| buf.append("\n"); |
| buf.append("import java.util.Set;\n"); |
| buf.append("\n"); |
| buf.append("public class C {\n"); |
| buf.append("}\n"); |
| ICompilationUnit cu= pack1.createCompilationUnit("C.java", buf.toString(), false, null); |
| |
| String[] order= new String[] { "java.util", "java.new", "p" }; |
| |
| ImportRewrite imports= newImportsRewrite(cu, order, 2, 2, true); |
| |
| imports.addImport("java.net.Socket"); |
| imports.addImport("p.A"); |
| |
| apply(imports); |
| |
| buf= new StringBuffer(); |
| buf.append("package pack1;\n"); |
| buf.append("\n"); |
| buf.append("import java.util.Set;\n"); |
| buf.append("\n"); |
| buf.append("\n"); |
| buf.append("import java.net.Socket;\n"); |
| buf.append("\n"); |
| buf.append("\n"); |
| buf.append("import p.A;\n"); |
| buf.append("\n"); |
| buf.append("public class C {\n"); |
| buf.append("}\n"); |
| assertEqualString(cu.getSource(), buf.toString()); |
| } |
| |
| public void testAddImports2() throws Exception { |
| |
| IPackageFragment pack1= this.sourceFolder.createPackageFragment("pack1", false, null); |
| StringBuffer buf= new StringBuffer(); |
| buf.append("package pack1;\n"); |
| buf.append("\n"); |
| buf.append("import java.util.Set;\n"); |
| buf.append("import java.util.Vector;\n"); |
| buf.append("\n"); |
| buf.append("public class C {\n"); |
| buf.append("}\n"); |
| ICompilationUnit cu= pack1.createCompilationUnit("C.java", buf.toString(), false, null); |
| |
| String[] order= new String[] { "java", "java.util", "com", "pack" }; |
| |
| ImportRewrite imports= newImportsRewrite(cu, order, 2, 2, true); |
| imports.addImport("java.x.Socket"); |
| |
| apply(imports); |
| |
| buf= new StringBuffer(); |
| buf.append("package pack1;\n"); |
| buf.append("\n"); |
| buf.append("import java.x.Socket;\n"); |
| buf.append("\n"); |
| buf.append("import java.util.Set;\n"); |
| buf.append("import java.util.Vector;\n"); |
| buf.append("\n"); |
| buf.append("public class C {\n"); |
| buf.append("}\n"); |
| assertEqualString(cu.getSource(), buf.toString()); |
| } |
| |
| |
| public void testAddImports3() throws Exception { |
| |
| IPackageFragment pack1= this.sourceFolder.createPackageFragment("pack1", false, null); |
| StringBuffer buf= new StringBuffer(); |
| buf.append("package pack1;\n"); |
| buf.append("\n"); |
| buf.append("import java.util.Set; // comment\n"); |
| buf.append("\n"); |
| buf.append("public class C {\n"); |
| buf.append("}\n"); |
| ICompilationUnit cu= pack1.createCompilationUnit("C.java", buf.toString(), false, null); |
| |
| String[] order= new String[] { "java", "java.util", "com", "pack" }; |
| |
| ImportRewrite imports= newImportsRewrite(cu, order, 99, 99, true); |
| imports.addImport("java.util.Vector"); |
| |
| apply(imports); |
| |
| buf= new StringBuffer(); |
| buf.append("package pack1;\n"); |
| buf.append("\n"); |
| buf.append("import java.util.Set; // comment\n"); |
| buf.append("import java.util.Vector;\n"); |
| buf.append("\n"); |
| buf.append("public class C {\n"); |
| buf.append("}\n"); |
| assertEqualString(cu.getSource(), buf.toString()); |
| } |
| |
| public void testAddImports4() throws Exception { |
| getJavaProject("P").setOption(DefaultCodeFormatterConstants.FORMATTER_INSERT_SPACE_BEFORE_SEMICOLON, JavaCore.INSERT); |
| |
| IPackageFragment pack1= this.sourceFolder.createPackageFragment("pack1", false, null); |
| StringBuffer buf= new StringBuffer(); |
| buf.append("package pack1;\n"); |
| buf.append("\n"); |
| buf.append("import java.util.Set; // comment\n"); |
| buf.append("\n"); |
| buf.append("public class C {\n"); |
| buf.append("}\n"); |
| ICompilationUnit cu= pack1.createCompilationUnit("C.java", buf.toString(), false, null); |
| |
| String[] order= new String[] { "java", "java.util", "com", "pack" }; |
| |
| ImportRewrite imports= newImportsRewrite(cu, order, 99, 99, true); |
| imports.addImport("java.util.Vector"); |
| |
| apply(imports); |
| |
| buf= new StringBuffer(); |
| buf.append("package pack1;\n"); |
| buf.append("\n"); |
| buf.append("import java.util.Set; // comment\n"); |
| buf.append("import java.util.Vector ;\n"); |
| buf.append("\n"); |
| buf.append("public class C {\n"); |
| buf.append("}\n"); |
| assertEqualString(cu.getSource(), buf.toString()); |
| } |
| |
| public void testAddImportsWithGroupsOfUnmatched1() throws Exception { |
| |
| IPackageFragment pack1= this.sourceFolder.createPackageFragment("pack1", false, null); |
| StringBuffer buf= new StringBuffer(); |
| buf.append("package pack1;\n"); |
| buf.append("\n"); |
| buf.append("public class C {\n"); |
| buf.append("}\n"); |
| ICompilationUnit cu= pack1.createCompilationUnit("C.java", buf.toString(), false, null); |
| |
| String[] order= new String[] { "java", "", "org", "#", "pack" }; |
| |
| ImportRewrite imports= newImportsRewrite(cu, order, 99, 99, true); |
| imports.addImport("org.x.Y"); |
| imports.addImport("pack.P"); |
| imports.addImport("my.M"); |
| imports.addImport("java.util.Vector"); |
| imports.addStaticImport("stat.X", "CONST", true); |
| |
| apply(imports); |
| |
| buf= new StringBuffer(); |
| buf.append("package pack1;\n"); |
| buf.append("\n"); |
| buf.append("import java.util.Vector;\n"); |
| buf.append("\n"); |
| buf.append("import my.M;\n"); |
| buf.append("\n"); |
| buf.append("import org.x.Y;\n"); |
| buf.append("\n"); |
| buf.append("import static stat.X.CONST;\n"); |
| buf.append("\n"); |
| buf.append("import pack.P;\n"); |
| buf.append("\n"); |
| buf.append("public class C {\n"); |
| buf.append("}\n"); |
| assertEqualString(cu.getSource(), buf.toString()); |
| } |
| |
| public void testAddImportsWithGroupsOfUnmatched2() throws Exception { |
| |
| IPackageFragment pack1= this.sourceFolder.createPackageFragment("pack1", false, null); |
| StringBuffer buf= new StringBuffer(); |
| buf.append("package pack1;\n"); |
| buf.append("\n"); |
| buf.append("public class C {\n"); |
| buf.append("}\n"); |
| ICompilationUnit cu= pack1.createCompilationUnit("C.java", buf.toString(), false, null); |
| |
| String[] order= new String[] { "org", "com", "pack", "#", "" }; |
| |
| ImportRewrite imports= newImportsRewrite(cu, order, 99, 99, true); |
| imports.addImport("com.x.Y"); |
| imports.addImport("pack.P"); |
| imports.addImport("my.M"); |
| imports.addImport("org.Vector"); |
| imports.addStaticImport("stat.X", "CONST", true); |
| |
| apply(imports); |
| |
| buf= new StringBuffer(); |
| buf.append("package pack1;\n"); |
| buf.append("\n"); |
| buf.append("import org.Vector;\n"); |
| buf.append("\n"); |
| buf.append("import com.x.Y;\n"); |
| buf.append("\n"); |
| buf.append("import pack.P;\n"); |
| buf.append("\n"); |
| buf.append("import static stat.X.CONST;\n"); |
| buf.append("\n"); |
| buf.append("import my.M;\n"); |
| buf.append("\n"); |
| buf.append("public class C {\n"); |
| buf.append("}\n"); |
| assertEqualString(cu.getSource(), buf.toString()); |
| } |
| |
| public void testRemoveImports1() throws Exception { |
| |
| IPackageFragment pack1= this.sourceFolder.createPackageFragment("pack1", false, null); |
| StringBuffer buf= new StringBuffer(); |
| buf.append("package pack1;\n"); |
| buf.append("\n"); |
| buf.append("import java.util.Set;\n"); |
| buf.append("import java.util.Vector;\n"); |
| buf.append("import java.util.Map;\n"); |
| buf.append("\n"); |
| buf.append("import pack.List;\n"); |
| buf.append("import pack.List2;\n"); |
| buf.append("\n"); |
| buf.append("public class C {\n"); |
| buf.append("}\n"); |
| ICompilationUnit cu= pack1.createCompilationUnit("C.java", buf.toString(), false, null); |
| |
| String[] order= new String[] { "java", "com", "pack" }; |
| |
| ImportRewrite imports= newImportsRewrite(cu, order, 2, 2, true); |
| imports.removeImport("java.util.Set"); |
| imports.removeImport("pack.List"); |
| |
| apply(imports); |
| |
| buf= new StringBuffer(); |
| buf.append("package pack1;\n"); |
| buf.append("\n"); |
| buf.append("import java.util.Vector;\n"); |
| buf.append("import java.util.Map;\n"); |
| buf.append("\n"); |
| buf.append("import pack.List2;\n"); |
| buf.append("\n"); |
| buf.append("public class C {\n"); |
| buf.append("}\n"); |
| assertEqualString(cu.getSource(), buf.toString()); |
| } |
| |
| public void testRemoveImports2() throws Exception { |
| |
| IPackageFragment pack1= this.sourceFolder.createPackageFragment("pack1", false, null); |
| StringBuffer buf= new StringBuffer(); |
| buf.append("package pack1;\n"); |
| buf.append("\n"); |
| buf.append("import java.util.Set;\n"); |
| buf.append("import java.util.Vector; // comment\n"); |
| buf.append("\n"); |
| buf.append("public class C {\n"); |
| buf.append("}\n"); |
| ICompilationUnit cu= pack1.createCompilationUnit("C.java", buf.toString(), false, null); |
| |
| String[] order= new String[] { "java", "com", "pack" }; |
| |
| ImportRewrite imports= newImportsRewrite(cu, order, 2, 2, true); |
| imports.removeImport("java.util.Vector"); |
| |
| apply(imports); |
| |
| buf= new StringBuffer(); |
| buf.append("package pack1;\n"); |
| buf.append("\n"); |
| buf.append("import java.util.Set;\n"); |
| buf.append("\n"); |
| buf.append("public class C {\n"); |
| buf.append("}\n"); |
| assertEqualString(cu.getSource(), buf.toString()); |
| } |
| |
| |
| public void testAddImports_bug23078() throws Exception { |
| |
| IPackageFragment pack1= this.sourceFolder.createPackageFragment("pack1", false, null); |
| StringBuffer buf= new StringBuffer(); |
| buf.append("package pack1;\n"); |
| buf.append("\n"); |
| buf.append("import p.A.*;\n"); |
| buf.append("\n"); |
| buf.append("public class C {\n"); |
| buf.append("}\n"); |
| ICompilationUnit cu= pack1.createCompilationUnit("C.java", buf.toString(), false, null); |
| |
| String[] order= new String[] { }; |
| |
| ImportRewrite imports= newImportsRewrite(cu, order, 2, 2, true); |
| imports.addImport("p.Inner"); |
| |
| apply(imports); |
| |
| buf= new StringBuffer(); |
| buf.append("package pack1;\n"); |
| buf.append("\n"); |
| buf.append("import p.Inner;\n"); |
| buf.append("import p.A.*;\n"); |
| buf.append("\n"); |
| buf.append("public class C {\n"); |
| buf.append("}\n"); |
| assertEqualString(cu.getSource(), buf.toString()); |
| } |
| |
| public void testAddImports_bug25113() throws Exception { |
| |
| IPackageFragment pack1= this.sourceFolder.createPackageFragment("pack1", false, null); |
| StringBuffer buf= new StringBuffer(); |
| buf.append("package pack1;\n"); |
| buf.append("\n"); |
| buf.append("import java.awt.Panel;\n"); |
| buf.append("\n"); |
| buf.append("import java.math.BigInteger;\n"); |
| buf.append("\n"); |
| buf.append("public class C {\n"); |
| buf.append("}\n"); |
| ICompilationUnit cu= pack1.createCompilationUnit("C.java", buf.toString(), false, null); |
| |
| String[] order= new String[] { "java.awt", "java" }; |
| |
| ImportRewrite imports= newImportsRewrite(cu, order, 99, 99, true); |
| imports.addImport("java.applet.Applet"); |
| |
| apply(imports); |
| |
| buf= new StringBuffer(); |
| buf.append("package pack1;\n"); |
| buf.append("\n"); |
| buf.append("import java.awt.Panel;\n"); |
| buf.append("\n"); |
| buf.append("import java.applet.Applet;\n"); |
| buf.append("import java.math.BigInteger;\n"); |
| buf.append("\n"); |
| buf.append("public class C {\n"); |
| buf.append("}\n"); |
| assertEqualString(cu.getSource(), buf.toString()); |
| } |
| |
| public void testAddImports_bug42637() throws Exception { |
| |
| IPackageFragment pack1= this.sourceFolder.createPackageFragment("pack1", false, null); |
| StringBuffer buf= new StringBuffer(); |
| buf.append("package pack1;\n"); |
| buf.append("\n"); |
| buf.append("import java.lang.System;\n"); |
| buf.append("\n"); |
| buf.append("public class C {\n"); |
| buf.append("}\n"); |
| ICompilationUnit cu= pack1.createCompilationUnit("C.java", buf.toString(), false, null); |
| |
| String[] order= new String[] { "java" }; |
| |
| ImportRewrite imports= newImportsRewrite(cu, order, 99, 99, true); |
| imports.addImport("java.io.Exception"); |
| |
| apply(imports); |
| |
| buf= new StringBuffer(); |
| buf.append("package pack1;\n"); |
| buf.append("\n"); |
| buf.append("import java.io.Exception;\n"); |
| buf.append("import java.lang.System;\n"); |
| buf.append("\n"); |
| buf.append("public class C {\n"); |
| buf.append("}\n"); |
| assertEqualString(cu.getSource(), buf.toString()); |
| } |
| |
| public void testAddImports_bug121428() throws Exception { |
| |
| IPackageFragment pack1= this.sourceFolder.createPackageFragment("pack1", false, null); |
| StringBuffer buf= new StringBuffer(); |
| buf.append("package pack1;\n"); |
| buf.append("\n"); |
| buf.append("/** comment */\n"); |
| buf.append("import java.lang.System;\n"); |
| buf.append("\n"); |
| buf.append("public class C {\n"); |
| buf.append("}\n"); |
| ICompilationUnit cu= pack1.createCompilationUnit("C.java", buf.toString(), false, null); |
| |
| String[] order= new String[] { "java" }; |
| |
| ImportRewrite imports= newImportsRewrite(cu, order, 99, 99, false); |
| imports.addImport("java.io.Exception"); |
| |
| apply(imports); |
| |
| buf= new StringBuffer(); |
| buf.append("package pack1;\n"); |
| buf.append("\n"); |
| buf.append("/** comment */\n"); |
| buf.append("import java.io.Exception;\n"); |
| buf.append("\n"); |
| buf.append("public class C {\n"); |
| buf.append("}\n"); |
| assertEqualString(cu.getSource(), buf.toString()); |
| } |
| |
| public void testAddStaticImports1() throws Exception { |
| |
| IPackageFragment pack1= this.sourceFolder.createPackageFragment("pack1", false, null); |
| StringBuffer buf= new StringBuffer(); |
| buf.append("package pack1;\n"); |
| buf.append("\n"); |
| buf.append("import java.lang.System;\n"); |
| buf.append("\n"); |
| buf.append("public class C {\n"); |
| buf.append("}\n"); |
| ICompilationUnit cu= pack1.createCompilationUnit("C.java", buf.toString(), false, null); |
| |
| String[] order= new String[] { "#", "java" }; |
| |
| ImportRewrite imports= newImportsRewrite(cu, order, 99, 99, true); |
| imports.addStaticImport("java.lang.Math", "min", true); |
| imports.addImport("java.lang.Math"); |
| imports.addStaticImport("java.lang.Math", "max", true); |
| |
| apply(imports); |
| |
| buf= new StringBuffer(); |
| buf.append("package pack1;\n"); |
| buf.append("\n"); |
| buf.append("import static java.lang.Math.max;\n"); |
| buf.append("import static java.lang.Math.min;\n"); |
| buf.append("\n"); |
| buf.append("import java.lang.System;\n"); |
| buf.append("\n"); |
| buf.append("public class C {\n"); |
| buf.append("}\n"); |
| assertEqualString(cu.getSource(), buf.toString()); |
| } |
| |
| public void testAddStaticImports2() throws Exception { |
| |
| IPackageFragment pack1= this.sourceFolder.createPackageFragment("pack1", false, null); |
| StringBuffer buf= new StringBuffer(); |
| buf.append("package pack1;\n"); |
| buf.append("\n"); |
| buf.append("import java.lang.System;\n"); |
| buf.append("\n"); |
| buf.append("public class C {\n"); |
| buf.append("}\n"); |
| ICompilationUnit cu= pack1.createCompilationUnit("C.java", buf.toString(), false, null); |
| |
| String[] order= new String[] { "#", "java" }; |
| |
| ImportRewrite imports= newImportsRewrite(cu, order, 99, 99, true); |
| imports.addStaticImport("xx.MyConstants", "SIZE", true); |
| imports.addStaticImport("xy.MyConstants", "*", true); |
| imports.addImport("xy.MyConstants"); |
| |
| apply(imports); |
| |
| buf= new StringBuffer(); |
| buf.append("package pack1;\n"); |
| buf.append("\n"); |
| buf.append("import static xx.MyConstants.SIZE;\n"); |
| buf.append("import static xy.MyConstants.*;\n"); |
| buf.append("\n"); |
| buf.append("import java.lang.System;\n"); |
| buf.append("\n"); |
| buf.append("import xy.MyConstants;\n"); |
| buf.append("\n"); |
| buf.append("public class C {\n"); |
| buf.append("}\n"); |
| assertEqualString(cu.getSource(), buf.toString()); |
| } |
| |
| public void testAddStaticImports3() throws Exception { |
| |
| IPackageFragment pack1= this.sourceFolder.createPackageFragment("pack1", false, null); |
| StringBuffer buf= new StringBuffer(); |
| buf.append("package pack1;\n"); |
| buf.append("\n"); |
| buf.append("import java.lang.System;\n"); |
| buf.append("\n"); |
| buf.append("public class C {\n"); |
| buf.append("}\n"); |
| ICompilationUnit cu= pack1.createCompilationUnit("C.java", buf.toString(), false, null); |
| |
| String[] order= new String[] { "#", "java" }; |
| |
| ImportRewrite imports= newImportsRewrite(cu, order, 99, 3, true); |
| imports.addStaticImport("java.lang.Math", "min", true); |
| imports.addStaticImport("java.lang.Math", "max", true); |
| imports.addStaticImport("java.lang.Math", "abs", true); |
| |
| imports.addStaticImport("java.io.File", "pathSeparator", true); |
| imports.addStaticImport("java.io.File", "separator", true); |
| |
| imports.addImport("java.util.List"); |
| imports.addImport("java.util.Vector"); |
| imports.addImport("java.util.ArrayList"); |
| |
| apply(imports); |
| |
| buf= new StringBuffer(); |
| buf.append("package pack1;\n"); |
| buf.append("\n"); |
| buf.append("import static java.io.File.pathSeparator;\n"); |
| buf.append("import static java.io.File.separator;\n"); |
| buf.append("import static java.lang.Math.*;\n"); |
| buf.append("\n"); |
| buf.append("import java.lang.System;\n"); |
| buf.append("import java.util.ArrayList;\n"); |
| buf.append("import java.util.List;\n"); |
| buf.append("import java.util.Vector;\n"); |
| buf.append("\n"); |
| buf.append("public class C {\n"); |
| buf.append("}\n"); |
| assertEqualString(cu.getSource(), buf.toString()); |
| } |
| |
| |
| private void createClassStub(String pack, String typeName, String typeKind) throws JavaModelException { |
| IPackageFragment pack1= this.sourceFolder.createPackageFragment(pack, false, null); |
| StringBuffer buf= new StringBuffer(); |
| buf.append("package ").append(pack).append(";\n"); |
| buf.append("public ").append(typeKind).append(" ").append(typeName).append(" {\n"); |
| buf.append("}\n"); |
| String content= buf.toString(); |
| |
| String name= typeName; |
| int idx= typeName.indexOf('<'); |
| if (idx != -1) { |
| name= typeName.substring(0, idx); |
| } |
| pack1.createCompilationUnit(name + ".java", content, false, null); |
| } |
| |
| |
| public void testImportStructureWithSignatures() throws Exception { |
| createClassStub("java.io", "IOException", "class"); |
| createClassStub("java.net", "URL", "class"); |
| createClassStub("java.util", "List<E>", "interface"); |
| createClassStub("java.net", "SocketAddress", "class"); |
| |
| IPackageFragment pack1= this.sourceFolder.createPackageFragment("test1", false, null); |
| StringBuffer buf= new StringBuffer(); |
| buf.append("package test1;\n"); |
| buf.append("import java.util.*;\n"); |
| buf.append("import java.net.*;\n"); |
| buf.append("import java.io.*;\n"); |
| buf.append("public class A {\n"); |
| buf.append(" public void foo() {\n"); |
| buf.append(" IOException s;\n"); |
| buf.append(" URL[][] t;\n"); |
| buf.append(" List<SocketAddress> x;\n"); |
| buf.append(" }\n"); |
| buf.append("}\n"); |
| String content= buf.toString(); |
| ICompilationUnit cu1= pack1.createCompilationUnit("A.java", content, false, null); |
| |
| buf= new StringBuffer(); |
| buf.append("package test1;\n"); |
| buf.append("public class B {\n"); |
| buf.append("}\n"); |
| String content2= buf.toString(); |
| ICompilationUnit cu2= pack1.createCompilationUnit("B.java", content2, false, null); |
| |
| String[] order= new String[] { "java.util", "java.io", "java.net" }; |
| int threshold= 99; |
| AST ast= AST.newAST(AST.JLS3); |
| ImportRewrite importsRewrite= newImportsRewrite(cu2, order, threshold, threshold, true); |
| { |
| IJavaElement[] elements= cu1.codeSelect(content.indexOf("IOException"), "IOException".length()); |
| assertEquals(1, elements.length); |
| String key= ((IType) elements[0]).getKey(); |
| String signature= new BindingKey(key).toSignature(); |
| |
| importsRewrite.addImportFromSignature(signature, ast); |
| } |
| { |
| IJavaElement[] elements= cu1.codeSelect(content.indexOf("URL"), "URL".length()); |
| assertEquals(1, elements.length); |
| String key= ((IType) elements[0]).getKey(); |
| String signature= new BindingKey(key).toSignature(); |
| |
| importsRewrite.addImportFromSignature(signature, ast); |
| } |
| { |
| IJavaElement[] elements= cu1.codeSelect(content.indexOf("List"), "List".length()); |
| assertEquals(1, elements.length); |
| String key= ((IType) elements[0]).getKey(); |
| String signature= new BindingKey(key).toSignature(); |
| |
| importsRewrite.addImportFromSignature(signature, ast); |
| } |
| apply(importsRewrite); |
| |
| buf= new StringBuffer(); |
| buf.append("package test1;\n"); |
| buf.append("\n"); |
| buf.append("import java.util.List;\n"); |
| buf.append("\n"); |
| buf.append("import java.io.IOException;\n"); |
| buf.append("\n"); |
| buf.append("import java.net.SocketAddress;\n"); |
| buf.append("import java.net.URL;\n"); |
| buf.append("\n"); |
| buf.append("public class B {\n"); |
| buf.append("}\n"); |
| |
| assertEqualStringIgnoreDelim(cu2.getSource(), buf.toString()); |
| } |
| |
| public void testImportStructureWithSignatures2() throws Exception { |
| createClassStub("java.util", "Map<S, T>", "interface"); |
| createClassStub("java.util", "Set<S>", "interface"); |
| createClassStub("java.net", "SocketAddress", "class"); |
| createClassStub("java.net", "ServerSocket", "class"); |
| |
| IPackageFragment pack1= this.sourceFolder.createPackageFragment("test1", false, null); |
| StringBuffer buf= new StringBuffer(); |
| buf.append("package test1;\n"); |
| buf.append("import java.util.*;\n"); |
| buf.append("import java.net.*;\n"); |
| buf.append("import java.io.*;\n"); |
| buf.append("public class A {\n"); |
| buf.append(" public void foo() {\n"); |
| buf.append(" Map<?, ? extends Set<? super ServerSocket>> z;\n"); |
| buf.append(" }\n"); |
| buf.append("}\n"); |
| String content= buf.toString(); |
| ICompilationUnit cu1= pack1.createCompilationUnit("A.java", content, false, null); |
| |
| buf= new StringBuffer(); |
| buf.append("package test1;\n"); |
| buf.append("public class B {\n"); |
| buf.append("}\n"); |
| String content2= buf.toString(); |
| ICompilationUnit cu2= pack1.createCompilationUnit("B.java", content2, false, null); |
| |
| String[] order= new String[] { "java.util", "java.io", "java.net" }; |
| int threshold= 99; |
| AST ast= AST.newAST(AST.JLS3); |
| ImportRewrite importsRewrite= newImportsRewrite(cu2, order, threshold, threshold, true); |
| { |
| IJavaElement[] elements= cu1.codeSelect(content.indexOf("Map"), "Map".length()); |
| assertEquals(1, elements.length); |
| String key= ((IType) elements[0]).getKey(); |
| String signature= new BindingKey(key).toSignature(); |
| |
| importsRewrite.addImportFromSignature(signature, ast); |
| } |
| |
| apply(importsRewrite); |
| |
| buf= new StringBuffer(); |
| buf.append("package test1;\n"); |
| buf.append("\n"); |
| buf.append("import java.util.Map;\n"); |
| buf.append("import java.util.Set;\n"); |
| buf.append("\n"); |
| buf.append("import java.net.ServerSocket;\n"); |
| buf.append("\n"); |
| buf.append("public class B {\n"); |
| buf.append("}\n"); |
| |
| assertEqualStringIgnoreDelim(cu2.getSource(), buf.toString()); |
| } |
| |
| |
| public void testAddedRemovedImportsAPI() throws Exception { |
| IPackageFragment pack1= this.sourceFolder.createPackageFragment("pack1", false, null); |
| StringBuffer buf= new StringBuffer(); |
| buf.append("package pack1;\n"); |
| buf.append("\n"); |
| buf.append("import java.util.Vector;\n"); |
| buf.append("\n"); |
| buf.append("public class C {\n"); |
| buf.append(" public final static int CONST= 9;\n"); |
| buf.append("}\n"); |
| ICompilationUnit cu= pack1.createCompilationUnit("C.java", buf.toString(), false, null); |
| |
| String[] order= new String[] { "#", "java" }; |
| |
| ImportRewrite imports= newImportsRewrite(cu, order, 99, 99, true); |
| imports.addStaticImport("java.lang.Math", "min", true); |
| imports.addImport("java.lang.Math"); |
| |
| assertAddedAndRemoved(imports, |
| new String[] { "java.lang.Math" }, new String[] {}, |
| new String[] { "java.lang.Math.min" }, new String[] {} |
| ); |
| |
| imports.addImport("java.lang.Math"); |
| imports.addStaticImport("java.lang.Math", "max", true); |
| |
| assertAddedAndRemoved(imports, |
| new String[] { "java.lang.Math" }, new String[] {}, |
| new String[] { "java.lang.Math.min", "java.lang.Math.max" }, new String[] {} |
| ); |
| |
| imports.removeImport("java.lang.Math"); |
| imports.removeImport("java.util.Vector"); |
| imports.removeStaticImport("java.lang.Math.dup"); |
| |
| assertAddedAndRemoved(imports, |
| new String[] { }, new String[] { "java.util.Vector"}, |
| new String[] { "java.lang.Math.min", "java.lang.Math.max" }, new String[] {} |
| ); |
| |
| imports.addImport("java.util.Vector"); |
| imports.addStaticImport("pack1.C", "CONST", true); |
| |
| assertAddedAndRemoved(imports, |
| new String[] { }, new String[] { }, |
| new String[] { "java.lang.Math.min", "java.lang.Math.max", "pack1.C.CONST" }, new String[] {} |
| ); |
| |
| apply(imports); |
| |
| buf= new StringBuffer(); |
| buf.append("package pack1;\n"); |
| buf.append("\n"); |
| buf.append("import static java.lang.Math.max;\n"); |
| buf.append("import static java.lang.Math.min;\n"); |
| buf.append("import static pack1.C.CONST;\n"); |
| buf.append("\n"); |
| buf.append("import java.util.Vector;\n"); |
| buf.append("\n"); |
| buf.append("public class C {\n"); |
| buf.append(" public final static int CONST= 9;\n"); |
| buf.append("}\n"); |
| assertEqualString(cu.getSource(), buf.toString()); |
| } |
| |
| public void testPackageInfo() throws Exception { |
| IPackageFragment pack1= this.sourceFolder.createPackageFragment("pack1", false, null); |
| StringBuffer buf= new StringBuffer(); |
| buf.append("\npackage pack1;"); |
| |
| ICompilationUnit cu= pack1.createCompilationUnit("package-info.java", buf.toString(), false, null); |
| |
| String[] order= new String[] { "#", "java" }; |
| |
| ImportRewrite imports= newImportsRewrite(cu, order, 99, 99, true); |
| imports.addImport("foo.Bar"); |
| |
| apply(imports); |
| |
| buf= new StringBuffer(); |
| buf.append("\npackage pack1;\n"); |
| buf.append("import foo.Bar;\n"); |
| assertEqualString(cu.getSource(), buf.toString()); |
| } |
| |
| |
| private void assertAddedAndRemoved(ImportRewrite imports, String[] expectedAdded, String[] expectedRemoved, String[] expectedAddedStatic, String[] expectedRemovedStatic) { |
| assertEqualStringsIgnoreOrder(imports.getAddedImports(), expectedAdded); |
| assertEqualStringsIgnoreOrder(imports.getAddedStaticImports(), expectedAddedStatic); |
| assertEqualStringsIgnoreOrder(imports.getRemovedImports(), expectedRemoved); |
| assertEqualStringsIgnoreOrder(imports.getRemovedStaticImports(), expectedRemovedStatic); |
| } |
| |
| private void assertEqualString(String actual, String expected) { |
| StringAsserts.assertEqualString(actual, expected); |
| } |
| |
| private void assertEqualStringsIgnoreOrder(String[] actual, String[] expecteds) { |
| StringAsserts.assertEqualStringsIgnoreOrder(actual, expecteds); |
| } |
| |
| private void assertEqualStringIgnoreDelim(String actual, String expected) throws IOException { |
| StringAsserts.assertEqualStringIgnoreDelim(actual, expected); |
| } |
| |
| private ImportRewrite newImportsRewrite(ICompilationUnit cu, String[] order, int normalThreshold, int staticThreshold, boolean restoreExistingImports) throws CoreException, BackingStoreException { |
| ImportRewrite rewrite= ImportRewrite.create(cu, restoreExistingImports); |
| rewrite.setImportOrder(order); |
| rewrite.setOnDemandImportThreshold(normalThreshold); |
| rewrite.setStaticOnDemandImportThreshold(staticThreshold); |
| return rewrite; |
| } |
| |
| private void apply(ImportRewrite rewrite) throws CoreException, MalformedTreeException, BadLocationException { |
| TextEdit edit= rewrite.rewriteImports(null); |
| |
| // not the efficient way! |
| ICompilationUnit compilationUnit= rewrite.getCompilationUnit(); |
| Document document= new Document(compilationUnit.getSource()); |
| edit.apply(document); |
| compilationUnit.getBuffer().setContents(document.get()); |
| } |
| |
| } |
